tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-05 20:23:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_otel.py): fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2024-09-23 08:10:06 -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
tests/otel_tests/test_otel.py
+1 -5
View File
@@ -111,11 +111,7 @@ async def test_chat_completion_check_otel_spans():
print("Parent trace spans: ", parent_trace_spans)
# either 5 or 6 traces depending on how many redis calls were made
assert (
len(parent_trace_spans) == 6
or len(parent_trace_spans) == 5
or len(parent_trace_spans) == 7
)
assert len(parent_trace_spans) >= 5
# 'postgres', 'redis', 'raw_gen_ai_request', 'litellm_request', 'Received Proxy Server Request' in the span
assert "postgres" in parent_trace_spans