phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix arize handle optional params (#6243)

Browse source
This commit is contained in:
Ishaan Jaff 2024-10-16 08:33:40 +05:30 • committed by GitHub
parent 1994100028
commit 6909d8e11b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 106 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/local_testing/test_arize_ai.py
View file

@ -8,7 +8,7 @@ from dotenv import load_dotenv
from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
import litellm
from litellm._logging import verbose_logger
from litellm._logging import verbose_logger, verbose_proxy_logger
from litellm.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig
load_dotenv()
@ -18,6 +18,9 @@ import logging
@pytest.mark.asyncio()
async def test_async_otel_callback():
litellm.set_verbose = True
verbose_proxy_logger.setLevel(logging.DEBUG)
verbose_logger.setLevel(logging.DEBUG)
litellm.success_callback = ["arize"]
await litellm.acompletion(
@ -27,3 +30,5 @@ async def test_async_otel_callback():
temperature=0.1,
user="OTEL_USER",
)
await asyncio.sleep(2)