From 8ed41dee097a3bdf54fb4f9a219f35691b15af5f Mon Sep 17 00:00:00 2001 From: Hannes Burrichter Date: Tue, 21 May 2024 18:25:24 +0200 Subject: [PATCH] Revert set Langfuse output to null for embedding responses --- litellm/integrations/langfuse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index 7a78befaba..5e402134d2 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -129,7 +129,7 @@ class LangFuseLogger: or isinstance(response_obj, litellm.EmbeddingResponse) ): input = prompt - output = None + output = response_obj["data"] elif response_obj is not None and isinstance( response_obj, litellm.ModelResponse ):