codex-telegram 0.4.7__tar.gz → 0.4.9__tar.gz

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codex-telegram
-Version: 0.4.7
+Version: 0.4.9
 Summary: Bridge for controlling the Codex App Server from Telegram and Web UI
 Author: heungtae
 License:                                  Apache License

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/codex/client.py

@@ -1,5 +1,6 @@
 import asyncio
 import inspect
+from dataclasses import dataclass
 from typing import Any, Callable
 import logging
 
@@ -10,6 +11,12 @@ from utils.config import get
 logger = logging.getLogger("codex-telegram.codex")
 
 
+@dataclass(slots=True)
+class _ApprovalWaiter:
+    future: asyncio.Future[str]
+    loop: asyncio.AbstractEventLoop
+
+
 class CodexClient:
     def __init__(self):
         self.protocol = Protocol()
@@ -17,7 +24,7 @@ class CodexClient:
         self._reader_task: asyncio.Task | None = None
         self._stderr_task: asyncio.Task | None = None
         self._pending: dict[int, asyncio.Future[JSONRPCResponse]] = {}
-        self._pending_approvals: dict[int, asyncio.Future[str]] = {}
+        self._pending_approvals: dict[int, _ApprovalWaiter] = {}
         self._event_handlers: dict[str, list[Callable]] = {}
         self._any_event_handlers: list[Callable] = []
         self._approval_handlers: list[Callable] = []
@@ -114,10 +121,18 @@ class CodexClient:
         self._approval_handlers.append(handler)
 
     def submit_approval_decision(self, request_id: int, decision: str) -> bool:
-        future = self._pending_approvals.get(request_id)
-        if future is None or future.done():
+        waiter = self._pending_approvals.get(request_id)
+        if waiter is None or waiter.future.done():
             return False
-        future.set_result(decision)
+
+        def _resolve() -> None:
+            if not waiter.future.done():
+                waiter.future.set_result(decision)
+
+        if waiter.loop.is_running():
+            waiter.loop.call_soon_threadsafe(_resolve)
+        else:
+            _resolve()
         return True
     
     def _write(self, msg: JSONRPCRequest | JSONRPCNotification | JSONRPCResponse):
@@ -132,8 +147,8 @@ class CodexClient:
                 future.set_exception(exc)
         self._pending.clear()
         for future in self._pending_approvals.values():
-            if not future.done():
-                future.set_result("deny")
+            if not future.future.done():
+                future.future.set_result("deny")
         self._pending_approvals.clear()
     
     async def _read_stdout_stream(self):
@@ -499,8 +514,9 @@ class CodexClient:
                         )
                         return
 
-                future: asyncio.Future[str] = asyncio.Future()
-                self._pending_approvals[req_id] = future
+                loop = asyncio.get_running_loop()
+                future: asyncio.Future[str] = loop.create_future()
+                self._pending_approvals[req_id] = _ApprovalWaiter(future=future, loop=loop)
                 await self._emit_approval_request(method, req_id, params)
 
                 decision_choice = self._default_choice(auto_mode)

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/codex_telegram/__init__.py

@@ -1,3 +1,3 @@
 """codex-telegram package."""
 
-__version__ = "0.4.7"
+__version__ = "0.4.9"

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/codex_telegram.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codex-telegram
-Version: 0.4.7
+Version: 0.4.9
 Summary: Bridge for controlling the Codex App Server from Telegram and Web UI
 Author: heungtae
 License:                                  Apache License

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/tests/test_codex_client.py

@@ -1,4 +1,6 @@
+import asyncio
 import unittest
+import threading
 from unittest.mock import patch
 from types import SimpleNamespace
 
@@ -89,6 +91,45 @@ class CodexClientServerRequestTests(unittest.IsolatedAsyncioTestCase):
         answers = ((response.get("result") or {}).get("answers") or {})
         self.assertEqual({"answers": ["Approve this Session"]}, answers.get("q2"))
 
+    async def test_submit_approval_decision_is_thread_safe(self):
+        client = CodexClient()
+        written = []
+        client._write = lambda msg: written.append(msg)  # type: ignore[method-assign]
+        req = JSONRPCRequest(
+            method="item/commandExecution/requestApproval",
+            id=42,
+            params={"threadId": "thread-1"},
+        )
+
+        with patch(
+            "codex.client.get",
+            side_effect=lambda key, default=None: "interactive" if key == "approval.mode" else "approve" if key == "approval.auto_response" else default,
+        ):
+            task = asyncio.create_task(client._handle_server_request(req))
+            for _ in range(100):
+                if 42 in client._pending_approvals:
+                    break
+                await asyncio.sleep(0.01)
+            else:
+                self.fail("approval request was not registered")
+
+            accepted = {}
+
+            def approve_from_thread():
+                accepted["value"] = client.submit_approval_decision(42, "approve")
+
+            thread = threading.Thread(target=approve_from_thread)
+            thread.start()
+            thread.join()
+
+            await asyncio.wait_for(task, timeout=1.0)
+
+        self.assertTrue(accepted["value"])
+        self.assertEqual(1, len(written))
+        response = written[0].to_dict()
+        self.assertEqual(42, response.get("id"))
+        self.assertEqual({"decision": "accept"}, response.get("result"))
+
     async def test_tool_request_user_input_prefers_option_value_when_present(self):
         client = CodexClient()
         written = []

{codex-telegram-0.4.7 → codex-telegram-0.4.9}/web/workspace.py

@@ -62,7 +62,7 @@ def workspace_suggestions(workspace: str, prefix: str, limit: int) -> list[str]:
     partial_lower = partial.lower()
     recursive_mode = not base_rel and "/" not in normalized_prefix and bool(partial_lower)
 
-    if recursive_mode or (base_rel and partial_lower):
+    if recursive_mode:
         max_scan = 4000
         scanned = 0
         try: