jupyterlab-codex-sidebar 0.1.2 → 0.1.3

package/AGENTS.md

@@ -41,3 +41,8 @@
 - 사용자 프롬프트/출력 내용을 원문 저장하거나 네트워크로 전송하는 로직 변경
 - localStorage/IndexedDB/세션 파일에 사용자 컨텍스트를 저장하는 로직 추가
 - 외부 CLI/도구 실행 경로를 하드코딩하거나, 권한 범위를 확장하는 동작 변경
+
+## UI Language Policy
+
+- User-facing UI labels, buttons, notices, dialogs, and system messages must always be displayed in English.
+- Do not introduce Korean (or other non-English) UI strings in code changes.

package/README.md

@@ -64,26 +64,25 @@ The backend runs `codex` as a local subprocess per request, streams JSONL events
 ## Install / Run
 
 ### Quick start (recommended for local development)
-Run:
+There are two development workflows:
+
+- `install_dev.sh` : install/link only (does not start JupyterLab)
+- `run_dev.sh` : install first, then start JupyterLab
+
+Install only:
 
 ```bash
-bash run_jupyterlab_codex.sh
+bash install_dev.sh
 ```
 
-The script will:
-1. install JS dependencies (`jlpm install`)
-2. build frontend (`jlpm build`)
-3. install Python package editable (`python -m pip install -e .`)
-4. install server config snippet and enable extension
-5. link labextension in `share/jupyter/labextensions/`
-6. launch `jupyter lab`
-
-You can pass JupyterLab options directly:
+Install + run:
 
 ```bash
-bash run_jupyterlab_codex.sh --ServerApp.port=8888
+bash run_dev.sh --ServerApp.port=8888
 ```
 
+`run_dev.sh` internally runs `install_dev.sh` first.
+
 ### Manual local install
 1. Build frontend
 
@@ -252,12 +251,25 @@ JupyterLab 4 우측 사이드바에서 Codex CLI(`codex exec --json`)를 채팅
 
 ## 설치/실행
 ### 빠른 실행(권장)
-아래 스크립트는 “개발/로컬 실행”에 필요한 과정을 한 번에 수행합니다.
+개발용 스크립트가 분리되어 있습니다.
+
+- `install_dev.sh` : 설치/링크만 수행 (`jupyter lab` 실행 없음)
+- `run_dev.sh` : 설치 후 JupyterLab 실행
+
+설치만:
 
 ```bash
-bash run_jupyterlab_codex.sh
+bash install_dev.sh
 ```
 
+설치 + 실행:
+
+```bash
+bash run_dev.sh --ServerApp.port=8888
+```
+
+`run_dev.sh`는 내부적으로 `install_dev.sh`를 먼저 실행한 뒤 JupyterLab을 시작합니다.
+
 스크립트가 하는 일(요약):
 - JS 의존성 설치(`jlpm install`) 및 빌드(`jlpm build`)
 - 파이썬 패키지 editable 설치(`python -m pip install -e .`)
@@ -268,7 +280,7 @@ bash run_jupyterlab_codex.sh
 추가로 JupyterLab 옵션을 넘기고 싶다면, 스크립트 뒤에 그대로 붙이면 됩니다.
 
 ```bash
-bash run_jupyterlab_codex.sh --ServerApp.port=8888
+bash run_dev.sh --ServerApp.port=8888
 ```
 
 ### 수동 설치(개발/로컬)

package/{run_jupyterlab_codex.sh → install_dev.sh}

@@ -9,10 +9,22 @@ set -euo pipefail
 ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 cd "$ROOT_DIR"
 
-echo "[0/6] Checking Python/Jupyter prerequisites"
+PYTHON_BIN="${PYTHON_BIN:-}"
+if [ -z "$PYTHON_BIN" ]; then
+  if command -v python >/dev/null 2>&1; then
+    PYTHON_BIN="python"
+  elif command -v python3 >/dev/null 2>&1; then
+    PYTHON_BIN="python3"
+  else
+    echo "ERROR: python not found on PATH (also tried python3)."
+    exit 1
+  fi
+fi
+
+echo "[1/5] Checking Python/Jupyter prerequisites"
 
-if ! command -v python >/dev/null 2>&1; then
-  echo "ERROR: python not found on PATH"
+if ! command -v "$PYTHON_BIN" >/dev/null 2>&1; then
+  echo "ERROR: $PYTHON_BIN not found on PATH"
   exit 1
 fi
 
