From 71e14cbbda0582d60790fa14a4a42f4db6069fe0 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 31 Jan 2024 10:28:18 -0800 Subject: [PATCH] (fix) langfuse check for embedding --- 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 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