scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Based on the data-type using json

Browse source 
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
This commit is contained in:
Rajan Paneru 2024-05-10 10:06:50 +09:30
parent 66a1b581e5
commit c45085b728
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/integrations/langfuse.py
View file

@ -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
):