PyPI - python-codex - Versions diffs - 0.2.0__tar.gz → 0.2.2__tar.gz - Mend

python-codex 0.2.0tar.gz → 0.2.2tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (121) hide show

{python_codex-0.2.0 → python_codex-0.2.2}/PKG-INFO RENAMED Viewed

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-codex
-Version: 0.2.0
+Version: 0.2.2
 Summary: A minimal Python extraction of Codex's main agent loop
 License-File: LICENSE
 Requires-Python: >=3.6.2

{python_codex-0.2.0 → python_codex-0.2.2}/pycodex/interactive_session.py RENAMED Viewed

@@ -18,6 +18,7 @@ import typing
 EXIT_COMMANDS = {"/exit", "/quit"}
+HELP_COMMAND = "/help"
 HISTORY_COMMAND = "/history"
 TITLE_COMMAND = "/title"
 MODEL_COMMAND = "/model"
@@ -27,7 +28,7 @@ COMPACT_COMMAND = "/compact"
 LINK_COMMAND = "/link"
 UNLINK_COMMAND = "/unlink"
 EXTRA_COMMANDS_LINE = (
-    "Extra commands: /history, /title, /model, /resume, /compact, /link, /unlink"
+    "Extra commands: /help, /history, /title, /model, /resume, /compact, /link, /unlink"
 )
