From f50b890c576e0ce2612e68660bdd535d80c576bd Mon Sep 17 00:00:00 2001
From: jesus <jesus@berri.ai>
Date: Mon, 7 Sep 2026 16:05:38 +0000
Subject: [PATCH] test(langfuse): avoid patching litellm internals in session
 trace tests

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../integrations/test_langfuse.py             | 35 +++++++++----------
 1 file changed, 16 insertions(+), 19 deletions(-)

diff --git a/tests/test_litellm/integrations/test_langfuse.py b/tests/test_litellm/integrations/test_langfuse.py
index 608721998fb..f5668a8e254 100644
--- a/tests/test_litellm/integrations/test_langfuse.py
+++ b/tests/test_litellm/integrations/test_langfuse.py
@@ -908,26 +908,23 @@ class TestLangfuseUsageDetails(unittest.TestCase):
         kwargs = self._build_langfuse_kwargs(payload)
         kwargs["litellm_trace_id"] = "sess-abc"
         self.last_trace_kwargs = {}
+        # Below the prompt-management SDK floor so the call needs no prompt patching
+        self.logger.langfuse_sdk_version = "2.6.3"
 
-        with patch(
-            "litellm.integrations.langfuse.langfuse._add_prompt_to_generation_params",
-            side_effect=lambda generation_params, **kwargs: generation_params,
-            create=True,
-        ):
-            self.logger._log_langfuse_v2(
-                user_id="user-1",
-                metadata={"session_id": "sess-abc", "trace_id": metadata_trace_id},
-                litellm_params={"metadata": {"session_id": "sess-abc", "trace_id": metadata_trace_id}},
-                output=None,
-                start_time=datetime.datetime.utcnow(),
-                end_time=datetime.datetime.utcnow(),
-                kwargs=kwargs,
-                optional_params={},
-                input=None,
-                response_obj=None,
-                level="INFO",
-                litellm_call_id=litellm_call_id,
-            )
+        self.logger._log_langfuse_v2(
+            user_id="user-1",
+            metadata={"session_id": "sess-abc", "trace_id": metadata_trace_id},
+            litellm_params={"metadata": {"session_id": "sess-abc", "trace_id": metadata_trace_id}},
+            output=None,
+            start_time=datetime.datetime.utcnow(),
+            end_time=datetime.datetime.utcnow(),
+            kwargs=kwargs,
+            optional_params={},
+            input=None,
+            response_obj=None,
+            level="INFO",
+            litellm_call_id=litellm_call_id,
+        )
         return self.last_trace_kwargs
 
     def test_log_langfuse_v2_session_header_trace_id_falls_back_to_call_id(self):