From c45085b728ab2994db7cf0412a9aa076d1c11b77 Mon Sep 17 00:00:00 2001 From: Rajan Paneru Date: Fri, 10 May 2024 10:06:50 +0930 Subject: [PATCH] Based on the data-type using json The value of response_obj["choices"][0]["message"] is Message object and dict Added a conditional to use .json only iff it is Message Object --- litellm/integrations/langfuse.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index caf5437b242..140dced9d32 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -135,7 +135,9 @@ class LangFuseLogger: response_obj, litellm.ModelResponse ): input = prompt - output = response_obj["choices"][0]["message"].json() + output = response_obj["choices"][0]["message"] + if isinstance(output, litellm.Message): + output = output.json() elif response_obj is not None and isinstance( response_obj, litellm.TextCompletionResponse ):