Release v0.5.0 proxy reliability updates

Author: HaShiShark

.gitignore

@@ -1,4 +1,5 @@
 node_modules/
+.claude/
 release/
 python_dist/
 .venv/

codex_context.py

@@ -0,0 +1,44 @@
+from __future__ import annotations
+
+from typing import Any
+
+CONTEXT_CONTROL_COMMANDS = {"ctx", "/ctx", "context", "/context"}
+
+
+def text_value(value: Any) -> str:
+    return str(value or "")
+
+
+def is_context_control_command_text(text: Any) -> bool:
+    return text_value(text).strip().lower() in CONTEXT_CONTROL_COMMANDS
+
+
+def is_contextual_user_text(text: Any) -> bool:
+    trimmed = text_value(text).lstrip()
+    lowered = trimmed.lower()
+    return (
+        trimmed.startswith("# AGENTS.md instructions for ")
+        or lowered.startswith("<environment_context>")
+        or lowered.startswith("<skills>")
+        or lowered.startswith("<user_shell_command>")
+        or lowered.startswith("<turn_aborted>")
+        or lowered.startswith("<subagent_notification>")
+    )
+
+
+def is_conversation_record(record: dict[str, Any]) -> bool:
+    role = text_value(record.get("role")).strip()
+    if role == "assistant":
+        return True
+    if role != "user":
+        return False
+    text = text_value(record.get("text"))
+    return (
+        bool(text.strip())
+        and not is_contextual_user_text(text)
+        and not is_context_control_command_text(text)
+    )
+
+
+def conversation_record_count(transcript: list[dict[str, Any]]) -> int:
+    return sum(1 for record in transcript if is_conversation_record(record))

electron/context-window.cjs

@@ -4,7 +4,7 @@ const fs = require('node:fs');
 const http = require('node:http');
 const path = require('node:path');
 
-const HOST = '127.0.0.1';
+const HOST = process.env.HASH_CONTEXT_HOST || 'localhost';
 const BACKEND_PORT = 8765;
 const FRONTEND_PORT = 5174;
 const PROXY_PORT = 8787;
@@ -228,7 +228,7 @@ async function startBackend(root) {
     writeLog(`[backend:error] ${chunk.toString().trim()}`);
   });
 
-  await waitFor(BACKEND_PORT, '/api/init', 'Backend');
+  await waitFor(BACKEND_PORT, '/api/init', 'Backend', 90000);
   writeLog('backend ready');
 }
 

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "hash-context-codex-lab",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "private": true,
   "description": "Local context proxy and workbench companion for Codex.",
   "author": "",
