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

test otel

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-11 09:54:23 -07:00
parent e2c3b2b694
commit 15d35cd62f
2 changed files with 31 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/otel_tests/test_otel.py
View file

@ -100,12 +100,17 @@ async def test_chat_completion_check_otel_spans():
print("otel_spans: ", otel_spans)
all_otel_spans = otel_spans["otel_spans"]
most_recent_parent = str(otel_spans["most_recent_parent"])
print("Most recent OTEL parent: ", most_recent_parent)
print("\n spans grouped by parent: ", otel_spans["spans_grouped_by_parent"])
parent_trace_spans = otel_spans["spans_grouped_by_parent"][most_recent_parent]
assert len(all_otel_spans) == 5
print("Parent trace spans: ", parent_trace_spans)
assert len(parent_trace_spans) == 4
# 'postgres', 'redis', 'raw_gen_ai_request', 'litellm_request', 'Received Proxy Server Request' in the span
assert "postgres" in all_otel_spans
assert "redis" in all_otel_spans
assert "raw_gen_ai_request" in all_otel_spans
assert "litellm_request" in all_otel_spans
assert "Received Proxy Server Request" in all_otel_spans
assert "postgres" in parent_trace_spans
assert "redis" in parent_trace_spans
assert "raw_gen_ai_request" in parent_trace_spans
assert "litellm_request" in parent_trace_spans