Upload tool

tool.py

@@ -1,5 +1,6 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
+import logging
 
 class UserInputTool(Tool):
     name = "user_input"
@@ -7,9 +8,30 @@ class UserInputTool(Tool):
     inputs = {'question': {'type': 'string', 'description': 'The question to ask the user'}}
     output_type = "string"
 
-    def forward(self, question):
-        user_input = input(f"{question} => Type your answer here:")
-        return user_input
+    def __init__(self):
+        super().__init__()
+        import logging
+        self.logger = logging.getLogger(__name__)
+        self.logger.setLevel(logging.INFO)
+        self.user_input = ""  # Initialize user_input as an instance variable
 
-    def __init__(self, *args, **kwargs):
-        self.is_initialized = False
+    def _validate_question(self, question) -> tuple[bool, str]:
+        # Helper method to validate the question
+        if not isinstance(question, str):
+            return False, f"Question must be a string, got {type(question)}"
+        if not question.strip():
+            return False, "Question cannot be empty"
+        return True, question
+
+    async def forward(self, question: str) -> str:
+        # Validate the question first
+        success, response = self._validate_question(question)
+        if not success:
+            self.logger.error(response)
+            return f"Error: {response}"
+
+        # Ask the validated question
+        self.logger.info(f"Asking user: {question}")
+        self.user_input = input(f"{question} => Type your answer here:")  # Use instance variable
+        self.logger.info(f"Received user input: {self.user_input}")
+        return self.user_input