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

fix langfuse test

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-11 10:20:30 -07:00
parent ebb5c76e37
commit 97ba230b7a
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_alangfuse.py
View file

@ -248,12 +248,14 @@ async def test_langfuse_logging_metadata(langfuse_client):
"trace_name", "trace_name",
"trace_id", "trace_id",
"existing_trace_id", "existing_trace_id",
"update_trace_keys",
"trace_user_id", "trace_user_id",
"session_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"
@ -339,6 +341,13 @@ async def test_langfuse_logging_metadata(langfuse_client):
for generation_id, generation in zip(generation_ids, generations): for generation_id, generation in zip(generation_ids, generations):
assert generation.id == generation_id assert generation.id == generation_id
assert generation.trace_id == trace_id assert generation.trace_id == trace_id
print(
"diff in generation keys",
set(generation.metadata.keys()).difference(
expected_filtered_metadata_keys
),
)
assert set(generation.metadata.keys()).isdisjoint( assert set(generation.metadata.keys()).isdisjoint(
expected_filtered_metadata_keys expected_filtered_metadata_keys
) )