From 8eb842dcf584c9cf6ff9d7675899fb84f39e35e7 Mon Sep 17 00:00:00 2001 From: Rajan Paneru Date: Fri, 10 May 2024 22:04:44 +0930 Subject: [PATCH] revered the patch so that the fix can be applied in the main place --- litellm/integrations/langfuse.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 140dced9d..caf5437b2 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -135,9 +135,7 @@ class LangFuseLogger: response_obj, litellm.ModelResponse ): input = prompt - output = response_obj["choices"][0]["message"] - if isinstance(output, litellm.Message): - output = output.json() + output = response_obj["choices"][0]["message"].json() elif response_obj is not None and isinstance( response_obj, litellm.TextCompletionResponse ):