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

test(test_otel.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-09-23 08:09:54 -07:00
parent 030b2e1bae
commit 2a8eb492a1
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/otel_tests/test_otel.py
View file

@ -111,11 +111,7 @@ async def test_chat_completion_check_otel_spans():
print("Parent trace spans: ", parent_trace_spans) print("Parent trace spans: ", parent_trace_spans)
# either 5 or 6 traces depending on how many redis calls were made # either 5 or 6 traces depending on how many redis calls were made
assert ( assert len(parent_trace_spans) >= 5
len(parent_trace_spans) == 6
or len(parent_trace_spans) == 5
or len(parent_trace_spans) == 7
)
# 'postgres', 'redis', 'raw_gen_ai_request', 'litellm_request', 'Received Proxy Server Request' in the span # 'postgres', 'redis', 'raw_gen_ai_request', 'litellm_request', 'Received Proxy Server Request' in the span
assert "postgres" in parent_trace_spans assert "postgres" in parent_trace_spans