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 = {
"trace_name",
"trace_id",
"existing_trace_id",
"update_trace_keys",
"trace_user_id",
"session_id",
"tags",
"generation_name",
"generation_id",
"prompt",
"cache_hit",
}
trace_metadata = {
"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.trace_id == trace_id
print(
"diff in generation keys",
set(generation.metadata.keys()).difference(
"common keys in trace",
set(generation.metadata.keys()).intersection(
expected_filtered_metadata_keys
),
)