agentic-python-coder 3.3.0__py3-none-any.whl → 3.4.0__py3-none-any.whl

agentic_python_coder/kernel.py

@@ -10,6 +10,7 @@ import logging
 import os
 import re
 import shutil
+import signal
 import threading
 import time
 import uuid
@@ -311,7 +312,21 @@ except ImportError:
             return self._execute_raw(code, deadline_timeout)
 
     def shutdown(self):
-        """Shutdown the kernel and clean up."""
+        """Shutdown the kernel and clean up.
+
+        Kernels run in their own process group (jupyter_client launches with
+        start_new_session=True). After the regular shutdown, defensively kill
+        the whole group: uv-wrapped kernels are a tree (uv parent + python
+        child) and a surviving member would be orphaned.
+        """
+        # Capture the kernel's process group before shutdown clears it
+        pgid = None
+        try:
+            if hasattr(self, "km") and self.km:
+                pgid = getattr(getattr(self.km, "provisioner", None), "pgid", None)
+        except Exception:
+            pass
+
         try:
             if hasattr(self, "kc") and self.kc:
                 try:
@@ -330,6 +345,15 @@ except ImportError:
             # Ignore errors during shutdown
             pass
 
+        # Defensive: ensure no process-tree member survived. Never signal our
+        # own group (pgid could be bogus if start_new_session did not apply).
+        if pgid and pgid > 0:
+            try:
+                if pgid != os.getpgid(0):
+                    os.killpg(pgid, signal.SIGKILL)
+            except (ProcessLookupError, PermissionError, OSError):
+                pass
+
 
 # =============================================================================
 # Core Registry Functions

agentic_python_coder/mcp_server.py

@@ -7,6 +7,9 @@ import ast
 import asyncio
 import json
 import logging
+import os
+import signal
+import threading
 
 from mcp.server import Server
 from mcp.server.stdio import stdio_server
@@ -20,6 +23,7 @@ from agentic_python_coder.kernel import (
     kernel_exists,
     list_kernels,
     restart_kernel,
+    shutdown_all_kernels,
 )
 
 # Configure logging
@@ -240,6 +244,30 @@ No side effects - safe to call anytime.""",
                 },
             },
         ),
+        Tool(
+            name="submit_code",
+            description="""Submit your final, verified, self-contained Python program.
+
+Call this ONCE at the end of a solve, only after you have executed the program
+via python_exec and your verification passed. The submission must be a
+standalone program: all imports included, no reliance on session state.
+
+The code is syntax-checked with compile():
+- {"ok": true} if it parses
+- {"ok": false, "error": "<detail>"} on a syntax error — fix and resubmit
+
+The code is NOT executed and NOT written to disk; the client persists it.""",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "code": {
+                        "type": "string",
+                        "description": "Complete standalone Python program (final answer).",
+                    },
+                },
+                "required": ["code"],
+            },
+        ),
         Tool(
             name="python_interrupt",
             description="""Interrupt running code in the session.
@@ -391,6 +419,21 @@ versions
         result = await execute_with_timeout(code, timeout, kernel_id)
         return [TextContent(type="text", text=json.dumps(result))]
 
+    elif name == "submit_code":
+        code = arguments.get("code", "")
+        if not code.strip():
+            payload = {"ok": False, "error": "Empty submission"}
+        else:
+            try:
+                compile(code, "<submission>", "exec")
+                payload = {"ok": True}
+            except SyntaxError as e:
+                payload = {
+                    "ok": False,
+                    "error": f"Syntax error at line {e.lineno}: {e.msg}",
+                }
+        return [TextContent(type="text", text=json.dumps(payload))]
+
     elif name == "python_status":
         kernel_id = get_kernel_id(arguments)
         is_active = kernel_exists(kernel_id)
@@ -521,9 +564,39 @@ async def run_server():
         await server.run(read, write, server.create_initialization_options())
 
 
+# Hard-exit fallback if kernel cleanup hangs after a termination signal
+_TERMINATION_CLEANUP_TIMEOUT = 10.0
+
+
+def _handle_termination(signum, frame):
+    """Shut down all kernels on SIGTERM/SIGINT, then exit.
+
+    The cleanup runs in a thread (not in signal-handler context) so it can
+    safely take the kernel-registry locks; a timer hard-exits if it hangs.
+    """
+
+    def _cleanup_and_exit():
+        try:
+            shutdown_all_kernels()
+        finally:
+            os._exit(128 + signum)
+
+    threading.Timer(
+        _TERMINATION_CLEANUP_TIMEOUT, lambda: os._exit(128 + signum)
+    ).start()
+    threading.Thread(target=_cleanup_and_exit, daemon=True).start()
+
+
 def main():
     """Entry point for coder-mcp command."""
