fix: azure model name

lunary-ai · Nov 5, 2024 · be8ad98 · be8ad98
1 parent 7dea1e3
commit be8ad98
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/lunary/__init__.py b/lunary/__init__.py
@@ -957,7 +957,8 @@ def on_llm_start(
                 )  # Sometimes, for example with ChatAnthropic, `invocation_params` is empty
 
                 name = (
-                    params.get("model")
+                    metadata.get("ls_model_name") # for Azure OpenAI
+                    or params.get("model")
                     or params.get("model_name")
                     or params.get("model_id")
                     or params.get("deployment_name")
@@ -1015,7 +1016,8 @@ def on_chat_model_start(
                 )  # Sometimes, for example with ChatAnthropic, `invocation_params` is empty
 
                 name = (
-                    params.get("model")
+                    metadata.get("ls_model_name") # for Azure OpenAI
+                    or params.get("model")
                     or params.get("model_name")
                     or params.get("model_id")
                     or params.get("deployment_name")

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lunary"
-version = "1.1.11"
+version = "1.1.12"
 description = "Observability, analytics and evaluations for AI agents and chatbots."
 authors = ["lunary <hello@lunary.ai>"]
 readme = "README.md"