diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py index 37737d40b..6f9b8d626 100644 --- a/litellm/tests/test_alangfuse.py +++ b/litellm/tests/test_alangfuse.py @@ -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 ), )