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

core[minor],community[minor]: Upgrade all @root_validator() to @pre_init (#23841)

Browse Source 
This PR introduces a @pre_init decorator that's a @root_validator(pre=True) but with all the defaults populated!
This commit is contained in:
Eugene Yurtsev
2024-07-08 16:09:29 -04:00
committed by GitHub
parent f152d6ed3d
commit 2c180d645e
114 changed files with 439 additions and 276 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
libs/community/langchain_community/llms/vertexai.py
View File

@@ -10,7 +10,8 @@ from langchain_core.callbacks.manager import (
)
from langchain_core.language_models.llms import BaseLLM
from langchain_core.outputs import Generation, GenerationChunk, LLMResult
from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
from langchain_core.pydantic_v1 import BaseModel, Field
from langchain_core.utils import pre_init
from langchain_community.utilities.vertexai import (
create_retry_decorator,
@@ -222,7 +223,7 @@ class VertexAI(_VertexAICommon, BaseLLM):
"""Get the namespace of the langchain object."""
return ["langchain", "llms", "vertexai"]
@root_validator()
@pre_init
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that the python package exists in environment."""
tuned_model_name = values.get("tuned_model_name")
@@ -409,7 +410,7 @@ class VertexAIModelGarden(_VertexAIBase, BaseLLM):
"Set result_arg to None if output of the model is expected to be a string."
"Otherwise, if it's a dict, provided an argument that contains the result."
@root_validator()
@pre_init
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that the python package exists in environment."""
try: