Merge pull request #4226 from Intellegam/main

Langfuse Integration ignore Embedding Output

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
             ):

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,