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

fix - langfuse trace

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-11 10:28:13 -07:00
parent 97ba230b7a
commit 6577719bf8
1 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_alangfuse.py
View file

@ -247,15 +247,11 @@ async def test_langfuse_logging_metadata(langfuse_client):
expected_filtered_metadata_keys = { expected_filtered_metadata_keys = {
"trace_name", "trace_name",
"trace_id", "trace_id",
"existing_trace_id",
"update_trace_keys",
"trace_user_id", "trace_user_id",
"session_id",
"tags", "tags",
"generation_name", "generation_name",
"generation_id", "generation_id",
"prompt", "prompt",
"cache_hit",
} }
trace_metadata = { trace_metadata = {
"trace_actual_metadata_key": "trace_actual_metadata_value" "trace_actual_metadata_key": "trace_actual_metadata_value"
@ -342,8 +338,8 @@ async def test_langfuse_logging_metadata(langfuse_client):
assert generation.id == generation_id assert generation.id == generation_id
assert generation.trace_id == trace_id assert generation.trace_id == trace_id
print( print(
"diff in generation keys", "common keys in trace",
set(generation.metadata.keys()).difference( set(generation.metadata.keys()).intersection(
expected_filtered_metadata_keys expected_filtered_metadata_keys
), ),
) )