codex-python 1.114.0__tar.gz → 1.114.1__tar.gz

{codex_python-1.114.0 → codex_python-1.114.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codex-python
-Version: 1.114.0
+Version: 1.114.1
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: Python :: 3.12

{codex_python-1.114.0 → codex_python-1.114.1}/codex/__init__.py

@@ -14,7 +14,7 @@ from codex.options import (
     TurnOptions,
 )
 
-__version__ = "1.114.0"
+__version__ = "1.114.1"
 
 __all__ = [
     "Codex",

{codex_python-1.114.0 → codex_python-1.114.1}/codex/app_server/_session.py

@@ -31,6 +31,7 @@ from codex.protocol import types as protocol
 _ModelT = TypeVar("_ModelT", bound=BaseModel)
 _RequestT = TypeVar("_RequestT", bound=BaseModel)
 _NotificationPredicate = Callable[[Notification], bool]
+_SubscriptionMessage = Notification | Exception | None
 
 
 class _NotificationSink:
@@ -41,7 +42,7 @@ class _NotificationSink:
     ) -> None:
         self.methods = methods
         self.predicate = predicate
-        self.queue: asyncio.Queue[Notification | None] = asyncio.Queue()
+        self.queue: asyncio.Queue[_SubscriptionMessage] = asyncio.Queue()
 
     def matches(self, method: str, notification: Notification) -> bool:
         if self.methods is not None and method not in self.methods:
@@ -52,13 +53,15 @@ class _NotificationSink:
 @dataclass(slots=True)
 class _AsyncNotificationSubscription:
     sink: _NotificationSink
-    queue: asyncio.Queue[Notification | None]
+    queue: asyncio.Queue[_SubscriptionMessage]
     close_callback: Callable[[], None]
 
     async def next(self) -> Notification:
         message = await self.queue.get()
         if message is None:
             raise StopAsyncIteration
+        if isinstance(message, Exception):
+            raise message
         return message
 
     async def close(self) -> None:
@@ -253,7 +256,7 @@ class _AsyncSession:
             self._reader_error = exc
             self._fail_pending(exc)
             for sink in list(self._notification_sinks):
-                await sink.queue.put(None)
+                await sink.queue.put(exc)
 
     async def _await_future(self, future: asyncio.Future[object]) -> object:
         while True:

{codex_python-1.114.0 → codex_python-1.114.1}/codex/app_server/transports.py

@@ -17,6 +17,8 @@ from codex.app_server.errors import (
 )
 from codex.app_server.options import AppServerProcessOptions, AppServerWebSocketOptions
 
+STDIO_STREAM_LIMIT_BYTES = 4 * 1024 * 1024
+
 
 class AsyncMessageTransport(Protocol):
     async def start(self) -> None: ...
@@ -70,6 +72,7 @@ class AsyncStdioTransport:
                 stdin=asyncio.subprocess.PIPE,
                 stdout=asyncio.subprocess.PIPE,
                 stderr=asyncio.subprocess.PIPE,
+                limit=STDIO_STREAM_LIMIT_BYTES,
                 env=_build_env(self._options),
             )
         except OSError as exc:
@@ -85,7 +88,7 @@ class AsyncStdioTransport:
 
     async def _drain_stderr(self, stderr: asyncio.StreamReader) -> None:
         while True:
-            line = await stderr.readline()
+            line = await _readline_with_limit_error(stderr, stream_name="stderr")
             if line == b"":
                 break
             self._stderr_lines.append(line.decode("utf-8", errors="replace").rstrip())
@@ -103,7 +106,7 @@ class AsyncStdioTransport:
     async def receive(self) -> JsonObject | None:
         if self._process is None or self._process.stdout is None:
             raise AppServerClosedError("app-server stdio transport is not running")
-        line = await self._process.stdout.readline()
+        line = await _readline_with_limit_error(self._process.stdout, stream_name="stdout")
         if line == b"":
             return None
         try:
@@ -240,3 +243,17 @@ def _exception_types(candidate: object) -> tuple[type[BaseException], ...]:
     if isinstance(candidate, type) and issubclass(candidate, BaseException):
         return (candidate,)
     return ()
+
+
+async def _readline_with_limit_error(
+    stream: asyncio.StreamReader,
+    *,
+    stream_name: str,
+) -> bytes:
+    try:
+        return await stream.readline()
+    except ValueError as exc:
+        raise AppServerProtocolError(
+            "app-server stdio "
+            f"{stream_name} line exceeded configured limit of {STDIO_STREAM_LIMIT_BYTES} bytes"
+        ) from exc

{codex_python-1.114.0 → codex_python-1.114.1}/crates/codex_native/Cargo.lock

@@ -20,7 +20,7 @@ dependencies = [
 
 [[package]]
 name = "codex_native"
-version = "1.114.0"
+version = "1.114.1"
 dependencies = [
  "pyo3",
 ]

{codex_python-1.114.0 → codex_python-1.114.1}/crates/codex_native/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "codex_native"
-version = "1.114.0"
+version = "1.114.1"
 edition = "2021"
 
 [lib]