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."""