-    asyncio.run(run_server())
+    signal.signal(signal.SIGTERM, _handle_termination)
+    signal.signal(signal.SIGINT, _handle_termination)
+    try:
+        asyncio.run(run_server())
+    finally:
+        # stdin EOF / transport close / normal exit: deterministic cleanup
+        # (atexit also runs this, but only on clean interpreter shutdown)
+        shutdown_all_kernels()
 
 
 if __name__ == "__main__":

{agentic_python_coder-3.3.0.dist-info → agentic_python_coder-3.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-python-coder
-Version: 3.3.0
+Version: 3.4.0
 Summary: A lightweight Python coding agent that writes, executes, and iterates on code through natural language instructions
 Author: Stefan Szeider
 License: Apache-2.0

{agentic_python_coder-3.3.0.dist-info → agentic_python_coder-3.4.0.dist-info}/RECORD

@@ -1,9 +1,9 @@
 agentic_python_coder/__init__.py,sha256=e6o-k__u5xNIiQAkZbHtbZC2-NwpeOdi4MdQtAqJBjE,2112
 agentic_python_coder/agent.py,sha256=gzDFf4n7J1JdWawYPzWQFVwcHZ2rsRHUzVZVcGYjQWo,14451
 agentic_python_coder/cli.py,sha256=HS9nlqsiWZAvg0Hc8ExoVT8eEcy2t3Bux6UGukERWqQ,14663
-agentic_python_coder/kernel.py,sha256=PQrKMi0_mKVeG1q2wXTu3-W2ZYKwYNjtPI_3A4m4Gvg,21284
+agentic_python_coder/kernel.py,sha256=JFN3gtZp3S-bGbQf2B-iQAE3AF8Q1TyQg4wrsx4Fdok,22261
 agentic_python_coder/llm.py,sha256=zjsk4-Kdx92hFSxywPGIkLv8zflWnE2cMEcpeXu6ttw,6998
-agentic_python_coder/mcp_server.py,sha256=2Z5O9n5QgZrgn3GdFljjF7Yq5e9tE1zNlAWBPyckrlI,18503
+agentic_python_coder/mcp_server.py,sha256=7Qs5kJ9_W238vIywN5OIMG01vuPzfQ1cy1xFR8C7nGM,21054
 agentic_python_coder/project_md.py,sha256=4xkQHlWb4cDPEkvegw1YwcfaTJ3aFbrl0CYcC_xxABY,3249
 agentic_python_coder/runner.py,sha256=Kbw1ceYJ4MVfUKgSIkiuBhMb8ZtbZuJwbBeTUsvPRts,13045
 agentic_python_coder/tools.py,sha256=tjhQbsF3n3bFIegnEOtE8czjzbtMSXy6B9gYeb-YAJ4,12088
@@ -41,8 +41,8 @@ agentic_python_coder/models/sonnet46.json,sha256=Ops1FpBoyx9PBGRhIXG8u-YGBgIui9D
 agentic_python_coder/models/sonnet5.json,sha256=kf6DIkQlcGVZ40IzRv3GXFtNncZF_gKqQeGzZAxE_KM,87
 agentic_python_coder/prompts/system.md,sha256=hREDVAhJXwCBAgJ3AE9JD6j-JRHREUPlkRSvThOvM3A,3807
 agentic_python_coder/prompts/system_todo.md,sha256=TuNeeUNS9KqyF62Da6t2mlUCYhKsgzwGe5ic_vFPDqw,5054
-agentic_python_coder-3.3.0.dist-info/METADATA,sha256=ec4FIlQitVNa_5bMQhu2NPpJV-5VwDRQy1hzK_4b_G8,12872
-agentic_python_coder-3.3.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-agentic_python_coder-3.3.0.dist-info/entry_points.txt,sha256=XIW3nZzL4kJ-Igohilj_mTvBcCyHBouDq2jmGzDILrk,107
-agentic_python_coder-3.3.0.dist-info/licenses/LICENSE,sha256=R9-M_V-SPdN71IphOVzM7kCNpUYs_hwj7rtroTwDhs0,11362
-agentic_python_coder-3.3.0.dist-info/RECORD,,
+agentic_python_coder-3.4.0.dist-info/METADATA,sha256=A0vE8oBRtMdgx3kwnaVwZSzhCgTi-VXBRCxcJotjXY0,12872
+agentic_python_coder-3.4.0.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+agentic_python_coder-3.4.0.dist-info/entry_points.txt,sha256=XIW3nZzL4kJ-Igohilj_mTvBcCyHBouDq2jmGzDILrk,107
+agentic_python_coder-3.4.0.dist-info/licenses/LICENSE,sha256=R9-M_V-SPdN71IphOVzM7kCNpUYs_hwj7rtroTwDhs0,11362
+agentic_python_coder-3.4.0.dist-info/RECORD,,