Set Langfuse output to null for embedding responses

This commit is contained in:
Hannes Burrichter 2024-05-11 18:25:53 +02:00
parent be85145b86
commit 84141917cf

View file

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