Update tools/FinalAnswerTool

tools/FinalAnswerTool

@@ -1,42 +1,20 @@
 from smolagents import LiteLLMModel
-from settings import settings
-from utils import InputTokenRateLimiter
+from src.settings import Settings
+from src.utils import InputTokenRateLimiter
+from smolagents.tools import Tool
+from litellm import completion
+import os
 
-class FinalAnswerTool:
-    def __init__(self):
-        self.model = LiteLLMModel(
-            model_id=settings.llm_model_id,
-            api_key=settings.llm_api_key,
-            temperature=0.1,
-            max_tokens=20,
-        )
-        self.token_rate_limiter = InputTokenRateLimiter()
-        self.expected_tokens_per_step = 10000
+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):
-        self.token_rate_limiter.maybe_wait(self.expected_tokens_per_step)
-        response = self.model.generate(
-            [
-                {  "role": "system",
-                    "content": [
-                        {
-                            "type": "text",
-                            "text": (
-                                "You are an assistant that answers questions with exactly one word. "
-                                "Always choose the most direct, unambiguous word. "
-                                "Do NOT use sentences. If the answer is unknown or invalid, respond 'N/A'."
-                            ),
-                        }
-                    ],
-                },
-                {
-                    "role": "user",
-                    "content": "Generate a final answer to the question given the context provided. Only use the context to provide the final answer. Do not provide any additional thinking or commentary in the response, the final answer only.",
-                }
-            ]
-        )
-        tokens_used = getattr(response, "token_usage", None)
-        if tokens_used:
-            self.token_rate_limiter.add_tokens(tokens_used.input_tokens)
-        return response.content
+    def forward(self, answer: str) -> str:
+        final_answer = f"FINAL ANSWER: {answer}"
+        return final_answer