From 4cdaca67dc51dba887289f56c6fead3c1a52f97d Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 15 Aug 2024 14:54:08 -0400 Subject: [PATCH] ai21[patch]: Upgrade @root_validators for pydantic 2 migration (#25454) Upgrade @root_validators usage to match pydantic 2 semantics --- libs/partners/ai21/langchain_ai21/chat_models.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/libs/partners/ai21/langchain_ai21/chat_models.py b/libs/partners/ai21/langchain_ai21/chat_models.py index 4e31f66ff0f..374f262e101 100644 --- a/libs/partners/ai21/langchain_ai21/chat_models.py +++ b/libs/partners/ai21/langchain_ai21/chat_models.py @@ -83,13 +83,11 @@ class ChatAI21(BaseChatModel, AI21Base): _chat_adapter: ChatAdapter - @root_validator() + @root_validator(pre=False, skip_on_failure=True) def validate_environment(cls, values: Dict) -> Dict: - values = super().validate_environment(values) - model = values.get("model") - - values["_chat_adapter"] = create_chat_adapter(model) # type: ignore - + """Validate the environment.""" + model = values["model"] + values["_chat_adapter"] = create_chat_adapter(model) return values class Config: