ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test - logfire

Browse Source
This commit is contained in:
Ishaan Jaff 2024-06-17 19:47:48 -07:00
parent 991604986d
commit 0e569bacf7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/tests/test_logfire.py
View File

@ -26,7 +26,7 @@ sys.path.insert(0, os.path.abspath("../.."))
def test_completion_logfire_logging(stream):
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
litellm.success_callback = ["logfire"]
litellm.callbacks = ["logfire"]
litellm.set_verbose = True
messages = [{"role": "user", "content": "what llm are u"}]
temperature = 0.3
@ -53,7 +53,7 @@ def test_completion_logfire_logging(stream):
async def test_acompletion_logfire_logging(stream):
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
litellm.success_callback = ["logfire"]
litellm.callbacks = ["logfire"]
litellm.set_verbose = True
messages = [{"role": "user", "content": "what llm are u"}]
temperature = 0.3