Update tools.py

tools.py

@@ -1,6 +1,10 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 
+simplify_system_message = """
+You are a general AI assistant. I will give you a question and answer, you should simplify answer. Your answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+"""     
+
 class FinalAnswerTool(Tool):
     name = "final_answer"
     description = "Provides a final answer to the given problem."
@@ -11,4 +15,43 @@ class FinalAnswerTool(Tool):
         return answer
 
     def __init__(self, *args, **kwargs):
-        self.is_initialized = False
+        self.is_initialized = False
+
+
+class SimplifyAnswerTool(Tool):
+    name = "simplify_answer"
+    description = (
+        "Simplify the final answer."
+        "Be sure to use this as the last step to prepare the final answer."
+    )
+    inputs = {
+        "question": {
+            "type": "string",
+            "description": "The question from the user.",
+        },
+        "answer": {
+            "type": "string",
+            "description": "The existing answer to be simplified.",
+        }        
+    }
+    output_type = "string"
+
+    def __init__(self, api_model: ApiModel):
+        super().__init__()
+        self.api_model = api_model
+
+    def forward(self, question: str, answer: str) -> str:
+        try:
+            response_message = self.api_model(messages=[
+                {"role": MessageRole.SYSTEM, "content": [{"type": "text", "text": simplify_system_message }]},
+                {"role": MessageRole.ASSISTANT, "content": [{"type": "text", "text": f"QUESTION: {question}"}]},
+                {"role": MessageRole.ASSISTANT, "content": [{"type": "text", "text": f"ANSWER: {answer}"}]}
+            ])
+            return response_message.content
+
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"