@@ -232,6 +233,9 @@ async def run_interactive_session(
                 continue
             if prompt_text in EXIT_COMMANDS:
                 break
+            if prompt_text == HELP_COMMAND:
+                view.write_line(EXTRA_COMMANDS_LINE)
+                continue
             if prompt_text == HISTORY_COMMAND:
                 view.show_history()
                 continue

{python_codex-0.2.0 → python_codex-0.2.2}/pyproject.toml RENAMED Viewed

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 [project]
 name = "python-codex"
-version = "0.2.0"
+version = "0.2.2"
 description = "A minimal Python extraction of Codex's main agent loop"
 readme = "README.md"
 requires-python = ">=3.6.2"

{python_codex-0.2.0 → python_codex-0.2.2}/tests/test_workspace_server.py RENAMED Viewed

@@ -8,6 +8,7 @@ import pytest
 from pycodex import (
     Agent,
+    AgentEvent,
     AssistantMessage,
     BaseTool,
     CliSubmissionQueue,
@@ -193,6 +194,7 @@ def test_workspace_app_missing_board_returns_pending_page(tmp_path) -> None:
     app = create_app(lambda: link, board)
     with TestClient(app) as client:
+        shell = client.get("/")
         response = client.get("/board")
         status = client.get("/api/board")
@@ -200,6 +202,9 @@ def test_workspace_app_missing_board_returns_pending_page(tmp_path) -> None:
     assert "Board pending" in response.text
     assert str(board) in response.text
     assert status.json() == {"exists": False}
+    assert "let lastBoardSignature = null;" in shell.text
+    assert "if (lastBoardSignature === null)" in shell.text
+    assert "boardFrame.src = `board?t=${Date.now()}`;" in shell.text
 def test_workspace_app_session_snapshot_uses_turns_not_raw_messages(tmp_path) -> None:
@@ -218,6 +223,9 @@ def test_workspace_app_session_snapshot_uses_turns_not_raw_messages(tmp_path) ->
     assert snapshot["turns"][0]["response"] == "pong"
     assert snapshot["turns"][0]["thinking"] == ""
+    sessions = response.json()["sessions"]
+    assert sessions[0]["last_assistant"] == "pong"
 def test_workspace_app_session_endpoint_returns_spinner_and_turns(tmp_path) -> None:
     board = tmp_path / "board.html"
@@ -236,6 +244,26 @@ def test_workspace_app_session_endpoint_returns_spinner_and_turns(tmp_path) -> N
     assert snapshot["turns"][0]["response"] == "response"
+def test_workspace_view_tracks_context_remaining_percent() -> None:
+    view = WebSessionView()
+    assert view.snapshot()["context_remaining_percent"] is None
+    view.set_context_window_tokens(100_000)
+    assert view.snapshot()["context_remaining_percent"] == 100
+    assert view.summary()["context_remaining_percent"] == 100
+    view.handle_event(
+        AgentEvent(
+            kind="token_count",
+            turn_id="turn_1",
+            payload={"usage": {"total_tokens": 20_800}},
+        )
+    )
+    assert view.snapshot()["context_remaining_percent"] == 90
+    assert view.summary()["context_remaining_percent"] == 90
 def test_workspace_app_shell_uses_spinner_without_send_button(tmp_path) -> None:
     board = tmp_path / "board.html"
     board.write_text("<!doctype html><title>Board</title>", encoding="utf-8")
@@ -247,7 +275,30 @@ def test_workspace_app_shell_uses_spinner_without_send_button(tmp_path) -> None:
         response = client.get("/")
     assert response.status_code == 200
-    assert 'id="spinner"' in response.text
+    assert 'class="spinner-wrap"' in response.text
+    assert 'class="prompt-row"' in response.text
+    assert 'id="spinner" class="spinner" type="button" role="checkbox"' in response.text
+    assert 'id="toast" class="toast" role="status"' in response.text
+    assert '<textarea id="prompt" rows="1"' in response.text
+    assert ".spinner-wrap:not(:has(.spinner:not(:empty)))" in response.text
+    assert response.text.index('class="prompt-row"') < response.text.index('id="contextMeter"')
+    assert response.text.index('id="contextMeter"') < response.text.index('id="prompt"')
+    assert 'id="contextMeter" class="context-meter hidden" aria-hidden="true"' in response.text
+    assert 'id="contextMeterValue" class="context-meter-value"' in response.text
+    assert 'contextMeter.style.setProperty("--context-remaining", `${value}%`)' in response.text
+    assert "contextMeter.setAttribute(\"aria-valuenow\"" not in response.text
+    assert "contextMeterValue.textContent = String(value)" in response.text
+    assert "sessionId === activeSessionId && spinnerText" in response.text
+    assert "spinner.addEventListener(\"click\", toggleSpinnerNotification)" in response.text
+    assert "new Notification(`session: ${title}`" in response.text
+    assert "body: lastAssistant" in response.text
+    assert "notifySessionDone(sessionId, session)" in response.text
+    assert "notify-on-done enabled" in response.text
+    assert "notify-on-done canceled" in response.text
+    assert "browser notifications blocked" in response.text
+    assert "requestNotificationPermission" in response.text
+    assert 'document.addEventListener("visibilitychange"' in response.text
+    assert 'document.visibilityState !== "visible"' in response.text
     assert ">Send<" not in response.text
     assert "Enter sends. Shift+Enter adds a newline." not in response.text
     assert "compositionstart" in response.text
@@ -281,7 +332,18 @@ def test_workspace_spinner_tool_call_preview_is_longer_than_title() -> None:
     assert spinner.startswith("calling exec_command(")
-def test_workspace_app_message_uses_shared_interactive_commands(tmp_path) -> None:
+@pytest.mark.parametrize(
+    ("command", "expected_response"),
+    [
+        ("/history", "No history yet."),
+        ("/help", "/history"),
+    ],
+)
+def test_workspace_app_message_uses_shared_interactive_commands(
+    tmp_path,
+    command: str,
+    expected_response: str,
+) -> None:
     board = tmp_path / "board.html"
     board.write_text("<!doctype html><title>Board</title>", encoding="utf-8")
     model = ScriptedModelClient(
@@ -295,10 +357,11 @@ def test_workspace_app_message_uses_shared_interactive_commands(tmp_path) -> Non
     app = create_app(build_session, board)
     with TestClient(app) as client:
-        response = client.post("/api/session/message", json={"prompt": "/history"})
+        response = client.post("/api/session/message", json={"prompt": command})
         snapshot = _wait_for_snapshot(
             client,
-            lambda item: item["turns"] and item["turns"][-1]["response"] == "No history yet.",
+            lambda item: item["turns"]
+            and expected_response in item["turns"][-1]["response"],
         )
     assert response.status_code == 200
@@ -871,6 +934,8 @@ def test_workspace_app_session_list_uses_lightweight_summary(tmp_path) -> None:
                 "running": True,
                 "spinner": "thinking",
                 "turn_count": 123,
+                "last_assistant": "latest answer",
+                "context_remaining_percent": 48,
             }
         def snapshot(self):
@@ -886,6 +951,8 @@ def test_workspace_app_session_list_uses_lightweight_summary(tmp_path) -> None:
     assert response.json()["sessions"][0]["running"] is True
     assert response.json()["sessions"][0]["spinner"] == "thinking"
     assert response.json()["sessions"][0]["turn_count"] == 123
+    assert response.json()["sessions"][0]["last_assistant"] == "latest answer"
+    assert response.json()["sessions"][0]["context_remaining_percent"] == 48
 def test_workspace_app_websocket_ping(tmp_path) -> None:

{python_codex-0.2.0 → python_codex-0.2.2}/workspace_server/app.py RENAMED Viewed

@@ -25,7 +25,12 @@ from pycodex.utils.session_persist import (
     load_resumed_session_path,
 )
 from pycodex.utils import uuid7_string
-from pycodex.utils.visualize import IDLE_LISTENING_STATUS, shorten_title, tool_summary
+from pycodex.utils.visualize import (
+    IDLE_LISTENING_STATUS,
+    percent_of_context_window_remaining,
+    shorten_title,
+    tool_summary,
+)
 import typing
@@ -180,6 +185,8 @@ class WebSessionView:
         self._title = ""
         self._spinner_status = ""
         self._stream_buffer = ""
+        self._context_window_tokens: "typing.Union[int, None]" = None
+        self._context_remaining_percent: "typing.Union[int, None]" = None
         self._closed = False
         self._server_loop: "typing.Union[asyncio.AbstractEventLoop, None]" = None
         self._worker_loop: "typing.Union[asyncio.AbstractEventLoop, None]" = None
@@ -347,7 +354,11 @@ class WebSessionView:
         self,
         context_window_tokens: "typing.Union[int, None]",
     ) -> None:
-        del context_window_tokens
+        with self._lock:
+            self._context_window_tokens = context_window_tokens
+            self._context_remaining_percent = (
+                100 if context_window_tokens is not None else None
+            )
     def subscribe(self) -> "asyncio.Queue":
         queue: "asyncio.Queue" = asyncio.Queue()
@@ -386,6 +397,7 @@ class WebSessionView:
                 "spinner": self._spinner_status,
                 "model": "pycodex",
                 "title": self._title,
+                "context_remaining_percent": self._context_remaining_percent,
                 "turns": [_public_turn(turn) for turn in self._turns[-80:]],
             }
