python-codex 0.4.3__tar.gz → 0.4.5__tar.gz

{python_codex-0.4.3 → python_codex-0.4.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-codex
-Version: 0.4.3
+Version: 0.4.5
 Summary: A minimal Python extraction of Codex's main agent loop
 License-File: LICENSE
 Requires-Python: >=3.6.2
@@ -808,7 +808,9 @@ event JSON format.
   The global `get_agent_runtime_environment()` and
   separate `Agent(..., runtime_environment=...)` entry points are removed.
 - Each sub-agent gets its own context manager and runtime services. The sub-agent
-  service retains terminal event status, rather than inspecting Task results.
+  service retains terminal event status and queues one notification per terminal
+  event. The parent consumes these notifications before its next normal model
+  request, without requiring `wait_agent`; an idle parent waits for its next turn.
 - Every Agent records its session, including direct and child Agents; external
   callers no longer construct or inject `SessionRolloutRecorder`. The outer queue
   remains optional and no worker starts implicitly. Empty initial history and a

{python_codex-0.4.3 → python_codex-0.4.5}/README.md

@@ -788,7 +788,9 @@ event JSON format.
   The global `get_agent_runtime_environment()` and
   separate `Agent(..., runtime_environment=...)` entry points are removed.
 - Each sub-agent gets its own context manager and runtime services. The sub-agent
-  service retains terminal event status, rather than inspecting Task results.
+  service retains terminal event status and queues one notification per terminal
+  event. The parent consumes these notifications before its next normal model
+  request, without requiring `wait_agent`; an idle parent waits for its next turn.
 - Every Agent records its session, including direct and child Agents; external
   callers no longer construct or inject `SessionRolloutRecorder`. The outer queue
   remains optional and no worker starts implicitly. Empty initial history and a

{python_codex-0.4.3 → python_codex-0.4.5}/pycodex/agent.py

@@ -253,6 +253,10 @@ class Agent:
             while True:
                 phase = "pre_turn" if turn.iteration == 0 else "mid_turn"
                 await self._maybe_auto_compact(turn.turn_id, phase)
+                manager = self.tool_registry.runtime_environment.subagent_manager
+                notifications = manager.take_pending_notifications()
+                if notifications:
+                    self._append_history(notifications)
                 if turn.iteration == 0:
                     self._append_history(UserMessage(text=text) for text in texts)
                 response = await self._sample(turn)

{python_codex-0.4.3 → python_codex-0.4.5}/pycodex/tools/runtime_services.py

@@ -15,7 +15,7 @@ from ..events import (
     TurnInterruptedEvent,
     TurnStartedEvent,
 )
-from ..protocol import ConversationItem, ToolCall, ToolResult
+from ..protocol import ConversationItem, ToolCall, ToolResult, UserMessage
 from ..utils import uuid7_string
 from ..utils.compat import Literal
 
@@ -221,6 +221,7 @@ class SubAgentManager:
     def __init__(self) -> "None":
         self._runtime_builder: "typing.Union[AgentRuntimeBuilder, None]" = None
         self._agents: "typing.Dict[str, ManagedAgent]" = {}
+        self._pending_notifications: "typing.List[UserMessage]" = []
         self._condition = None
         self._available_nicknames: "typing.List[str]" = []
         self._nickname_random = random.Random()
@@ -235,6 +236,11 @@ class SubAgentManager:
     ) -> "None":
         self._runtime_builder = builder
 
+    def take_pending_notifications(self) -> "typing.Tuple[UserMessage, ...]":
+        messages = tuple(self._pending_notifications)
+        self._pending_notifications.clear()
+        return messages
+
     async def spawn_agent(
         self,
         message: "typing.Union[str, None]",
@@ -396,8 +402,27 @@ class SubAgentManager:
             }
         elif isinstance(event, TurnInterruptedEvent):
             managed.last_status = {"errored": "TurnInterrupted: turn interrupted"}
-        elif not isinstance(event, (TurnStartedEvent, CompactStartedEvent)):
+        elif isinstance(event, (TurnStartedEvent, CompactStartedEvent)):
+            asyncio.create_task(self._notify_waiters())
+            return
+        else:
             return
+        self._pending_notifications.append(
+            UserMessage(
+                text=(
+                    "<subagent_notification>\n"
+                    + json.dumps(
+                        {
+                            "agent_id": managed.agent_id,
+                            "status": managed.last_status,
+                        },
+                        ensure_ascii=False,
+                        separators=(",", ":"),
+                    )
+                    + "\n</subagent_notification>"
+                )
+            )
+        )
         asyncio.create_task(self._notify_waiters())
 
     async def _notify_waiters(self) -> "None":

{python_codex-0.4.3 → python_codex-0.4.5}/pycodex/tools/wait_agent_tool.py

@@ -9,10 +9,7 @@ Expected behavior:
   wait times out.
 """
 
-import json
-import typing
-
-from ..protocol import JSONDict, JSONValue, UserMessage
+from ..protocol import JSONDict, JSONValue
 from .agent_tool_schemas import AGENT_STATUS_SCHEMA
 from .base_tool import BaseTool, ToolContext
 from .runtime_services import SubAgentManager
@@ -80,26 +77,6 @@ class WaitAgentTool(BaseTool):
         timeout_ms = self._timeout_ms(args)
         return await self._subagent_manager.wait_agents(agent_ids, timeout_ms)
 
-    def follow_up_messages(
-        self, output: "JSONValue"
-    ) -> "typing.Tuple[UserMessage, ...]":
-        if not isinstance(output, dict):
-            return ()
-        return tuple(
-            UserMessage(
-                text=(
-                    "<subagent_notification>\n"
-                    + json.dumps(
-                        {"agent_id": agent_id, "status": status},
-                        ensure_ascii=False,
-                        separators=(",", ":"),
-                    )
-                    + "\n</subagent_notification>"
-                )
-            )
-            for agent_id, status in output["status"].items()
-        )
-
     def _timeout_ms(self, args: "JSONDict") -> "int":
         value = int(args.get("timeout_ms", DEFAULT_WAIT_AGENT_TIMEOUT_MS))
         return min(

{python_codex-0.4.3 → python_codex-0.4.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "python-codex"
-version = "0.4.3"
+version = "0.4.5"
 description = "A minimal Python extraction of Codex's main agent loop"
 readme = "README.md"
 requires-python = ">=3.6.2"
@@ -40,7 +40,7 @@ test = [
 ]
 dev = [
     { include-group = "test" },
-    "pycodex-ws==0.4.3",
+    "pycodex-ws==0.4.5",
     "black>=22.8.0",
     "debugpy>=1.8.20; python_version >= '3.8'",
     "isort>=5.10.1",