From f73a5a61d1460b5025c0a4c1b230d25604d1292b Mon Sep 17 00:00:00 2001 From: Max Deichmann Date: Tue, 21 Nov 2023 17:06:44 +0100 Subject: [PATCH] support new azure langchain integration --- langfuse/callback.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/langfuse/callback.py b/langfuse/callback.py index 6423d43ae..f21e2ee91 100644 --- a/langfuse/callback.py +++ b/langfuse/callback.py @@ -485,7 +485,9 @@ def __on_llm_action( elif kwargs["invocation_params"]["_type"] == "huggingface_hub": model_name = kwargs["invocation_params"]["repo_id"] elif kwargs["invocation_params"]["_type"] == "azure-openai-chat": - if serialized["kwargs"].get("deployment_name") and serialized["kwargs"].get("model_version"): + if kwargs.get("invocation_params").get("model") and serialized["kwargs"].get("model_version"): + model_name = kwargs.get("invocation_params").get("model") + "-" + serialized["kwargs"]["model_version"] + elif serialized["kwargs"].get("deployment_name") and serialized["kwargs"].get("model_version"): model_name = serialized["kwargs"]["deployment_name"] + "-" + serialized["kwargs"]["model_version"] else: model_name = kwargs["invocation_params"]["engine"]