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

fix test_awesome_otel_with_message_logging_off

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-09 17:59:07 -07:00
parent 16b6b56c8b
commit 569f3ddda9
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_async_opentelemetry.py
View file

@ -34,6 +34,8 @@ async def test_awesome_otel_with_message_logging_off():
) )
litellm.callbacks = [otel_logger] litellm.callbacks = [otel_logger]
litellm.success_callback = []
litellm.failure_callback = []
response = await litellm.acompletion( response = await litellm.acompletion(
model="gpt-3.5-turbo", model="gpt-3.5-turbo",
@ -42,7 +44,7 @@ async def test_awesome_otel_with_message_logging_off():
) )
print("response", response) print("response", response)
await asyncio.sleep(4) await asyncio.sleep(5)
assert otel_logger.kwargs["messages"] == [ assert otel_logger.kwargs["messages"] == [
{"role": "user", "content": "redacted-by-litellm"} {"role": "user", "content": "redacted-by-litellm"}