@@ -69,6 +69,7 @@
     },
     "files": [
       "electron/**/*",
+      "codex_context.py",
       "web_server.py",
       "main.py",
       "hash.html",

package.json

@@ -18,8 +18,14 @@
 from pathlib import Path
 from typing import Any
 
+from codex_context import (
+    conversation_record_count,
+    is_context_control_command_text,
+    is_contextual_user_text,
+)
+
 
-HOST = os.environ.get("HASH_CONTEXT_PROXY_HOST", "127.0.0.1")
+HOST = os.environ.get("HASH_CONTEXT_PROXY_HOST", os.environ.get("HASH_CONTEXT_HOST", "localhost"))
 PORT = int(os.environ.get("HASH_CONTEXT_PROXY_PORT", "8787"))
 OPENAI_UPSTREAM_BASE_URL = os.environ.get(
     "HASH_CONTEXT_OPENAI_UPSTREAM_BASE_URL",
@@ -36,7 +42,6 @@
 CODEX_PROXY_BASE_URL = f"http://{HOST}:{PORT}/v1"
 INTERNAL_CONTEXT_HEADER = "x-hash-context-internal"
 INTERNAL_CONTEXT_VALUE = "context-workbench"
-CONTEXT_CONTROL_COMMANDS = {"ctx", "/ctx", "context", "/context"}
 CONTEXT_CONTROL_NOTICE_TEXT = "Hash Context: opened workbench."
 CONTROL_PORT = int(os.environ.get("HASH_CONTEXT_CONTROL_PORT", "8790"))
 LOCAL_COMPACT_PROMPT_PREFIX = "You are performing a CONTEXT CHECKPOINT COMPACTION."
@@ -507,10 +512,6 @@ def flush_assistant() -> None:
     return records
 
 
-def is_context_control_command_text(text: str) -> bool:
-    return str(text or "").strip().lower() in CONTEXT_CONTROL_COMMANDS
-
-
 def context_control_command_from_input(input_items: Any) -> str:
     transcript = input_items_to_transcript(input_items)
     for record in reversed(transcript):
@@ -938,7 +939,7 @@ def open_context_workbench(session_id: str) -> tuple[bool, str]:
     path = "/show"
     if session_id:
         path = f"{path}?session_id={urllib.parse.quote(session_id, safe='')}"
-    conn = http.client.HTTPConnection("127.0.0.1", CONTROL_PORT, timeout=2)
+    conn = http.client.HTTPConnection(os.environ.get("HASH_CONTEXT_HOST", "localhost"), CONTROL_PORT, timeout=2)
     try:
         conn.request("POST", path, body=b"", headers={"Content-Length": "0"})
         response = conn.getresponse()
@@ -1115,19 +1116,6 @@ def is_local_compact_summary_text(text: str) -> bool:
     return str(text or "").startswith(f"{LOCAL_COMPACT_SUMMARY_PREFIX}\n\n")
 
 
-def is_contextual_user_text(text: str) -> bool:
-    trimmed = str(text or "").lstrip()
-    lowered = trimmed.lower()
-    return (
-        trimmed.startswith("# AGENTS.md instructions for ")
-        or lowered.startswith("<environment_context>")
-        or lowered.startswith("<skills>")
-        or lowered.startswith("<user_shell_command>")
-        or lowered.startswith("<turn_aborted>")
-        or lowered.startswith("<subagent_notification>")
-    )
-
-
 def is_initial_context_prefix_record(record: dict[str, Any]) -> bool:
     role = str(record.get("role") or "").strip()
     if role in {"system", "developer"}:
@@ -1161,6 +1149,13 @@ def with_fresh_initial_context_prefix(
     return clean_transcript([*prefix, *body])
 
 
+def should_replace_transcript_from_control_intercept(
+    existing_transcript: list[dict[str, Any]],
+    candidate_transcript: list[dict[str, Any]],
+) -> bool:
+    return conversation_record_count(candidate_transcript) >= conversation_record_count(existing_transcript)
+
+
 def local_compact_source_from_transcript(transcript: list[dict[str, Any]]) -> list[dict[str, Any]] | None:
     records = clean_transcript(transcript)
     if not records:
@@ -1410,8 +1405,12 @@ def record_control_intercept(self, session_id: str, body: dict[str, Any], header
             if session is None:
                 session = ProxySession(id=session_id, title=f"Codex {session_id[:8]}")
                 self.sessions[session_id] = session
-            source_transcript = strip_context_edit_notice_records(input_items_to_transcript(body.get("input")))
-            session.transcript = clean_transcript(source_transcript)
+            source_transcript = clean_transcript(
+                strip_context_edit_notice_records(input_items_to_transcript(body.get("input")))
+            )
+            existing_transcript = clean_transcript(session.transcript)
+            if should_replace_transcript_from_control_intercept(existing_transcript, source_transcript):
+                session.transcript = source_transcript
             session.pending_transcript = None
             session.local_compact_source_transcript = None
             session.status = "override" if session.edited_transcript is not None else "mirror"

proxy_server.py

@@ -26,6 +26,8 @@ import { normalizeConversation, normalizeReasoningOptions } from './utils';
 const DEMO_SESSION_ID = 'hash-context-preview';
 const LIVE_REFRESH_IDLE_MS = 1600;
 const LIVE_REFRESH_RUNNING_MS = 800;
+const PENDING_CONTEXT_REFRESH_MS = 200;
+const PENDING_CONTEXT_REFRESH_MAX_MS = 5000;
 const LOCAL_EDIT_GRACE_MS = 1500;
 const MIN_WORKBENCH_WINDOW_WIDTH = 760;
 const MIN_WORKBENCH_WINDOW_HEIGHT = 520;
@@ -55,6 +57,19 @@ function isProxyBusy(status?: string, isRunning?: boolean): boolean {
   return Boolean(isRunning || status === 'running' || status === 'compacting');
 }
 
+function hasPendingAssistantMessage(messages: MessageRecord[]): boolean {
+  const lastAssistant = [...messages].reverse().find((message) => message.role === 'an');
+  if (!lastAssistant) {
+    return false;
+  }
+
+  return Boolean(
+    lastAssistant.pending
+    || (!lastAssistant.text.trim() && lastAssistant.blocks.some((block) => block.kind === 'thinking'))
+    || lastAssistant.blocks.some((block) => block.kind === 'reasoning' && block.status === 'streaming'),
+  );
+}
+
 type LoadInitOptions = {
   silent?: boolean;
   targetSessionId?: string;
@@ -530,6 +545,28 @@ export default function WorkbenchWindow() {
   );
 
   const currentPendingRestore = sessionId ? pendingRestores[sessionId] || null : null;
+  const hasPendingAssistant = useMemo(() => hasPendingAssistantMessage(messages), [messages]);
+
+  useEffect(() => {
+    if (!hasPendingAssistant) {
+      return undefined;
+    }
+
+    const startedAt = Date.now();
+    const intervalId = window.setInterval(() => {
+      if (Date.now() - startedAt >= PENDING_CONTEXT_REFRESH_MAX_MS) {
+        window.clearInterval(intervalId);
+        return;
+      }
+      if (Date.now() - lastLocalEditAtRef.current < LOCAL_EDIT_GRACE_MS) {
+        return;
+      }
+
+      void loadInit({ silent: true });
+    }, PENDING_CONTEXT_REFRESH_MS);
+
+    return () => window.clearInterval(intervalId);
+  }, [hasPendingAssistant, loadInit]);
 
   const commitContextConversation = useCallback(
     async (

react_app/src/WorkbenchWindow.tsx

@@ -172,11 +172,12 @@ function areScrollMetricsEqual(left: ScrollMetrics, right: ScrollMetrics) {
 function canExpandMessage(record: MessageRecord, previewText: string, isPreviewTruncated: boolean) {
   void previewText;
   const textValue = record.text || '';
+  const trimmedTextValue = textValue.trim();
   const hasStructuredContent = Boolean(
     record.attachments.length
     || record.toolEvents.length
     || record.blocks.length > 1
-    || /\r?\n/.test(textValue),
+    || /\r?\n/.test(trimmedTextValue),
   );
 
   return hasStructuredContent || isPreviewTruncated;

react_app/src/components/ContextMapSidebar.tsx

@@ -22,11 +22,11 @@ export default defineConfig(({ command }) => ({
     },
     proxy: {
       '/api/proxy/sessions': {
-        target: 'http://127.0.0.1:8787',
+        target: 'http://localhost:8787',
         changeOrigin: true,
       },
       '/api': {
-        target: 'http://127.0.0.1:8765',
+        target: 'http://localhost:8765',
         changeOrigin: true,
       },
     },

react_app/vite.config.ts

@@ -8,6 +8,8 @@ try {
 } catch {
 }
 
+$loopbackHost = if ($env:HASH_CONTEXT_HOST) { $env:HASH_CONTEXT_HOST } else { "localhost" }
+
 function Write-HookJson {
   param(
     [hashtable] $Payload
@@ -134,7 +136,7 @@ function Sync-LocalCodexSession {
 
   try {
     $payload = @{ session_id = $SessionId; title = "Codex $($SessionId.Substring(0, 8))" } | ConvertTo-Json -Compress
-    Invoke-WebRequest -Uri "http://127.0.0.1:8765/api/codex-local-session-sync" -Method Post -Body $payload -ContentType "application/json" -UseBasicParsing -TimeoutSec 8 | Out-Null
+    Invoke-WebRequest -Uri "http://${loopbackHost}:8765/api/codex-local-session-sync" -Method Post -Body $payload -ContentType "application/json" -UseBasicParsing -TimeoutSec 8 | Out-Null
     Write-HookLog "local-session-sync ok session_id=$SessionId"
   } catch {
     Write-HookLog "local-session-sync failed session_id=$SessionId error=$($_.Exception.Message)"
@@ -169,7 +171,7 @@ function Write-BackgroundLog {
 `$sessionId = '$escapedSessionId'
 try {
   `$payload = @{ session_id = `$sessionId; title = "Codex `$(`$sessionId.Substring(0, 8))" } | ConvertTo-Json -Compress
-  Invoke-WebRequest -Uri "http://127.0.0.1:8765/api/codex-local-session-sync" -Method Post -Body `$payload -ContentType "application/json" -UseBasicParsing -TimeoutSec 20 | Out-Null
+  Invoke-WebRequest -Uri "http://${loopbackHost}:8765/api/codex-local-session-sync" -Method Post -Body `$payload -ContentType "application/json" -UseBasicParsing -TimeoutSec 20 | Out-Null
   Write-BackgroundLog "local-session-sync ok session_id=`$sessionId"
 } catch {
   Write-BackgroundLog "local-session-sync failed session_id=`$sessionId error=`$(`$_.Exception.Message)"
@@ -194,7 +196,7 @@ function Consume-ContextEditMarker {
 
   try {
     $payload = @{ session_id = $SessionId } | ConvertTo-Json -Compress
-    $response = Invoke-WebRequest -Uri "http://127.0.0.1:8765/api/context-edit-marker-consume" -Method Post -Body $payload -ContentType "application/json; charset=utf-8" -UseBasicParsing -TimeoutSec 2
+    $response = Invoke-WebRequest -Uri "http://${loopbackHost}:8765/api/context-edit-marker-consume" -Method Post -Body $payload -ContentType "application/json; charset=utf-8" -UseBasicParsing -TimeoutSec 2
     $body = $response.Content | ConvertFrom-Json
     if ($body -and $body.marker) {
       Write-HookLog "context-edit-marker consumed session_id=$SessionId marker=$($response.Content)"
@@ -262,7 +264,7 @@ try {
   if (-not $sessionId) {
     $sessionId = Find-LatestHistorySessionId -Prompt $prompt
   }
-  $showUrl = "http://127.0.0.1:$controlPort/show"
+  $showUrl = "http://${loopbackHost}:$controlPort/show"
   if ($sessionId) {
     $showUrl = "$showUrl`?session_id=$([uri]::EscapeDataString($sessionId))"
   }