Refactor imports and enhance system prompts in agents and tools modules

agent/agents.py

@@ -1,13 +1,12 @@
 import time
 
 import litellm
-from google import genai
 from google.genai import types
-from google.genai.types import GenerateContentConfig, ThinkingConfig
+from google.genai.types import ThinkingConfig
 from pydantic import BaseModel
-from smolagents import CodeAgent, Tool, VisitWebpageTool, WebSearchTool
+from smolagents import CodeAgent, VisitWebpageTool, WebSearchTool
 
-from agent.prompts import system_prompt
+from agent.prompts import formatter_system_prompt, system_prompt
 from agent.tools import (
     CodeExecutionTool,
     DownloadFile,
@@ -101,13 +100,15 @@ class BasicAgent:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # Run the agent to find the best catering service
 
-        answer = self.code_agent.run(f"Task id: {task_id}, Question: {question}")
+        answer = self.code_agent.run(
+            f"{system_prompt}\nTask id: {task_id}, Question: {question}"
+        )
         print(f"Agent found answer: {answer}")
         response = client.models.generate_content(
             model=model,
             contents=f"Here is the provided question: {question}, infomation or answer: {str(answer)} \n Now, resposne with the answer that followed the rules.",
             config=types.GenerateContentConfig(
-                system_instruction=system_prompt,
+                system_instruction=formatter_system_prompt,
                 response_mime_type="application/json",
                 temperature=0.0,
                 response_schema=final_answer,

agent/prompts.py

@@ -1,4 +1,6 @@
-system_prompt = """
+system_prompt = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL 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."""
+
+formatter_system_prompt = """
 I will provide you with the question and the information and answer, often derived from agent outputs. Your task is to distill this information into a single, precisely formatted final answer. The final answer should adhere strictly to the following formatting rules. Provide ONLY the requested value(s) and nothing else:
 Output Type: The final answer must be one of the following:
 A single number (in Western/Arabic digits).

agent/score.py

@@ -1,10 +1,8 @@
 # modified from https://huggingface.co/spaces/gaia-benchmark/leaderboard/blob/main/scorer.py
-import json
 import re
 import string
 import warnings
 
-import numpy as np
 
 
 def normalize_number_str(number_str: str) -> float:

agent/tools.py

@@ -1,13 +1,12 @@
 import io
 
 import requests
+import wikipedia
 from google import genai
 from google.genai import types
-from langchain_community.retrievers import WikipediaRetriever
+from markdownify import markdownify as md
 from PIL import Image
 from smolagents import Tool
-import wikipedia
-from markdownify import markdownify as md
 
 
 class WikipediaSearchTool(Tool):

agent/utils.py

@@ -1,4 +1,3 @@
-import os
 
 from google import genai
 from smolagents import LiteLLMModel