fix: polish code and handle some case

Signed-off-by: Chojan Shang <chojan.shang@vesoft.com>

Author: Chojan Shang

examples/mini_swe_agent/client.py

@@ -416,7 +416,7 @@ async def _run_connection(self) -> None:
             return
 
         # Autostep loop: take queued prompts and send; if none and mode != human, keep stepping
-        while True:
+        while self.agent_state != "STOPPED":
             blocks: list[ContentBlock1]
             try:
                 blocks = self._outbox.get_nowait()
@@ -450,6 +450,11 @@ def _ask_new():
 
                     threading.Thread(target=_ask_new, daemon=True).start()
             except Exception as e:
+                # Break on connection shutdowns to stop background thread cleanly
+                msg = str(e)
+                if isinstance(e, (BrokenPipeError, ConnectionResetError)) or "Broken pipe" in msg or "closed" in msg:
+                    self.agent_state = "STOPPED"
+                    break
                 self.call_from_thread(lambda: self.enqueue_message(UIMessage("assistant", f"prompt error: {e}")))
             # Tiny delay to avoid busy-looping
             await asyncio.sleep(0.05)

examples/mini_swe_agent/duet.py

@@ -16,7 +16,8 @@ async def main() -> None:
     try:
         from dotenv import load_dotenv  # type: ignore
 
-        load_dotenv(dotenv_path=str(root.parents[2] / ".env"), override=True)
+        # Load .env from repo root: examples/mini_swe_agent -> examples -> REPO
+        load_dotenv(dotenv_path=str(root.parents[1] / ".env"), override=True)
     except Exception:
         pass
 
@@ -60,8 +61,30 @@ async def main() -> None:
         with contextlib.suppress(OSError):
             os.close(fd)
 
-    # Wait for processes to finish; no relay needed
-    await asyncio.gather(agent.wait(), client.wait())
+    # If either process exits, terminate the other gracefully
+    agent_task = asyncio.create_task(agent.wait())
+    client_task = asyncio.create_task(client.wait())
+    done, pending = await asyncio.wait({agent_task, client_task}, return_when=asyncio.FIRST_COMPLETED)
+
+    # Terminate the peer process
+    if agent_task in done and client.returncode is None:
+        with contextlib.suppress(ProcessLookupError):
+            client.terminate()
+    if client_task in done and agent.returncode is None:
+        with contextlib.suppress(ProcessLookupError):
+            agent.terminate()
+
+    # Wait a bit, then kill if still running
+    try:
+        await asyncio.wait_for(asyncio.gather(agent.wait(), client.wait()), timeout=3)
+    except asyncio.TimeoutError:
+        with contextlib.suppress(ProcessLookupError):
+            if agent.returncode is None:
+                agent.kill()
+        with contextlib.suppress(ProcessLookupError):
+            if client.returncode is None:
+                client.kill()
+        await asyncio.gather(agent.wait(), client.wait())
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -40,6 +40,7 @@ dev = [
     "mkdocs-material>=8.5.10",
     "mkdocstrings[python]>=0.26.1",
     "mini-swe-agent>=1.10.0",
+    "python-dotenv>=1.1.1",
 ]
 
 [build-system]