From 97ba230b7ae21bb27ade2893ebd166ed3d0013f1 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 11 May 2024 10:20:30 -0700
Subject: [PATCH] fix langfuse test

---
 litellm/tests/test_alangfuse.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py
index 5a495550c..37737d40b 100644
--- a/litellm/tests/test_alangfuse.py
+++ b/litellm/tests/test_alangfuse.py
@@ -248,12 +248,14 @@ async def test_langfuse_logging_metadata(langfuse_client):
         "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"
@@ -339,6 +341,13 @@ async def test_langfuse_logging_metadata(langfuse_client):
         for generation_id, generation in zip(generation_ids, generations):
             assert generation.id == generation_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(
                 expected_filtered_metadata_keys
             )