diff --git a/litellm/integrations/langfuse.py b/litellm/integrations/langfuse.py index d76ba18360..a8dfd955eb 100644 --- a/litellm/integrations/langfuse.py +++ b/litellm/integrations/langfuse.py @@ -64,7 +64,7 @@ class LangFuseLogger: pass # end of processing langfuse ######################## - if kwargs.get("call_type") == "embedding" or isinstance( + if kwargs.get("call_type", None) == "embedding" or isinstance( response_obj, litellm.EmbeddingResponse ): input = prompt