From 0e569bacf7a1ccf2b4d6aa57089cb135ac7d46cb Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 17 Jun 2024 19:47:48 -0700 Subject: [PATCH] test - logfire --- litellm/tests/test_logfire.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_logfire.py b/litellm/tests/test_logfire.py index 5ce06ea69c..34bd75ccae 100644 --- a/litellm/tests/test_logfire.py +++ b/litellm/tests/test_logfire.py @@ -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