fix

src/agent/base_agent.py

@@ -44,8 +44,7 @@ class BaseAgent(ABC):
             max_steps=self.max_steps
         )
         agent.system_prompt += ("\nWhen answering, provide ONLY the precise answer requested. "
-                                "Do not include explanations, steps, reasoning, or additional text. "
-                                "Be direct and specific. GAIA benchmark requires exact matching answers.")
-        # print(agent.system_prompt)
-        print("Agent initialized.")
+                                "Do not include explanations, steps, reasoning, or additional text. ")
+        print(agent.system_prompt)
+        # print("Agent initialized.")
         return agent

src/try_cuda.py

@@ -0,0 +1,2 @@
+import torch
+print(torch.cuda.is_available())