This commit is contained in:
Tanushree Sharma
2026-03-17 13:02:37 -07:00
parent 5a27af5224
commit 7edc4ccd86
2 changed files with 12 additions and 4 deletions

View File

@@ -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]:

View File

@@ -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}