Update agent.py

agent.py

@@ -17,11 +17,14 @@ from smolagents import (
 from tools import GetTaskFileTool, LoadXlsxFileTool, LoadTextFileTool
 
 sys_instruction = (
-    "You are a general AI assistant. Answer each question by reporting your thoughts, "
-    "then submit ONLY a concise text using the 'final_answer' tool. "
-    "Final answer MUST be a number, a few words, or a comma-separated list of numbers and/or strings. "
-    "For numbers, avoid commas and units unless specified. For strings, avoid articles and abbreviations, "
-    "and write digits in full unless stated otherwise. Apply these rules for list elements as well."
+    "You are a general-purpose AI assistant. For each question, first reason through your answer, "
+    "then use the 'final_answer' tool to submit ONLY the final output.\n"
+    "Your final answer MUST be one of the following:\n"
+    "- A number (without commas or units, unless explicitly instructed),\n"
+    "- A short phrase,\n"
+    "- A comma-separated list of numbers or strings.\n"
+    "For strings, avoid articles and abbreviations, and write digits fully unless told otherwise. "
+    "Apply these formatting rules consistently to each item in a list."
 )
 
 prompts = yaml.safe_load(
@@ -31,28 +34,34 @@ prompts["system_prompt"] = sys_instruction + prompts["system_prompt"]
 
 # req_instruction = (
 req_instruction = (
-    "You are an expert and helpful agent named {{name}}.\n"
+    "You are a highly capable and autonomous agent named {{name}}, designed to solve complex tasks efficiently.\n"
     "A valued client has assigned you the following task:\n"
     "---\n"
     "Task:\n"
     "{{task}}\n"
     "---\n"
-    "Use your tools as needed. Before completing the task, plan your actions carefully.\n"
-    "While completing the task, think step by step. And after completing the task, carefully double check your solution.\n\n"
-    "If you respond correctly, you will be rewarded with a very high bonus.\n\n"
+    "To complete this task successfully, follow these steps carefully:\n"
+    "    1. Comprehend the task and identify the intended goal.\n"
+    "    2. Break the task into clear, logical steps.\n"
+    "    3. Select and prepare the tools or resources you need.\n"
+    "    4. Set up the required environment or context.\n"
+    "    5. Execute each step methodically.\n"
+    "    6. Monitor outcomes and identify any deviations.\n"
+    "    7. Revise your plan if necessary based on feedback.\n"
+    "    8. Maintain internal state and track progress.\n"
+    "    9. Verify that the goal has been fully achieved.\n"
+    "   10. Present the final result clearly and concisely.\n"
+    "If you succeed, you will be rewarded with a significant bonus.\n\n"
     "Your final_answer MUST be:\n"
-    "- a number,\n"
-    "- a short phrase,\n"
-    "- or a comma-separated list of numbers or strings (no articles or abbreviations).\n\n"
-    "Only the content passed to the final_answer tool will be preserved—everything else will be discarded."
+    "- a number (retain its original type; do not include units),\n"
+    "- a concise phrase,\n"
+    "- or a comma-separated list of numbers or strings (no articles, no abbreviations).\n\n"
+    "Only the content passed to the final_answer tool will be preserved. Any other content will be discarded."
 )
+
 prompts['managed_agent']['task'] = req_instruction
 prompts['managed_agent']['report'] = "{{final_answer}}"
 
-# print(prompts["system_prompt"])
-# print(prompts['planning'])
-# print(prompts['managed_agent'])
-# print(prompts['final_answer'])
 
 def get_model(
         model_id: str = "gpt-4.1-mini",
@@ -109,4 +118,9 @@ def get_agent(
         additional_authorized_imports = ["pandas"],
         name="GAIAAgent",
     )
-    return agent
+    return agent
+
+def test():
+    """test tests something. """
+    # Test the agent with a simple task
+    return True