fix using arize as success callback

This commit is contained in:
Ishaan Jaff 2024-07-22 15:32:49 -07:00
parent d3bced56bb
commit a0600a30d8

View file

@ -18,7 +18,7 @@ import logging
@pytest.mark.asyncio() @pytest.mark.asyncio()
async def test_async_otel_callback(): async def test_async_otel_callback():
litellm.set_verbose = True litellm.set_verbose = True
litellm.callbacks = ["arize"] litellm.success_callback = ["arize"]
await litellm.acompletion( await litellm.acompletion(
model="gpt-3.5-turbo", model="gpt-3.5-turbo",