update code for better test

Author: nanjiangwill

agent/agents.py

@@ -62,6 +62,7 @@ def run(
         lint_cmd: str,
         fnames: list[str],
         log_dir: Path,
+        test_first: bool = False,
     ) -> AgentReturn:
         """Start aider agent"""
         if test_cmd:
@@ -111,7 +112,14 @@ def run(
         coder.stream = True
 
         # Run the agent
-        coder.run(message)
+        if test_first:
+            test_errors = coder.commands.cmd_test(test_cmd)
+            if test_errors:
+                coder.run(test_errors)
+        else:
+            coder.run(message)
+
+        # #### TMP
 
         # #### TMP
         # import time

agent/run_agent.py

@@ -139,6 +139,7 @@ def run_agent_for_repo(
                     lint_cmd,
                     target_edit_files,
                     test_log_dir,
+                    test_first=True,
                 )
                 # after running the agent, update the money display
                 update_queue.put(

agent/run_agent_no_rich.py

@@ -130,6 +130,7 @@ def run_agent_for_repo(
                     lint_cmd,
                     target_edit_files,
                     test_log_dir,
+                    test_first=True,
                 )
                 # cost = agent_return.last_cost
         else: