Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,63 @@
|
||||
"""Tests for _repair_tool_call_arguments — malformed JSON repair pipeline."""
|
||||
|
||||
import json
|
||||
|
||||
from run_agent import _repair_tool_call_arguments
|
||||
|
||||
|
||||
class TestRepairToolCallArguments:
|
||||
"""Verify each repair stage in the pipeline."""
|
||||
|
||||
# -- Stage 1: empty / whitespace-only --
|
||||
|
||||
def test_empty_string_returns_empty_object(self):
|
||||
assert _repair_tool_call_arguments("", "t") == "{}"
|
||||
|
||||
|
||||
|
||||
# -- Stage 2: Python None literal --
|
||||
|
||||
|
||||
|
||||
# -- Stage 3: trailing comma repair --
|
||||
|
||||
|
||||
def test_trailing_comma_in_array(self):
|
||||
result = _repair_tool_call_arguments('{"a": [1, 2,]}', "t")
|
||||
parsed = json.loads(result)
|
||||
assert parsed == {"a": [1, 2]}
|
||||
|
||||
|
||||
# -- Stage 4: unclosed brackets --
|
||||
|
||||
|
||||
|
||||
# -- Stage 5: excess closing delimiters --
|
||||
|
||||
|
||||
|
||||
# -- Stage 6: last resort --
|
||||
|
||||
|
||||
def test_unrepairable_partial_returns_empty_object(self):
|
||||
# Truncated in the middle of a string key — bracket closing won't help
|
||||
assert _repair_tool_call_arguments('{"truncated": "val', "t") == "{}"
|
||||
|
||||
# -- Valid JSON passthrough (this path is via except, but still works) --
|
||||
|
||||
|
||||
# -- Combined repairs --
|
||||
|
||||
|
||||
|
||||
# -- Stage 0: strict=False (literal control chars in strings) --
|
||||
# llama.cpp backends sometimes emit literal tabs/newlines inside JSON
|
||||
# string values. strict=False accepts these; we re-serialise to the
|
||||
# canonical wire form (#12068).
|
||||
|
||||
|
||||
|
||||
|
||||
# -- Stage 4: control-char escape fallback --
|
||||
|
||||
|
||||
Reference in New Issue
Block a user