core, partners: implement standard tracing params for LLMs (#25410)

libs/partners/openai/langchain_openai/llms/azure.py

@@ -5,6 +5,7 @@ import os
 from typing import Any, Callable, Dict, List, Mapping, Optional, Union
 
 import openai
+from langchain_core.language_models import LangSmithParams
 from langchain_core.pydantic_v1 import Field, SecretStr, root_validator
 from langchain_core.utils import convert_to_secret_str, get_from_dict_or_env
 
@@ -195,6 +196,17 @@ class AzureOpenAI(BaseOpenAI):
         openai_params = {"model": self.deployment_name}
         return {**openai_params, **super()._invocation_params}
 
+    def _get_ls_params(
+        self, stop: Optional[List[str]] = None, **kwargs: Any
+    ) -> LangSmithParams:
+        """Get standard params for tracing."""
+        params = super()._get_ls_params(stop=stop, **kwargs)
+        invocation_params = self._invocation_params
+        params["ls_provider"] = "azure"
+        if model_name := invocation_params.get("model"):
+            params["ls_model_name"] = model_name
+        return params
+
     @property
     def _llm_type(self) -> str:
         """Return type of llm."""

libs/partners/openai/tests/unit_tests/llms/test_azure.py

@@ -0,0 +1,23 @@
+from typing import Any
+
+from langchain_openai import AzureOpenAI
+
+
+def test_azure_model_param(monkeypatch: Any) -> None:
+    monkeypatch.delenv("OPENAI_API_BASE", raising=False)
+    llm = AzureOpenAI(
+        openai_api_key="secret-api-key",  # type: ignore[call-arg]
+        azure_endpoint="endpoint",
+        api_version="version",
+        azure_deployment="gpt-35-turbo-instruct",
+    )
+
+    # Test standard tracing params
+    ls_params = llm._get_ls_params()
+    assert ls_params == {
+        "ls_provider": "azure",
+        "ls_model_type": "llm",
+        "ls_model_name": "gpt-35-turbo-instruct",
+        "ls_temperature": 0.7,
+        "ls_max_tokens": 256,
+    }

libs/partners/openai/tests/unit_tests/llms/test_base.py

@@ -14,6 +14,16 @@ def test_openai_model_param() -> None:
     llm = OpenAI(model_name="foo")  # type: ignore[call-arg]
     assert llm.model_name == "foo"
 
+    # Test standard tracing params
+    ls_params = llm._get_ls_params()
+    assert ls_params == {
+        "ls_provider": "openai",
+        "ls_model_type": "llm",
+        "ls_model_name": "foo",
+        "ls_temperature": 0.7,
+        "ls_max_tokens": 256,
+    }
+
 
 def test_openai_model_kwargs() -> None:
     llm = OpenAI(model_kwargs={"foo": "bar"})