Rename agent14.py to agent.py

agent14.py → agent.py

@@ -217,7 +217,7 @@ class MagAgent:
             - Use the name of the file ONLY FROM the "FILE:" section. THIS IS ALWAYS A FILE.
             IMPORTANT: When giving the final answer, output only the direct required result without any extra text like "Final Answer:" or explanations. YOU MUST RESPOND IN THE EXACT FORMAT AS THE QUESTION.
             QUESTION: {question}
-            FILE: {context}
+            {file_section}
             ANSWER:
             """
         )
@@ -240,10 +240,11 @@ class MagAgent:
                 while not self.rate_limiter.consume(1):
                     print(f"Rate limit reached. Waiting...")
                     await asyncio.sleep(4)
-            # Format the prompt with question and file_path as context
-            task = self.prompt.format(
+            # Conditionally include FILE: section only if file_path is provided
+            file_section = f"FILE: {file_path}" if file_path else ""
+            task = self.prompt_template.format(
                 question=question,
-                context=file_path if file_path else ""
+                file_section=file_section
             )
             print(f"Calling agent.run...")
             response = await asyncio.to_thread(self.agent.run, task=task)