@@ -21,9 +33,9 @@ if ! command -v node >/dev/null 2>&1; then
   exit 1
 fi
 
-if ! python -c 'import jupyterlab' >/dev/null 2>&1; then
+if ! "$PYTHON_BIN" -c 'import jupyterlab' >/dev/null 2>&1; then
   echo "JupyterLab not found in this Python environment. Installing (jupyterlab>=4,<5; jupyter_server>=2,<3)..."
-  python -m pip install -q "jupyterlab>=4,<5" "jupyter_server>=2,<3"
+  "$PYTHON_BIN" -m pip install -q "jupyterlab>=4,<5" "jupyter_server>=2,<3"
 fi
 
 JLPM="jlpm"
@@ -33,21 +45,22 @@ fi
 
 if ! command -v jupyter >/dev/null 2>&1; then
   echo "ERROR: jupyter command not found (expected after installing jupyterlab)."
-  echo "Try: python -m pip install \"jupyterlab>=4,<5\""
+  echo "Try: $PYTHON_BIN -m pip install \"jupyterlab>=4,<5\""
   exit 1
 fi
 
-echo "[1/6] Installing JS dependencies"
+echo "[2/5] Installing JS dependencies"
 $JLPM install
 
-echo "[2/6] Building JupyterLab extension"
+echo "[3/5] Building JupyterLab extension"
 $JLPM run build
 
-echo "[3/6] Installing Python package (editable)"
-python -m pip install -e "$ROOT_DIR" --no-deps
+echo "[4/5] Installing Python package (editable)"
+"$PYTHON_BIN" -m pip install -e "$ROOT_DIR" --no-deps
+
+echo "[5/5] Enabling server extension and linking labextension"
 
-echo "[4/6] Enabling server extension"
-PREFIX="${CONDA_PREFIX:-$(python -c 'import sys; print(sys.prefix)')}"
+PREFIX="${CONDA_PREFIX:-$("$PYTHON_BIN" -c 'import sys; print(sys.prefix)')}"
 
 JUPYTER_CFG_DIR="$PREFIX/etc/jupyter/jupyter_server_config.d"
 mkdir -p "$JUPYTER_CFG_DIR"
@@ -58,12 +71,5 @@ jupyter server extension list | sed -n '1,120p' || true
 
 LABEXT_DIR="$PREFIX/share/jupyter/labextensions"
 mkdir -p "$LABEXT_DIR"
-
-echo "[5/6] Linking labextension into $LABEXT_DIR"
 ln -sfn "$ROOT_DIR/jupyterlab_codex/labextension" "$LABEXT_DIR/jupyterlab-codex-sidebar"
-
-echo "[6/6] Current labextension status"
 jupyter labextension list
-
-echo "Starting JupyterLab..."
-exec jupyter lab --no-browser --ServerApp.open_browser=False "$@"

package/jupyterlab_codex/cli_defaults.py

@@ -28,9 +28,7 @@ def load_cli_defaults_for_ui() -> Dict[str, Any]:
     effective_model = env_model or (config_model.strip() if config_model else None)
     effective_reasoning = (config_reasoning or "").strip().lower() or None
 
