Set Langfuse output to null for embedding responses

2024-06-16 15:14:34 +02:00 · 2024-06-16 15:14:34 +02:00 · d338a94a57
commit d338a94a57
parent 056913fd70
1 changed files with 1 additions and 1 deletions
--- a/litellm/integrations/langfuse.py
+++ b/litellm/integrations/langfuse.py
@ -167,7 +167,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
            ):