From 6577719bf8a78db887c1de254e00f01c83b01ebf Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 11 May 2024 10:28:13 -0700
Subject: [PATCH] fix - langfuse trace

---
 litellm/tests/test_alangfuse.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

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
                 ),
             )