github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-02 03:26:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Remove usage of @root_validator(allow_reuse=True) (#25235)

Browse Source 
Remove usage of @root_validator(allow_reuse=True)
This commit is contained in:
Eugene Yurtsev
2024-08-09 10:57:42 -04:00
committed by GitHub
parent a2b4c33bd6
commit 6e57aa7c36
12 changed files with 76 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/langchain_community/chat_models/gpt_router.py
View File

@@ -167,7 +167,7 @@ class GPTRouter(BaseChatModel):
"""Number of chat completions to generate for each prompt."""
max_tokens: int = 256
@root_validator(allow_reuse=True)
@root_validator(pre=True)
def validate_environment(cls, values: Dict) -> Dict:
values["gpt_router_api_base"] = get_from_dict_or_env(
values,
@@ -183,7 +183,10 @@ class GPTRouter(BaseChatModel):
"GPT_ROUTER_API_KEY",
)
)
return values
@root_validator(pre=True, skip_on_failure=True)
def post_init(cls, values: Dict) -> Dict:
try:
from gpt_router.client import GPTRouterClient