improve: mordernizing validation by migrating pydantic from 1.x to 2.x (#4592)

api/core/workflow/nodes/tool/entities.py

@@ -1,6 +1,7 @@
 from typing import Any, Literal, Union
 
-from pydantic import BaseModel, validator
+from pydantic import BaseModel, field_validator
+from pydantic_core.core_schema import ValidationInfo
 
 from core.workflow.entities.base_node_data_entities import BaseNodeData
 
@@ -13,13 +14,14 @@ class ToolEntity(BaseModel):
     tool_label: str # redundancy
     tool_configurations: dict[str, Any]
 
-    @validator('tool_configurations', pre=True, always=True)
-    def validate_tool_configurations(cls, value, values):
+    @classmethod
+    @field_validator('tool_configurations', mode='before')
+    def validate_tool_configurations(cls, value, values: ValidationInfo) -> dict[str, Any]:
         if not isinstance(value, dict):
             raise ValueError('tool_configurations must be a dictionary')
         
-        for key in values.get('tool_configurations', {}).keys():
-            value = values.get('tool_configurations', {}).get(key)
+        for key in values.data.get('tool_configurations', {}).keys():
+            value = values.data.get('tool_configurations', {}).get(key)
             if not isinstance(value, str | int | float | bool):
                 raise ValueError(f'{key} must be a string')
             
@@ -30,10 +32,11 @@ class ToolNodeData(BaseNodeData, ToolEntity):
         value: Union[Any, list[str]]
         type: Literal['mixed', 'variable', 'constant']
 
-        @validator('type', pre=True, always=True)
-        def check_type(cls, value, values):
+        @classmethod
+        @field_validator('type', mode='before')
+        def check_type(cls, value, validation_info: ValidationInfo):
             typ = value
-            value = values.get('value')
+            value = validation_info.data.get('value')
             if typ == 'mixed' and not isinstance(value, str):
                 raise ValueError('value must be a string')
             elif typ == 'variable':
@@ -45,7 +48,7 @@ class ToolNodeData(BaseNodeData, ToolEntity):
             elif typ == 'constant' and not isinstance(value, str | int | float | bool):
                 raise ValueError('value must be a string, int, float, or bool')
             return typ
-        
+
     """
     Tool Node Schema
     """