wehub-resource-sync
fed8b2eed7
Backend release / release (push) Waiting to run
Bandit Security Scan / bandit_scan (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / manifest (push) Blocked by required conditions
Build and push DocsGPT FE Docker image for development / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / manifest (push) Blocked by required conditions
Python linting / ruff (push) Waiting to run
Run python tests with pytest / Run tests and count coverage (3.12) (push) Waiting to run
React Widget Build / build (push) Waiting to run
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
"""Engine robustness: fenced/prose structured output recovery in _parse_structured_output."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from application.agents.workflows.workflow_engine import WorkflowEngine
|
|
|
|
|
|
def _engine() -> WorkflowEngine:
|
|
return WorkflowEngine.__new__(WorkflowEngine)
|
|
|
|
|
|
def test_parse_structured_output_bare_json():
|
|
ok, val = _engine()._parse_structured_output('{"x": true}')
|
|
assert ok and val == {"x": True}
|
|
|
|
|
|
def test_parse_structured_output_json_fence():
|
|
ok, val = _engine()._parse_structured_output('```json\n{"a": 1, "b": null}\n```')
|
|
assert ok and val == {"a": 1, "b": None}
|
|
|
|
|
|
def test_parse_structured_output_unlabelled_fence():
|
|
ok, val = _engine()._parse_structured_output("text\n```\n{\"y\": [1, 2]}\n```\nmore")
|
|
assert ok and val == {"y": [1, 2]}
|
|
|
|
|
|
def test_parse_structured_output_prose_wrapped_object():
|
|
ok, val = _engine()._parse_structured_output('Result: {"z": "v"} done')
|
|
assert ok and val == {"z": "v"}
|
|
|
|
|
|
def test_parse_structured_output_non_json():
|
|
ok, val = _engine()._parse_structured_output("not json at all")
|
|
assert ok is False and val is None
|
|
|
|
|
|
def test_parse_structured_output_empty():
|
|
ok, val = _engine()._parse_structured_output(" ")
|
|
assert ok is False and val is None
|
|
|
|
|
|
def test_strip_json_fence_prefers_fence_over_braces():
|
|
out = WorkflowEngine._strip_json_fence('lead {"outer": 1} ```json\n{"in": 2}\n```')
|
|
assert out == '{"in": 2}'
|
|
|
|
|
|
def test_parse_structured_output_prose_wrapped_array():
|
|
ok, val = _engine()._parse_structured_output(
|
|
'Here are the results: [{"name": "a"}, {"name": "b"}]'
|
|
)
|
|
assert ok and val == [{"name": "a"}, {"name": "b"}]
|
|
|
|
|
|
def test_strip_json_fence_earliest_bracket_wins_for_array():
|
|
# A top-level array must not be mis-sliced from its first "{" to its last
|
|
# "}" (which would drop the array framing or extract an inner object).
|
|
assert WorkflowEngine._strip_json_fence('x [1, 2, {"k": 3}] y') == '[1, 2, {"k": 3}]'
|
|
|
|
|
|
def test_strip_json_fence_earliest_bracket_wins_for_object():
|
|
# An object that opens before any "[" still wins.
|
|
assert WorkflowEngine._strip_json_fence('x {"a": [1, 2]} y') == '{"a": [1, 2]}'
|