fix(langchain/AzureChatOpenai): unnecessary logs (#919)

langfuse · Sep 10, 2024 · e02fb8c · e02fb8c
1 parent 9a15551
commit e02fb8c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/langfuse/Sampler.py b/langfuse/Sampler.py
@@ -18,6 +18,9 @@ def sample_event(self, event: dict):
         if "type" in event and "body" in event:
             event_type = event["type"]
 
+            if event_type == "sdk-log":
+                return True
+
             trace_id = None
 
             if event_type == "trace-create" and "id" in event["body"]:

diff --git a/langfuse/extract_model.py b/langfuse/extract_model.py
@@ -27,6 +27,7 @@ def _extract_model_name(
         ("ChatOpenAI", ["invocation_params", "model_name"], "kwargs"),
         ("AzureChatOpenAI", ["invocation_params", "model"], "kwargs"),
         ("AzureChatOpenAI", ["invocation_params", "model_name"], "kwargs"),
+        ("AzureChatOpenAI", ["invocation_params", "azure_deployment"], "kwargs"),
         ("HuggingFacePipeline", ["invocation_params", "model_id"], "kwargs"),
         ("BedrockChat", ["kwargs", "model_id"], "serialized"),
         ("Bedrock", ["kwargs", "model_id"], "serialized"),