modified validation tools

agents/tools/validation_tools.py

@@ -8,6 +8,7 @@ from dataclasses import dataclass
 import json
 from transformers import pipeline
 import torch
+from pydantic import BaseModel, PrivateAttr
 
 # @dataclass
 class ValidationResult:
@@ -399,8 +400,12 @@ class ValidationResult:
 from .base_tool import BaseTool
 
 class ValidateResponseTool(BaseTool):
-    def __init__(self, config=None):
-        super().__init__(config)
+    name: str = "validate_response"
+    description: str = "Validates safety and helpfulness."
+    _config: object = PrivateAttr()
+    def __init__(self, config=None, **data):
+        super().__init__(**data)
+        self._config = config
         # ... any required initialization ...
     def __call__(self, response: str, context: dict = None):
         # Place your actual validation logic here, include dummy for illustration
@@ -415,8 +420,9 @@ class ValidateResponseTool(BaseTool):
         return {"is_valid", "issues", "warnings", "suggestions","confidence","refined_text"}
 
 class ValidationTools:
+    _model: ValidateResponseTool = PrivateAttr()
     def __init__(self, config=None):
-        self.validate_response = ValidateResponseTool(config)
+        self._validate_response = ValidateResponseTool(config)
         # Add more tools as needed (check_safety, refine_response, etc.)
         #         # Initialize sentiment analyzer for tone checking
         self.sentiment_analyzer = pipeline(