diff --git a/libs/core/langchain_core/language_models/base.py b/libs/core/langchain_core/language_models/base.py index d998e389cb7..570076290e0 100644 --- a/libs/core/langchain_core/language_models/base.py +++ b/libs/core/langchain_core/language_models/base.py @@ -314,10 +314,8 @@ class BaseLanguageModel( stop: list[str] | None = None, **kwargs: Any, ) -> LangSmithParams: - """Wrap _get_ls_params to always include ls_integration.""" - ls_params = self._get_ls_params(stop=stop, **kwargs) - ls_params["ls_integration"] = "langchain_chat_model" - return ls_params + """Wrap _get_ls_params to include any additional default parameters.""" + return self._get_ls_params(stop=stop, **kwargs) @property def _identifying_params(self) -> Mapping[str, Any]: diff --git a/libs/core/langchain_core/language_models/chat_models.py b/libs/core/langchain_core/language_models/chat_models.py index 92b79e20759..25dc2dd16f3 100644 --- a/libs/core/langchain_core/language_models/chat_models.py +++ b/libs/core/langchain_core/language_models/chat_models.py @@ -827,6 +827,16 @@ class BaseChatModel(BaseLanguageModel[AIMessage], ABC): return ls_params + def _get_ls_params_with_defaults( + self, + stop: list[str] | None = None, + **kwargs: Any, + ) -> LangSmithParams: + """Wrap _get_ls_params to always include ls_integration.""" + ls_params = self._get_ls_params(stop=stop, **kwargs) + ls_params["ls_integration"] = "langchain_chat_model" + return ls_params + def _get_llm_string(self, stop: list[str] | None = None, **kwargs: Any) -> str: if self.is_lc_serializable(): params = {**kwargs, "stop": stop}