mirror of
https://github.com/hwchase17/langchain.git
synced 2026-04-12 23:42:51 +00:00
factor
This commit is contained in:
@@ -967,10 +967,15 @@ class ChatAnthropic(BaseChatModel):
|
||||
all_required_field_names = get_pydantic_field_names(cls)
|
||||
return _build_model_kwargs(values, all_required_field_names)
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-anthropic", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_model_profile(self) -> Self:
|
||||
"""Set model profile if not overridden."""
|
||||
self._add_version("langchain-anthropic", __version__)
|
||||
if self.profile is None:
|
||||
self.profile = _get_default_model_profile(self.model)
|
||||
if (
|
||||
|
||||
@@ -225,10 +225,15 @@ class ChatDeepSeek(BaseChatOpenAI):
|
||||
ls_params["ls_provider"] = "deepseek"
|
||||
return ls_params
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-deepseek", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
"""Validate necessary environment vars and client params."""
|
||||
self._add_version("langchain-deepseek", __version__)
|
||||
if self.api_base == DEFAULT_API_BASE and not (
|
||||
self.api_key and self.api_key.get_secret_value()
|
||||
):
|
||||
|
||||
@@ -396,10 +396,15 @@ class ChatFireworks(BaseChatModel):
|
||||
all_required_field_names = get_pydantic_field_names(cls)
|
||||
return _build_model_kwargs(values, all_required_field_names)
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-fireworks", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
"""Validate that api key and python package exists in environment."""
|
||||
self._add_version("langchain-fireworks", __version__)
|
||||
if self.n < 1:
|
||||
msg = "n must be at least 1."
|
||||
raise ValueError(msg)
|
||||
|
||||
@@ -543,10 +543,15 @@ class ChatGroq(BaseChatModel):
|
||||
raise ImportError(msg) from exc
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-groq", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_model_profile(self) -> Self:
|
||||
"""Set model profile if not overridden."""
|
||||
self._add_version("langchain-groq", __version__)
|
||||
if self.profile is None:
|
||||
self.profile = _get_default_model_profile(self.model_name)
|
||||
return self
|
||||
|
||||
@@ -598,10 +598,15 @@ class ChatMistralAI(BaseChatModel):
|
||||
overall_token_usage[k] = v
|
||||
return {"token_usage": overall_token_usage, "model_name": self.model}
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-mistralai", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
"""Validate api key, python package exists, temperature, and top_p."""
|
||||
self._add_version("langchain-mistralai", __version__)
|
||||
if isinstance(self.mistral_api_key, SecretStr):
|
||||
api_key_str: str | None = self.mistral_api_key.get_secret_value()
|
||||
else:
|
||||
|
||||
@@ -788,10 +788,15 @@ class ChatOllama(BaseChatModel):
|
||||
|
||||
return params
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-ollama", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_clients(self) -> Self:
|
||||
"""Set clients to use for ollama."""
|
||||
self._add_version("langchain-ollama", __version__)
|
||||
client_kwargs = self.client_kwargs or {}
|
||||
|
||||
cleaned_url, auth_headers = parse_url_with_auth(self.base_url)
|
||||
|
||||
@@ -320,10 +320,15 @@ class OllamaLLM(BaseLLM):
|
||||
params["ls_max_tokens"] = max_tokens
|
||||
return params
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-ollama", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_clients(self) -> Self:
|
||||
"""Set clients to use for ollama."""
|
||||
self._add_version("langchain-ollama", __version__)
|
||||
client_kwargs = self.client_kwargs or {}
|
||||
|
||||
cleaned_url, auth_headers = parse_url_with_auth(self.base_url)
|
||||
|
||||
@@ -970,10 +970,11 @@ class BaseChatOpenAI(BaseChatModel):
|
||||
|
||||
return values
|
||||
|
||||
def model_post_init(self, _context: Any, /) -> None:
|
||||
"""Add `langchain-openai` version to metadata."""
|
||||
super().model_post_init(_context)
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-openai", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
|
||||
@@ -118,10 +118,15 @@ class AzureOpenAI(BaseOpenAI):
|
||||
"""Return whether this model can be serialized by LangChain."""
|
||||
return True
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-openai", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
"""Validate that api key and python package exists in environment."""
|
||||
self._add_version("langchain-openai", __version__)
|
||||
if self.n < 1:
|
||||
msg = "n must be at least 1."
|
||||
raise ValueError(msg)
|
||||
|
||||
@@ -302,10 +302,15 @@ class ChatOpenRouter(BaseChatModel):
|
||||
values["model_kwargs"] = extra
|
||||
return values
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-openrouter", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def validate_environment(self) -> Self:
|
||||
"""Validate configuration and build the SDK client."""
|
||||
self._add_version("langchain-openrouter", __version__)
|
||||
if not (self.openrouter_api_key and self.openrouter_api_key.get_secret_value()):
|
||||
msg = "OPENROUTER_API_KEY must be set."
|
||||
raise ValueError(msg)
|
||||
|
||||
@@ -469,10 +469,15 @@ class ChatXAI(BaseChatOpenAI): # type: ignore[override]
|
||||
params["ls_provider"] = "xai"
|
||||
return params
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _set_version(self) -> Self:
|
||||
"""Set package version in metadata."""
|
||||
self._add_version("langchain-xai", __version__)
|
||||
return self
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _warn_search_parameters_deprecated(self) -> Self:
|
||||
"""Emit deprecation warning if search_parameters (Live Search) is used."""
|
||||
self._add_version("langchain-xai", __version__)
|
||||
if self.search_parameters:
|
||||
warnings.warn(
|
||||
"search_parameters (Live Search) is deprecated by xAI and is ignored. "
|
||||
|
||||
Reference in New Issue
Block a user