-    # UI only supports these levels today.
-    if effective_reasoning not in {"low", "medium", "high", "xhigh"}:
-        effective_reasoning = None
+    effective_reasoning = effective_reasoning or None
 
     return {
         "model": effective_model,
@@ -107,4 +105,3 @@ def _parse_toml_scalar(value: str) -> Any:
         # Keep this simple; config values we care about are typically plain strings.
         return value[1:-1]
     return value
-

package/jupyterlab_codex/handlers.py

@@ -54,6 +54,20 @@ def _coerce_command_path(value: Any) -> str:
     return value.strip()
 
 
+def _coerce_session_context_key(value: Any) -> str:
+    if not isinstance(value, str):
+        return ""
+    return value.strip()
+
+
+def _coerce_bool_flag(value: Any) -> bool:
+    if isinstance(value, bool):
+        return value
+    if isinstance(value, str):
+        return value.strip().lower() in {"1", "true", "y", "yes", "on"}
+    return False
+
+
 def _build_command_not_found_hint(requested_path: str) -> dict[str, str]:
     requested_label = requested_path or "codex"
     detected = shutil.which("codex")
@@ -85,6 +99,7 @@ class CodexWSHandler(WebSocketHandler):
     def open(self):
         self.write_message(json.dumps({"type": "status", "state": "ready"}))
         self._send_cli_defaults()
+        self._send_model_catalog()
         self._send_rate_limits_snapshot()
 
     async def on_message(self, message: str):
@@ -104,6 +119,14 @@ class CodexWSHandler(WebSocketHandler):
             await self._handle_send(payload)
             return
 
+        if msg_type == "delete_session":
+            self._handle_delete_session(payload)
+            return
+
+        if msg_type == "delete_all_sessions":
+            self._handle_delete_all_sessions(payload)
+            return
+
         if msg_type == "cancel":
             await self._handle_cancel(payload)
             return
@@ -129,14 +152,60 @@ class CodexWSHandler(WebSocketHandler):
         except Exception:
             return
 
+    def _send_model_catalog(self) -> None:
+        async def _send() -> None:
+            models = await self._runner.list_available_models()
+            if not models:
+                return
+            try:
+                self.write_message(json.dumps({"type": "cli_defaults", "availableModels": models}))
+            except Exception:
+                return
+
+        try:
+            asyncio.create_task(_send())
+        except Exception:
+            return
+
     async def _handle_start_session(self, payload: Dict[str, Any]):
-        session_id = payload.get("sessionId") or str(uuid.uuid4())
+        requested_session_id = payload.get("sessionId") or ""
+        if not isinstance(requested_session_id, str):
+            requested_session_id = str(requested_session_id)
+        requested_session_id = requested_session_id.strip()
+        force_new_thread = _coerce_bool_flag(payload.get("forceNewThread"))
         notebook_path = payload.get("notebookPath", "")
+        if not isinstance(notebook_path, str):
+            notebook_path = str(notebook_path)
+        notebook_path = notebook_path.strip()
+        session_context_key = _coerce_session_context_key(payload.get("sessionContextKey"))
         notebook_os_path = self._resolve_notebook_os_path(notebook_path)
 
-        self._store.ensure_session(session_id, notebook_path, notebook_os_path)
+        resolved_session_id = requested_session_id
+        if force_new_thread:
+            previous_session_id = self._store.resolve_session_for_notebook(notebook_path, notebook_os_path)
+            if previous_session_id and previous_session_id != resolved_session_id:
+                self._store.delete_session(previous_session_id)
+            if not resolved_session_id:
+                resolved_session_id = str(uuid.uuid4())
+        else:
+            resolved_session_id = self._store.resolve_session_for_notebook(notebook_path, notebook_os_path) or resolved_session_id
+            if not resolved_session_id:
+                resolved_session_id = str(uuid.uuid4())
+
+        self._store.ensure_session(resolved_session_id, notebook_path, notebook_os_path)
         if notebook_path:
-            self._store.update_notebook_path(session_id, notebook_path, notebook_os_path)
+            self._store.update_notebook_path(resolved_session_id, notebook_path, notebook_os_path)
+
+        raw_history = self._store.load_messages(resolved_session_id)
+        history = []
+        for item in raw_history:
+            role = item.get("role")
+            content = item.get("content")
+            if role not in {"user", "assistant", "system"}:
+                continue
+            if not isinstance(content, str):
+                continue
+            history.append({"role": role, "content": content})
 
         paired_ok, paired_path, paired_os_path, paired_message = _compute_pairing_status(
             notebook_path, notebook_os_path
@@ -146,8 +215,10 @@ class CodexWSHandler(WebSocketHandler):
                 {
                     "type": "status",
                     "state": "ready",
-                    "sessionId": session_id,
+                    "sessionId": resolved_session_id,
                     "notebookPath": notebook_path,
+                    "sessionContextKey": session_context_key,
+                    "history": history,
                     "pairedOk": paired_ok,
                     "pairedPath": paired_path,
                     "pairedOsPath": paired_os_path,
@@ -157,8 +228,12 @@ class CodexWSHandler(WebSocketHandler):
         )
 
     async def _handle_send(self, payload: Dict[str, Any]):
-        session_id = payload.get("sessionId") or str(uuid.uuid4())
+        session_id = payload.get("sessionId") or ""
+        if not isinstance(session_id, str):
+            session_id = str(session_id)
+        session_id = session_id.strip() or str(uuid.uuid4())
         content = payload.get("content", "")
+        session_context_key = _coerce_session_context_key(payload.get("sessionContextKey"))
         selection = payload.get("selection", "")
         cell_output = payload.get("cellOutput", "")
         images_payload = payload.get("images")
@@ -217,9 +292,9 @@ class CodexWSHandler(WebSocketHandler):
                         "type": "error",
                         "runId": run_id,
                         "sessionId": session_id,
+                        "sessionContextKey": session_context_key,
                         "notebookPath": notebook_path,
-                        "message": paired_message
-                        or "Jupytext paired file is required for this extension.",
+                        "message": paired_message or "Jupytext paired file is required for this extension.",
                         "pairedOk": paired_ok,
                         "pairedPath": paired_path,
                         "pairedOsPath": paired_os_path,
@@ -234,6 +309,7 @@ class CodexWSHandler(WebSocketHandler):
                         "state": "ready",
                         "runId": run_id,
                         "sessionId": session_id,
+                        "sessionContextKey": session_context_key,
                         "notebookPath": notebook_path,
                         "pairedOk": paired_ok,
                         "pairedPath": paired_path,
@@ -267,6 +343,7 @@ class CodexWSHandler(WebSocketHandler):
                         "state": "running",
                         "runId": run_id,
                         "sessionId": session_id,
+                        "sessionContextKey": session_context_key,
                         "notebookPath": notebook_path,
                         "pairedOk": paired_ok,
                         "pairedPath": paired_path,
@@ -290,6 +367,7 @@ class CodexWSHandler(WebSocketHandler):
                                 "type": "output",
                                 "runId": run_id,
                                 "sessionId": session_id,
+                                "sessionContextKey": session_context_key,
                                 "notebookPath": notebook_path,
                                 "text": text,
                             }
@@ -306,6 +384,7 @@ class CodexWSHandler(WebSocketHandler):
                                 "type": "event",
                                 "runId": run_id,
                                 "sessionId": session_id,
+                                "sessionContextKey": session_context_key,
                                 "notebookPath": notebook_path,
                                 "payload": event,
                             }
