local runs, home_path config, submodule miniagi (#50)

agent/benchmarks_example.py

@@ -0,0 +1,35 @@
+import os
+import sys
+from typing import Tuple
+import pexpect
+
+
+def run_specific_agent(task: str) -> Tuple[str, int]:
+    # Ensure the directory for the project exists
+    os.makedirs("workspace_path", exist_ok=True)
+
+    # Run the agent command
+    child = pexpect.spawn(f"python example.py {task}")
+
+    # Create a loop to continuously read output
+    while True:
+        try:
+            child.expect("\n")  # This waits until a newline appears
+            print(child.before.decode())  # This prints the line
+        except pexpect.EOF:
+            break  # No more output, break the loop
+
+    # Check the exit status
+    child.close()  # Close the child process
+
+    # Return child process's exit status and any error messages
+    return child.before.decode(), child.exitstatus
+
+
+if __name__ == "__main__":
+    # The first argument is the script name itself, second is the task
+    if len(sys.argv) != 2:
+        print("Usage: python script.py <task>")
+        sys.exit(1)
+    task = sys.argv[1]
+    run_specific_agent(task)