@@ -396,6 +408,8 @@ class WebSessionView:
                 "spinner": self._spinner_status,
                 "title": self._title,
                 "turn_count": len(self._turns),
+                "last_assistant": _last_assistant_text(self._turns),
+                "context_remaining_percent": self._context_remaining_percent,
             }
     def _apply_runtime_event(self, event: "AgentEvent") -> None:
@@ -403,6 +417,9 @@ class WebSessionView:
         payload = getattr(event, "payload", {})
         if not isinstance(payload, dict):
             payload = {}
+        if kind == "token_count":
+            self._update_context_window(payload.get("usage"))
+            return
         turn_id = str(payload.get("turn_id") or getattr(event, "turn_id", "") or "")
         submission_id = str(payload.get("submission_id") or turn_id or "")
         turn = self._turns_by_submission_id.get(submission_id)
@@ -475,6 +492,22 @@ class WebSessionView:
             turn["status"] = "interrupted"
             self._stream_buffer = ""
+    def _update_context_window(self, usage: "object") -> None:
+        if self._context_window_tokens is None:
+            return
+        if not isinstance(usage, dict):
+            self._context_remaining_percent = None
+            return
+        try:
+            total_tokens = int(usage["total_tokens"])
+        except (KeyError, TypeError, ValueError):
+            self._context_remaining_percent = None
+            return
+        self._context_remaining_percent = percent_of_context_window_remaining(
+            total_tokens,
+            self._context_window_tokens,
+        )
     def _apply_spinner_event(
         self,
         kind: str,
@@ -974,6 +1007,10 @@ class WorkspaceSessionManager:
                     "running": bool(summary.get("running")),
                     "spinner": summary.get("spinner") or "",
                     "turn_count": summary.get("turn_count") or 0,
+                    "last_assistant": summary.get("last_assistant") or "",
+                    "context_remaining_percent": summary.get(
+                        "context_remaining_percent"
+                    ),
                 }
             )
         return result
@@ -1178,9 +1215,21 @@ def _session_summary(session) -> "typing.Dict[str, object]":
         "running": bool(snapshot.get("running")),
         "spinner": snapshot.get("spinner") or "",
         "turn_count": len(snapshot.get("turns") or []),
+        "last_assistant": _last_assistant_text(snapshot.get("turns") or []),
+        "context_remaining_percent": snapshot.get("context_remaining_percent"),
     }
+def _last_assistant_text(turns: "typing.Iterable[typing.Dict[str, object]]") -> str:
+    for turn in reversed(list(turns)):
+        if str(turn.get("kind") or "assistant") == "control":
+            continue
+        response = str(turn.get("response") or "").strip()
+        if response:
+            return response
+    return ""
 def _public_turn(turn: "typing.Dict[str, object]") -> "typing.Dict[str, object]":
     return typing.cast(
         "typing.Dict[str, object]",

python-codex 0.2.0__tar.gz → 0.2.2__tar.gz

python-codex 0.2.0tar.gz → 0.2.2tar.gz