Revert set Langfuse output to null for embedding responses

This commit is contained in:
Hannes Burrichter 2024-05-21 18:25:24 +02:00
parent 82391d270c
commit 8ed41dee09

View file

@ -129,7 +129,7 @@ class LangFuseLogger:
or isinstance(response_obj, litellm.EmbeddingResponse) or isinstance(response_obj, litellm.EmbeddingResponse)
): ):
input = prompt input = prompt
output = None output = response_obj["data"]
elif response_obj is not None and isinstance( elif response_obj is not None and isinstance(
response_obj, litellm.ModelResponse response_obj, litellm.ModelResponse
): ):