From d338a94a5715985b271700cdf95bf5e2b91ebda4 Mon Sep 17 00:00:00 2001 From: Hannes Burrichter Date: Sun, 16 Jun 2024 15:14:34 +0200 Subject: [PATCH 1/2] 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 3c8f8c85cc..d2cfae41ad 100644 --- 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 ): From 27df6621641e96587616388dd75a2e2d342b1347 Mon Sep 17 00:00:00 2001 From: Hannes Burrichter Date: Sun, 16 Jun 2024 16:09:35 +0200 Subject: [PATCH 2/2] Fix missing Azure api version in Langfuse `create_async_task` test --- litellm/tests/test_alangfuse.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/tests/test_alangfuse.py b/litellm/tests/test_alangfuse.py index 2acacbb961..2303dc9e8c 100644 --- a/litellm/tests/test_alangfuse.py +++ b/litellm/tests/test_alangfuse.py @@ -177,6 +177,7 @@ def create_async_task(**completion_kwargs): """ completion_args = { "model": "azure/chatgpt-v-2", + "api_version": "2024-02-01", "messages": [{"role": "user", "content": "This is a test"}], "max_tokens": 5, "temperature": 0.7,