Windows robustness for claude/codex backends (+ hardened JSON fallback) (#79)
* Robustness for the claude/codex backends on Windows: argv overflow, subprocess encoding, tolerant JSON, test-eval dirs
Fixes surfaced running SkillOpt end-to-end on the bundled `claude` backend
(local Claude CLI) on Windows. None changes the OpenAI/GPT happy path.
1. skillopt/engine/trainer.py — the final test-eval directory
(test_eval_final/) is written to before being created; add
os.makedirs(..., exist_ok=True), matching the two sibling test-eval dirs.
Without it, summary.json raises FileNotFoundError when a rollout yields
zero predictions.
2. skillopt/model/claude_backend.py
a. Pass the prompt via stdin (not argv): on Windows the whole command line
is capped at ~32 KB and a large optimizer prompt (the success-analyst
minibatch carrying several report trajectories) overflows it with
[WinError 206], killing the run after retries.
b. Pass the system prompt via --append-system-prompt-file (a temp file),
not argv. The system prompt here is the skill being optimized, which
SkillOpt grows over training; since the ~32 KB cap applies to the SUM of
all argv, a grown skill would re-hit [WinError 206] even with the prompt
on stdin.
c. Pin the subprocess encoding to utf-8 (errors="replace"). With text=True
and no encoding=, stdin is encoded with the system codepage; on a zh-CN
box (cp936/GBK) a prompt containing an emoji or some Latin-1 characters
raises UnicodeEncodeError before the CLI even starts, failing every retry.
3. skillopt/model/codex_backend.py — the same utf-8 encoding pin on its
subprocess.run(input=...) call (identical unpinned-encoding pattern).
4. skillopt/utils/json_utils.py — extract_json() returned None for valid-
looking JSON that strict json.loads rejects (unescaped ASCII quotes inside
CJK string values, trailing commas), silently dropping the analyst's edits
on non-schema backends (Claude/Qwen): reflect produces N edits, 0 applied.
Add a json_repair fallback, but only on a single unambiguous object — a
balanced-brace extractor plus a refuse-on-multiple-objects guard — so a
chain-of-thought "scratch + final" response can't make repair silently
return the wrong (discarded) object, which would be worse than None (None is
detectable and retryable; a wrong-but-valid edit is applied blind). Declare
json_repair in requirements.txt and the claude/qwen optional extras so the
fallback is actually present (it otherwise no-ops, dropping edits silently).
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
(cherry picked from commit dca74a683e)
* fix(json_utils): harden tolerant JSON fallback from PR #77
Follow-up fixes on top of the cherry-picked Windows-robustness change:
1. Make _top_level_brace_objects() fully string-aware in its OUTER scan, not
just inside an object. A '{' inside quoted prose (e.g. '"set it to {x}"')
no longer starts a candidate object, so extract_json() returns None for
prose pseudo-JSON instead of repairing it into a bogus dict — which would
be strictly worse than dropping the edit, since extract_json feeds the
optimizer's skill edits.
2. Pick the repair candidate BEFORE importing json_repair, so the missing-
dependency RuntimeWarning only fires when there is genuinely a single
malformed object that could have been repaired. Ordinary no-JSON / prose
replies (the common case) now return None silently instead of warning on
every call.
3. Resolve dependency-metadata inconsistency: json_repair is optional, so add
it to the `all` extra (it was already in `claude`/`qwen`) and demote it
from a hard requirement to an optional/commented entry in requirements.txt,
matching the project's convention for backend-specific deps.
Adds regression tests for prose-with-braces (-> None), no-warning-on-plain-
text, single-object repair, and multi-object ambiguity. Existing 22 json
tests still pass with and without json_repair installed.
Co-Authored-By: Claude <noreply@anthropic.com>
---------
Co-authored-by: samuelgoofus-boop <260247789+samuelgoofus-boop@users.noreply.github.com>
Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -3,7 +3,11 @@ from __future__ import annotations
|
||||
|
||||
import pytest
|
||||
|
||||
from skillopt.utils.json_utils import extract_json, extract_json_array
|
||||
from skillopt.utils.json_utils import (
|
||||
_top_level_brace_objects,
|
||||
extract_json,
|
||||
extract_json_array,
|
||||
)
|
||||
|
||||
|
||||
class TestExtractJson:
|
||||
@@ -61,6 +65,52 @@ class TestExtractJson:
|
||||
assert extract_json(text) is None
|
||||
|
||||
|
||||
class TestTopLevelBraceObjects:
|
||||
"""_top_level_brace_objects — string/escape-aware top-level object scan."""
|
||||
|
||||
def test_single_clean_object(self) -> None:
|
||||
assert _top_level_brace_objects('{"a": 1}') == ['{"a": 1}']
|
||||
|
||||
def test_two_top_level_objects(self) -> None:
|
||||
assert _top_level_brace_objects('{"a":1}\n{"b":2}') == ['{"a":1}', '{"b":2}']
|
||||
|
||||
def test_brace_inside_quoted_prose_is_ignored(self) -> None:
|
||||
"""A '{' inside a quoted string must NOT start an object (the bug)."""
|
||||
# Brace-shaped content inside a string, with no real object → no spans.
|
||||
assert _top_level_brace_objects('label is "set it to {x: 1}" done') == []
|
||||
|
||||
def test_real_object_after_quoted_brace(self) -> None:
|
||||
"""Quoted-prose braces are skipped; a later real object is still found."""
|
||||
text = 'note "{wrong: 1}" then actual {"edit": "right"}'
|
||||
assert _top_level_brace_objects(text) == ['{"edit": "right"}']
|
||||
|
||||
|
||||
class TestExtractJsonTolerantFallback:
|
||||
"""extract_json — json_repair fallback for malformed non-OpenAI output."""
|
||||
|
||||
def test_prose_pseudo_json_returns_none(self) -> None:
|
||||
"""Regression: brace-shaped prose inside quotes must not be 'repaired'
|
||||
into a bogus dict. It returned {'op': 'delete'} before the fix."""
|
||||
text = 'The literal string "{op: delete}" appears in prose, not as JSON.'
|
||||
assert extract_json(text) is None
|
||||
|
||||
def test_no_warning_on_plain_text(self, recwarn: pytest.WarningsRecorder) -> None:
|
||||
"""No json_repair warning for ordinary no-JSON replies (no candidate)."""
|
||||
assert extract_json("Just plain text without JSON.") is None
|
||||
assert extract_json("") is None
|
||||
assert [w for w in recwarn.list if issubclass(w.category, RuntimeWarning)] == []
|
||||
|
||||
def test_trailing_comma_repaired_when_available(self) -> None:
|
||||
"""With json_repair installed, a single malformed object is repaired."""
|
||||
pytest.importorskip("json_repair")
|
||||
assert extract_json('{"edit": "add", "text": "x",}') == {"edit": "add", "text": "x"}
|
||||
|
||||
def test_two_malformed_objects_too_ambiguous(self) -> None:
|
||||
"""Multiple top-level objects are ambiguous → None, never guess."""
|
||||
pytest.importorskip("json_repair")
|
||||
assert extract_json('{"first": true,} noise {"second": true,}') is None
|
||||
|
||||
|
||||
class TestExtractJsonArray:
|
||||
"""extract_json_array — extract a JSON array from LLM response text."""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user