revered the patch so that the fix can be applied in the main place

This commit is contained in:
Rajan Paneru 2024-05-10 22:04:44 +09:30
parent c45085b728
commit 8eb842dcf5

View file

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