From cdc8b4f0374429013493036c1104313715ef7167 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 22 Jul 2024 11:30:24 -0700 Subject: [PATCH] test - arize ai basic logging --- litellm/tests/test_arize_ai.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 litellm/tests/test_arize_ai.py diff --git a/litellm/tests/test_arize_ai.py b/litellm/tests/test_arize_ai.py new file mode 100644 index 000000000..7c38db4c6 --- /dev/null +++ b/litellm/tests/test_arize_ai.py @@ -0,0 +1,29 @@ +import asyncio +import logging +import os +import time + +import pytest +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.integrations.opentelemetry import OpenTelemetry, OpenTelemetryConfig + +load_dotenv() +import logging + + +@pytest.mark.asyncio() +async def test_async_otel_callback(): + litellm.set_verbose = True + litellm.callbacks = ["arize"] + + await litellm.acompletion( + model="gpt-3.5-turbo", + messages=[{"role": "user", "content": "hi test from local arize"}], + mock_response="hello", + temperature=0.1, + user="OTEL_USER", + )