Add tests and fix research tools

1 files changed, 17 insertions, 8 deletions

diff --git a/main.py b/main.py
index 96eb7ef..128339a 100644
--- a/main.py
+++ b/main.py
@@ -41,6 +41,7 @@ AGENT_TIMEOUT = int(os.environ.get("SLOPBOT_AGENT_TIMEOUT", "120"))
 AGENT_MODEL = os.environ.get("SLOPBOT_AGENT_MODEL", "parasail/kimi-k26")
 AGENT_BIN = os.environ.get("SLOPBOT_AGENT", "agent")
 AGENT_PATH = os.environ.get("SLOPBOT_AGENT_PATH", os.environ.get("PATH", ""))
+AGENT_DEBUG = os.environ.get("SLOPBOT_AGENT_DEBUG", "").lower() in {"1", "true", "yes", "on"}
 WORKERS = int(os.environ.get("SLOPBOT_WORKERS", "4"))
 AGENT_WORKERS = int(os.environ.get("SLOPBOT_AGENT_WORKERS", "1"))
 HISTORY_LIMIT = int(os.environ.get("SLOPBOT_HISTORY_LIMIT", "20"))
@@ -126,16 +127,24 @@ def _agent(prompt: str) -> tuple[int, str, str]:
     """Run agent once; return (returncode, stdout, stderr)."""
     env = os.environ.copy()
     env["PATH"] = AGENT_PATH
+    command = [
+        AGENT_BIN,
+        "--model", AGENT_MODEL,
+        "--eval-cwd", str(BASE_DIR),
+    ]
+    if AGENT_DEBUG:
+        command.append("--debug")
+    command.append(prompt)
+
     result = subprocess.run(
-        [
-            AGENT_BIN,
-            "--model", AGENT_MODEL,
-            "--eval-cwd", str(BASE_DIR),
-            prompt,
-        ],
-        capture_output=True, text=True, timeout=AGENT_TIMEOUT, env=env,
+        command,
+        stdout=subprocess.PIPE,
+        stderr=None if AGENT_DEBUG else subprocess.PIPE,
+        text=True,
+        timeout=AGENT_TIMEOUT,
+        env=env,
     )
-    return result.returncode, result.stdout.strip(), result.stderr.strip()
+    return result.returncode, result.stdout.strip(), (result.stderr or "").strip()
 
 
 def _unique_session_name(msg: dict) -> str: