Set Langfuse output to null for embedding responses

This commit is contained in:
Hannes Burrichter 2024-06-16 15:14:34 +02:00
parent 056913fd70
commit d338a94a57

View file

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