@@ -348,6 +427,7 @@ class CodexWSHandler(WebSocketHandler):
                             "type": "done",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "exitCode": exit_code,
                             "fileChanged": file_changed,
@@ -365,6 +445,7 @@ class CodexWSHandler(WebSocketHandler):
                             "state": "ready",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "pairedOk": paired_ok,
                             "pairedPath": paired_path,
@@ -381,6 +462,7 @@ class CodexWSHandler(WebSocketHandler):
                             "type": "done",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "exitCode": None,
                             "cancelled": True,
@@ -399,6 +481,7 @@ class CodexWSHandler(WebSocketHandler):
                             "state": "ready",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "pairedOk": paired_ok,
                             "pairedPath": paired_path,
@@ -414,6 +497,7 @@ class CodexWSHandler(WebSocketHandler):
                     "type": "error",
                     "runId": run_id,
                     "sessionId": session_id,
+                    "sessionContextKey": session_context_key,
                     "notebookPath": notebook_path,
                     "message": hint["message"],
                     "pairedOk": paired_ok,
@@ -431,6 +515,7 @@ class CodexWSHandler(WebSocketHandler):
                             "state": "ready",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "pairedOk": paired_ok,
                             "pairedPath": paired_path,
@@ -446,6 +531,7 @@ class CodexWSHandler(WebSocketHandler):
                             "type": "error",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "message": str(exc),
                         }
@@ -458,6 +544,7 @@ class CodexWSHandler(WebSocketHandler):
                             "state": "ready",
                             "runId": run_id,
                             "sessionId": session_id,
+                            "sessionContextKey": session_context_key,
                             "notebookPath": notebook_path,
                             "pairedOk": paired_ok,
                             "pairedPath": paired_path,
@@ -479,8 +566,44 @@ class CodexWSHandler(WebSocketHandler):
             "task": task,
             "sessionId": session_id,
             "notebookPath": notebook_path,
+            "sessionContextKey": session_context_key,
         }
 
+    def _handle_delete_session(self, payload: Dict[str, Any]) -> None:
+        session_id = payload.get("sessionId")
+        if not isinstance(session_id, str):
+            session_id = str(session_id) if session_id else ""
+        session_id = session_id.strip()
+        if session_id:
+            self._store.delete_session(session_id)
+
+    def _handle_delete_all_sessions(self, payload: Dict[str, Any]) -> None:
+        del payload
+        response: Dict[str, Any] = {
+            "type": "delete_all_sessions",
+            "ok": False,
+            "deletedCount": 0,
+            "failedCount": 0,
+            "message": "Unknown error"
+        }
+        try:
+            deleted_count, failed_count = self._store.delete_all_sessions()
+            response["deletedCount"] = deleted_count
+            response["failedCount"] = failed_count
+            response["ok"] = failed_count == 0
+            response["message"] = (
+                f"Deleted {deleted_count} conversations" if deleted_count else "No conversations found to delete"
+            )
+            if failed_count:
+                response["message"] = f"Deleted {deleted_count} conversations, failed to delete {failed_count}"
+        except Exception as exc:  # pragma: no cover - defensive path
+            response["message"] = str(exc)
+
+        try:
+            self.write_message(json.dumps(response))
+        except Exception:
+            return
+
     def _send_rate_limits_snapshot(self, force: bool = False) -> None:
         try:
             snapshot = load_latest_rate_limits(force=force)
