Context Doctor plugin thumbnail

Context Doctor

Repair and beautify model JSON responses before they reach history, WebUI log, and tool execution using json_repair.

Author linkliti 6 stars Version 1.2.0 Updated
LlmAgentsMonologue
GitHub Repository Discussion How to install

README

Context Doctor

Repairs malformed JSON in model responses with json_repair, then beautifies output before it reaches history, log, and tool execution.

How it works

DirtyJson runs first (native A0). Context Doctor runs second: json_repair fixes what DirtyJson couldn't, validates against A0 schema (tool_name + tool_args required), outputs clean JSON.

Three hooks modify in-place, nothing appended:

  • response_stream_end - log item content, kvps, heading
  • hist_add_ai_response/end - history message, llm_result.response, log item
  • tool_execute_before - tool_args from full repaired response

Fallbacks

Input Output
Malformed A0 tool call Repaired + beautified JSON
Plain text with XML tags {} (history only, log untouched)
Plain text without XML {"thoughts": [raw]}

json_repair runs first. Fallbacks only apply if it returns nothing usable.

Settings

Experimental

Setting Default What it does
minify false Compact output, no whitespace

Debug

Setting Default What it does
use_standard_mode false Standard mode, no schema. Off = salvage mode with A0 schema.
update_log false Put repaired JSON in View Details. Kvps and heading always update.
suppress_xml true Replace XML-corrupted tool calls with {}. Disable to treat as raw text.

Structure

context_doctor/
├── plugin.yaml
├── default_config.yaml
├── helpers/context_doctor.py          # repair_and_beautify, transform_response, build_heading, update_log_item, schema
├── vendor/json_repair/                # bundled v0.63.3, no pip
├── extensions/python/
│   ├── response_stream_end/_20_context_doctor.py
│   ├── _functions/agent/Agent/hist_add_ai_response/end/_05_store_last_response.py
│   ├── _functions/agent/Agent/hist_add_ai_response/end/_20_context_doctor.py
│   ├── _functions/agent/Agent/hist_add_warning/start/_10_suppress_empty_response_warnings.py
│   └── tool_execute_before/_30_context_doctor.py
├── webui/config.html
├── tests/                             # unit + integration + e2e (92 tests)
│   └── vendor/                        # upstream json_repair suite (302 tests)

Install

Copy to /a0/usr/plugins/. Toggle on in WebUI.

Test

cd /a0/usr/plugins/context_doctor
/opt/venv-a0/bin/python -m pytest tests/ -q