diff --git a/tests/local_testing/test_arize_ai.py b/tests/local_testing/test_arize_ai.py index 9e668140d06..a2d3c9cab80 100644 --- a/tests/local_testing/test_arize_ai.py +++ b/tests/local_testing/test_arize_ai.py @@ -1,6 +1,7 @@ import asyncio import json import logging +import os from litellm import Choices import pytest @@ -32,6 +33,26 @@ async def test_async_otel_callback(): await asyncio.sleep(2) +@pytest.mark.asyncio() +async def test_async_dynamic_arize_config(): + litellm.set_verbose = True + + verbose_proxy_logger.setLevel(logging.DEBUG) + verbose_logger.setLevel(logging.DEBUG) + litellm.success_callback = ["arize"] + + await litellm.acompletion( + model="gpt-3.5-turbo", + messages=[{"role": "user", "content": "hi test from arize dynamic config"}], + temperature=0.1, + user="OTEL_USER", + arize_api_key=os.getenv("ARIZE_SPACE_2_API_KEY"), + arize_space_key=os.getenv("ARIZE_SPACE_2_KEY"), + ) + + await asyncio.sleep(2) + + @pytest.fixture def mock_env_vars(monkeypatch): monkeypatch.setenv("ARIZE_SPACE_KEY", "test_space_key")