Initial commit with full functionality extend app req tools

app.py

@@ -4,29 +4,33 @@ from tools.final_answer import FinalAnswerTool
 import yaml
 from Gradio_UI import GradioUI
 
-# Initialize components
-final_answer = FinalAnswerTool()
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct'
-)
+def create_agent():
+    # Initialize the final answer tool
+    final_answer = FinalAnswerTool()
+    
+    # Load prompt templates
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
+    
+    # Create the model
+    model = HfApiModel(
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+        max_tokens=2096,
+        temperature=0.5
+    )
+    
+    # Create and return the agent
+    return CodeAgent(
+        model=model,
+        tools=[final_answer],
+        max_steps=6,
+        verbosity_level=1,
+        prompt_templates=prompt_templates
+    )
 
-# Load prompt templates
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# Create agent
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer],
-    max_steps=6,
-    verbosity_level=1,
-    prompt_templates=prompt_templates
-)
-
-# Launch interface
 if __name__ == "__main__":
+    # Create and launch the agent
+    agent = create_agent()
     GradioUI(agent).launch(
         server_name="0.0.0.0",
         server_port=7860

tools/final_answer.py

@@ -1,23 +1,17 @@
 # tools/final_answer.py
-from dataclasses import dataclass, field
+from smolagents import Tool
+from typing import Optional
 
-@dataclass
-class FinalAnswerTool:
-    """A tool to format and deliver the final answer to the user."""
+class FinalAnswerTool(Tool):
+    """Tool for providing final answers to user queries."""
+    
+    name = "final_answer"
     
-    name: str = field(default="final_answer", init=False)
-    description: str = field(default="Tool to provide the final answer", init=False)
-
     def __call__(self, answer: str) -> str:
-        """Process and return the final response.
-
+        """Process and return the final answer.
         Args:
-            answer: The final answer text to return to the user.
-
+            answer: The answer text to be returned
         Returns:
-            The processed answer as a string.
+            str: The processed answer
         """
-        return answer
-
-    def __str__(self) -> str:
-        return self.name
+        return answer