github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-19 13:23:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

ai21[patch]: Upgrade @root_validators for pydantic 2 migration (#25454)

Browse Source 
Upgrade @root_validators usage to match pydantic 2 semantics
This commit is contained in:
Eugene Yurtsev 2024-08-15 14:54:08 -04:00 committed by GitHub
parent d72a08a60d
commit 4cdaca67dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/partners/ai21/langchain_ai21/chat_models.py
View File

@ -83,13 +83,11 @@ class ChatAI21(BaseChatModel, AI21Base):
_chat_adapter: ChatAdapter _chat_adapter: ChatAdapter
@root_validator() @root_validator(pre=False, skip_on_failure=True)
def validate_environment(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
values = super().validate_environment(values) """Validate the environment."""
model = values.get("model") model = values["model"]
values["_chat_adapter"] = create_chat_adapter(model)
values["_chat_adapter"] = create_chat_adapter(model) # type: ignore
return values return values
class Config: class Config: