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