python-codex 0.4.3__py3-none-any.whl → 0.4.4__py3-none-any.whl

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)

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":

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.dist-info → python_codex-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-codex
-Version: 0.4.3
+Version: 0.4.4
 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.dist-info → python_codex-0.4.4.dist-info}/RECORD

@@ -1,6 +1,6 @@
 pycodex/__init__.py,sha256=fHEyhylULeRo-5xEnl-UxT5ZpBzbFzRW7vLwdy_7oxU,2660
 pycodex/__main__.py,sha256=MHKZ_ae3fSLGTLUUMOx15fWdeOnJSHhq-zslRP5F5Lc,79
-pycodex/agent.py,sha256=zCIG2DV0oQntLXHFj8FVRkgBp5TCC0YJ7sz0lni9J1s,21177
+pycodex/agent.py,sha256=GF-mNTnxHW8qw-72QEXfGGvExZzl8c3m6emZBux6ddw,21418
 pycodex/bootstrap.py,sha256=U1X5kwORpS5JnPk9HhIiOVWPQzDLlk_ro_cY2knwL8I,12421
 pycodex/cli.py,sha256=MLvXpVwrb-HJuVWmPFmNMTR-4cqR3X08ZfAevYlVSQI,12646
 pycodex/context.py,sha256=yfNY2alT3caQQwhay47pSSrGUbgEAcLKhTVaZqtQ4yU,26610
@@ -37,7 +37,7 @@ pycodex/tools/read_file_tool.py,sha256=YK__uhIVldBjAQega5S_ZuZU5jgP_kLOwr-YhjAmR
 pycodex/tools/request_permissions_tool.py,sha256=nEt3_uw3lmFXyydqcN3PxuKwrEvaRPtfjA_U8nikCtU,3978
 pycodex/tools/request_user_input_tool.py,sha256=_7siHF2p3KiNDFwsTqKKIUw2kPCXMzLaR12PkZ1wdIQ,4050
 pycodex/tools/resume_agent_tool.py,sha256=bZvXEUNdc1RVD2N5Msjv63Rni62FQk4gj9LAXK1Vsy8,1613
-pycodex/tools/runtime_services.py,sha256=vZHIHGAeTA-DoEoHulAKvjcfHLXkuK1t3T0Mob9VJ8A,15044
+pycodex/tools/runtime_services.py,sha256=KPLtB1e7jmqEN9rhXV70rUMlhYNFsXja4CypIunzXLU,15962
 pycodex/tools/send_input_tool.py,sha256=w63SKM05_e3oXvdaBtL-hk941TvkWKuTUDEqTSyWEMQ,3761
 pycodex/tools/shell_command_tool.py,sha256=US63I5xiV5zfO1Oox2yHUg7F6XoKzmyxtohfaQIsmrw,4216
 pycodex/tools/shell_tool.py,sha256=lqFH2XosfBg7yQKhpB7i4vLGz9sNQtEZihD0rjsmBCQ,4068
@@ -45,7 +45,7 @@ pycodex/tools/spawn_agent_tool.py,sha256=Jx5jYpxsH_CQ3BLkXYPZuIINW5deO_zY8tyw8kZ
 pycodex/tools/unified_exec_manager.py,sha256=XPN6PgBnjZ5WFZATNIqpWKUAuw8HMMD7mlfHJQjJQZM,14544
 pycodex/tools/update_plan_tool.py,sha256=xwrSrerw5qfkbFc9Ltx1OZEmD__CWtryOy3UhL0-Jp4,3198
 pycodex/tools/view_image_tool.py,sha256=SCoiIHaQ51oYjYX69MRxjAK9b1P6_wOCIuZqzsC2rj8,4082
-pycodex/tools/wait_agent_tool.py,sha256=9Os6RzzCxIU5KJI5l6MzdwO6QKLNxLFwFXGeDMfYlP4,3828
+pycodex/tools/wait_agent_tool.py,sha256=pLsnk9TxSt4v7krC7s7vxi91pLisSNU0lATt9s-P_3E,3134
 pycodex/tools/wait_tool.py,sha256=0_OO_DvMsBhG3oDKZePTOjZTBq1lmWntXAzqIqvebdA,3206
 pycodex/tools/web_search_tool.py,sha256=3ao1IvJw7MwcP35UKLbqrM-nlm5Z-UUFVdEzkJYOV1o,988
 pycodex/tools/write_stdin_tool.py,sha256=jvBDchLLmoDITd7g-ZVeEVXHRhWKkmnjM5laL-53FOM,3609
@@ -83,8 +83,8 @@ responses_server/trajectory_dump.py,sha256=5_Wy6tslszrqx3j1JeP_boyTbzIicKy8No84P
 responses_server/tools/__init__.py,sha256=ivsBSEy0SBUhY-Uea5v1XMLXShkwHdCVl0id-1FwdZg,150
 responses_server/tools/custom_adapter.py,sha256=SnIiNUdomff3APtTriURAC9YCeEdhCDjzGMEYusfUdk,8340
 responses_server/tools/web_search.py,sha256=pnKH9JyTScRehWpZS3HpXRvjWsQvC0QZqxjEOCT4ndg,8936
-python_codex-0.4.3.dist-info/METADATA,sha256=hXum9V3ngy2KFpqKDcpssNeZrES-cHTFhiAGDTgGCcI,47510
-python_codex-0.4.3.dist-info/WHEEL,sha256=KGYbc1zXlYddvwxnNty23BeaKzh7YuoSIvIMO4jEhvw,87
-python_codex-0.4.3.dist-info/entry_points.txt,sha256=sNUVakoVuTrzJH505ZgRTQxmtRRPUHV_EH0i6EbYTyM,45
-python_codex-0.4.3.dist-info/licenses/LICENSE,sha256=0X8ifk312hYAORM4hlzg8wVSEXYKNmiPgWlB1YIy2Nw,10926
-python_codex-0.4.3.dist-info/RECORD,,
+python_codex-0.4.4.dist-info/METADATA,sha256=or8DXQM9IhgXgr3AY-tLyPNAdE2oGvcqY8l8EfV-PGs,47674
+python_codex-0.4.4.dist-info/WHEEL,sha256=KGYbc1zXlYddvwxnNty23BeaKzh7YuoSIvIMO4jEhvw,87
+python_codex-0.4.4.dist-info/entry_points.txt,sha256=sNUVakoVuTrzJH505ZgRTQxmtRRPUHV_EH0i6EbYTyM,45
+python_codex-0.4.4.dist-info/licenses/LICENSE,sha256=0X8ifk312hYAORM4hlzg8wVSEXYKNmiPgWlB1YIy2Nw,10926
+python_codex-0.4.4.dist-info/RECORD,,