@@ -503,6 +626,7 @@ class CodexWSHandler(WebSocketHandler):
 
         task = run_context["task"]
         task.cancel()
+        session_context_key = _coerce_session_context_key(run_context.get("sessionContextKey"))
         self.write_message(
             json.dumps(
                 {
@@ -510,6 +634,7 @@ class CodexWSHandler(WebSocketHandler):
                     "state": "ready",
                     "runId": run_id,
                     "sessionId": run_context["sessionId"],
+                    "sessionContextKey": session_context_key,
                     "notebookPath": run_context["notebookPath"],
                 }
             )
@@ -820,7 +945,7 @@ def _sanitize_reasoning_effort(value: Any) -> str | None:
     effort = value.strip().lower()
     if not effort:
         return None
-    if effort not in {"none", "minimal", "low", "medium", "high", "xhigh"}:
+    if not re.fullmatch(r"[a-z][a-z0-9._-]*", effort):
         return None
 
     return effort

package/jupyterlab_codex/labextension/package.json

@@ -1,6 +1,6 @@
 {
   "name": "jupyterlab-codex-sidebar",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "author": "ILHO AHN",
   "description": "Codex sidebar for JupyterLab",
   "license": "BSD-3-Clause",
@@ -37,7 +37,9 @@
     "@lumino/widgets": "^2.0.0",
     "dompurify": "^3.3.1",
     "highlight.js": "^11.11.1",
+    "katex": "^0.16.28",
     "marked": "^16.4.2",
+    "marked-katex-extension": "^5.1.7",
     "react": "^18.2.0",
     "react-dom": "^18.2.0"
   },
@@ -52,7 +54,7 @@
     "extension": true,
     "outputDir": "jupyterlab_codex/labextension",
     "_build": {
-      "load": "static/remoteEntry.00c43c61b0b2e7773f34.js",
+      "load": "static/remoteEntry.b2fdc03a1c4582e79156.js",
       "extension": "./extension"
     }
   }

package/jupyterlab_codex/labextension/static/124.ac0734653995d3fe00f2.js

@@ -0,0 +1 @@
+"use strict";(self.webpackChunkjupyterlab_codex_sidebar=self.webpackChunkjupyterlab_codex_sidebar||[]).push([[124],{8124(e,n,t){t.r(n),t.d(n,{default:()=>s});var r=t(4502);const a=/^(\${1,2})(?!\$)((?:\\.|[^\\\n])*?(?:\\.|[^\\\n\$]))\1(?=[\s?!\.,:？！。，：]|$)/,i=/^(\${1,2})(?!\$)((?:\\.|[^\\\n])*?(?:\\.|[^\\\n\$]))\1/,l=/^(\${1,2})\n((?:\\[^]|[^\\])+?)\n\1(?:\n|$)/;function s(e={}){return{extensions:[d(e,o(e,!1)),(n=o(e,!0),{name:"blockKatex",level:"block",tokenizer(e,n){const t=e.match(l);if(t)return{type:"blockKatex",raw:t[0],text:t[2].trim(),displayMode:2===t[1].length}},renderer:n})]};var n}function o(e,n){return t=>r.default.renderToString(t.text,{...e,displayMode:t.displayMode})+(n?"\n":"")}function d(e,n){const t=e&&e.nonStandard,r=t?i:a;return{name:"inlineKatex",level:"inline",start(e){let n,a=e;for(;a;){if(n=a.indexOf("$"),-1===n)return;if((t?n>-1:0===n||" "===a.charAt(n-1))&&a.substring(n).match(r))return n;a=a.substring(n+1).replace(/^\$+/,"")}},tokenizer(e,n){const t=e.match(r);if(t)return{type:"inlineKatex",raw:t[0],text:t[2].trim(),displayMode:2===t[1].length}},renderer:n}}}}]);