diff --git a/libs/community/langchain_community/utilities/apify.py b/libs/community/langchain_community/utilities/apify.py index efcae49e2d1..f80deb496d4 100644 --- a/libs/community/langchain_community/utilities/apify.py +++ b/libs/community/langchain_community/utilities/apify.py @@ -19,7 +19,7 @@ class ApifyWrapper(BaseModel): apify_client_async: Any apify_api_token: Optional[str] = None - @root_validator() + @root_validator(pre=True) def validate_environment(cls, values: Dict) -> Dict: """Validate environment. Validate that an Apify API token is set and the apify-client diff --git a/libs/community/langchain_community/utilities/arcee.py b/libs/community/langchain_community/utilities/arcee.py index 72170348583..3064fd16b4d 100644 --- a/libs/community/langchain_community/utilities/arcee.py +++ b/libs/community/langchain_community/utilities/arcee.py @@ -51,7 +51,7 @@ class DALMFilter(BaseModel): value: str _is_metadata: bool = False - @root_validator() + @root_validator(pre=True) def set_meta(cls, values: Dict) -> Dict: """document and name are reserved arcee keys. Anything else is metadata""" values["_is_meta"] = values.get("field_name") not in ["document", "name"] diff --git a/libs/community/langchain_community/utilities/arxiv.py b/libs/community/langchain_community/utilities/arxiv.py index bab2df8061b..394a97c0fdd 100644 --- a/libs/community/langchain_community/utilities/arxiv.py +++ b/libs/community/langchain_community/utilities/arxiv.py @@ -72,7 +72,7 @@ class ArxivAPIWrapper(BaseModel): return False return True - @root_validator() + @root_validator(pre=True) def validate_environment(cls, values: Dict) -> Dict: """Validate that the python package exists in environment.""" try: diff --git a/libs/community/langchain_community/utilities/asknews.py b/libs/community/langchain_community/utilities/asknews.py index 4ac5445568c..5e5fb246fc1 100644 --- a/libs/community/langchain_community/utilities/asknews.py +++ b/libs/community/langchain_community/utilities/asknews.py @@ -24,7 +24,7 @@ class AskNewsAPIWrapper(BaseModel): extra = Extra.forbid - @root_validator() + @root_validator(pre=True) def validate_environment(cls, values: Dict) -> Dict: """Validate that api credentials and python package exists in environment.""" diff --git a/libs/community/langchain_community/utilities/awslambda.py b/libs/community/langchain_community/utilities/awslambda.py index 8dbaca563ae..916c7627f3c 100644 --- a/libs/community/langchain_community/utilities/awslambda.py +++ b/libs/community/langchain_community/utilities/awslambda.py @@ -34,7 +34,7 @@ class LambdaWrapper(BaseModel): extra = Extra.forbid - @root_validator() + @root_validator(pre=True) def validate_environment(cls, values: Dict) -> Dict: """Validate that python package exists in environment.""" @@ -47,8 +47,6 @@ class LambdaWrapper(BaseModel): ) values["lambda_client"] = boto3.client("lambda") - values["function_name"] = values["function_name"] - return values def run(self, query: str) -> str: