Create FinalAnswerTool.py

tools/FinalAnswerTool.py

@@ -0,0 +1,20 @@
+from smolagents import LiteLLMModel
+from src.settings import Settings
+from src.utils import InputTokenRateLimiter
+from smolagents.tools import Tool
+from litellm import completion
+import os
+
+settings = Settings()
+print(settings.llm_model_id)
+class FinalAnswerTool(Tool):
+    name = "final_answer"
+    description = "Provides the exact, final answer to the given question."
+    inputs = {
+        "answer": {"type": "string", "description": "The final, correctly formatted answer string."},
+    }
+    output_type = "string"
+
+    def forward(self, answer: str) -> str:
+        final_answer = f"FINAL ANSWER: {answer}"
+        return final_answer