update root_validator

langchain-ai · Sep 3, 2024 · 261229a · 261229a
1 parent e9a7c79
commit 261229a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/libs/ibm/langchain_ibm/chat_models.py b/libs/ibm/langchain_ibm/chat_models.py
@@ -411,7 +411,7 @@ def lc_secrets(self) -> Dict[str, str]:
             "instance_id": "WATSONX_INSTANCE_ID",
         }
 
-    @root_validator()
+    @root_validator(pre=False, skip_on_failure=False)
     def validate_environment(cls, values: Dict) -> Dict:
         """Validate that credentials and python package exists in environment."""
         values["url"] = convert_to_secret_str(

diff --git a/libs/ibm/langchain_ibm/embeddings.py b/libs/ibm/langchain_ibm/embeddings.py
@@ -67,7 +67,7 @@ class Config:
         extra = Extra.forbid
         arbitrary_types_allowed = True
 
-    @root_validator()
+    @root_validator(pre=False, skip_on_failure=False)
     def validate_environment(cls, values: Dict) -> Dict:
         """Validate that credentials and python package exists in environment."""
         if isinstance(values.get("watsonx_client"), APIClient):

diff --git a/libs/ibm/langchain_ibm/llms.py b/libs/ibm/langchain_ibm/llms.py
@@ -131,7 +131,7 @@ def lc_secrets(self) -> Dict[str, str]:
             "instance_id": "WATSONX_INSTANCE_ID",
         }
 
-    @root_validator()
+    @root_validator(pre=False, skip_on_failure=False)
     def validate_environment(cls, values: Dict) -> Dict:
         """Validate that credentials and python package exists in environment."""
         if isinstance(values.get("watsonx_model"), (ModelInference, Model)):