diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 140dced9d3..caf5437b24 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 ):