github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-18 21:09:00 +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
@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: