Update agent.py

agent.py

@@ -134,35 +134,35 @@ def gaia_file_reader(file_id: str) -> str:
 # Final agent class
 # --------------------------------------------------------------------------- #
 class GeminiAgent:
-    """
-    Exposed to `app.py` – instantiated once and then called per question.
-    """
-
     def __init__(self):
-        model = GeminiModel()
-        tools = [
-            PythonInterpreterTool(),  # maths, csv, small image ops
-            DuckDuckGoSearchTool(),   # quick web look-ups
-            gaia_file_reader,         # our custom file tool
-        ]
         self.system_prompt = (
             "You are a concise, highly accurate assistant. "
             "Unless explicitly required, reply with ONE short sentence. "
             "Use the provided tools if needed. "
             "All answers are graded by exact string match."
         )
+
+        model = GeminiModel()
+        tools = [
+            PythonInterpreterTool(),
+            DuckDuckGoSearchTool(),
+            gaia_file_reader,
+        ]
+
+        # ✨  system_prompt removed – newest smolagents doesn't take it
         self.agent = CodeAgent(
             model=model,
             tools=tools,
-            system_prompt=self.system_prompt,
+            # any other kwargs (executor_type, additional_authorized_imports…)
+            verbosity_level=0,
         )
-        print("✅ GeminiAgent (google-genai) initialised.")
+        print("✅ GeminiAgent ready.")
 
-    # ----------- Main entry point for app.py ----------- #
     def __call__(self, question: str) -> str:
         file_ids = FILE_TAG.findall(question)
+
+        # -------- multimodal branch -------- #
         if file_ids:
-            # Build multimodal user content
             parts: list[gtypes.Part] = []
             text_part = FILE_TAG.sub("", question).strip()
             if text_part:
@@ -171,15 +171,23 @@ class GeminiAgent:
                 try:
                     img_bytes = _download_file(fid)
                     mime = mimetypes.guess_type(fid)[0] or "image/png"
-                    parts.append(gtypes.Part.from_bytes(data=img_bytes, mime_type=mime))
+                    parts.append(
+                        gtypes.Part.from_bytes(data=img_bytes, mime_type=mime)
+                    )
                 except Exception as exc:
-                    parts.append(gtypes.Part.from_text(f"[FILE {fid} ERROR: {exc}]"))
+                    parts.append(
+                        gtypes.Part.from_text(f"[FILE {fid} ERROR: {exc}]")
+                    )
             messages = [
                 {"role": "system", "content": self.system_prompt},
                 {"role": "user", "content": parts},
             ]
             answer = self.agent.model.call_messages(messages)
+
+        # -------- text-only branch -------- #
         else:
-            answer = self.agent(question)
-        # Trim trailing punctuation – GAIA scoring is case-/punctuation-sensitive
+            # prepend system prompt to the user question
+            full_prompt = f"{self.system_prompt}\n\n{question}"
+            answer = self.agent(full_prompt)
+
         return answer.rstrip(" .\n\r\t")