teleo/teleo-infrastructure
8
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions 1

Use provider-compatible Leo read schema

Browse source
This commit is contained in:
twentyOne2x 2026-07-16 08:53:53 +02:00
parent 9a77615e06
commit dcd31f5862
2 changed files with 20 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
scripts/leo_oos_readonly_guard.py
View file

@ -72,20 +72,6 @@ _ACTION_SPECS = {
READ_ONLY_BRIDGE_COMMANDS = frozenset(_ACTION_SPECS) READ_ONLY_BRIDGE_COMMANDS = frozenset(_ACTION_SPECS)
def _action_schema(action: str) -> dict[str, Any]:
spec = _ACTION_SPECS[action]
fields = (*spec["required"], *spec["optional"])
return {
"type": "object",
"properties": {
"action": {"type": "string", "enum": [action]},
**{field: _FIELD_SCHEMAS[field] for field in fields},
},
"required": ["action", *spec["required"]],
"additionalProperties": False,
}
STRUCTURED_READ_TOOL_SCHEMA = { STRUCTURED_READ_TOOL_SCHEMA = {
"name": STRUCTURED_READ_TOOL_NAME, "name": STRUCTURED_READ_TOOL_NAME,
"description": ( "description": (
@ -94,7 +80,19 @@ STRUCTURED_READ_TOOL_SCHEMA = {
), ),
"parameters": { "parameters": {
"type": "object", "type": "object",
"oneOf": [_action_schema(action) for action in sorted(_ACTION_SPECS)], "properties": {
"action": {
"type": "string",
"enum": sorted(_ACTION_SPECS),
"description": (
"Read operation. query is required for search, context, and search-proposals; claim_id is "
"required for show, evidence, and edges; proposal_id is required for show-proposal."
),
},
**_FIELD_SCHEMAS,
},
"required": ["action"],
"additionalProperties": False,
}, },
} }

15
tests/test_leo_oos_readonly_guard.py
View file

@ -78,15 +78,14 @@ def test_structured_read_rejects_writes_bad_types_irrelevant_fields_and_unbounde
guard.structured_read_argv(tool_args) guard.structured_read_argv(tool_args)
def test_schema_encodes_action_specific_required_fields() -> None: def test_schema_is_provider_compatible_while_handler_enforces_action_contracts() -> None:
branches = guard.STRUCTURED_READ_TOOL_SCHEMA["parameters"]["oneOf"] parameters = guard.STRUCTURED_READ_TOOL_SCHEMA["parameters"]
by_action = {branch["properties"]["action"]["enum"][0]: branch for branch in branches}
assert by_action["context"]["required"] == ["action", "query"] assert parameters["required"] == ["action"]
assert set(by_action["context"]["properties"]) == {"action", "query", "limit", "context_limit", "format"} assert parameters["additionalProperties"] is False
assert by_action["context"]["additionalProperties"] is False assert not ({"oneOf", "anyOf", "allOf"} & set(parameters))
assert by_action["show"]["required"] == ["action", "claim_id"] assert set(parameters["properties"]["action"]["enum"]) == set(guard.READ_ONLY_BRIDGE_COMMANDS)
assert by_action["status"]["required"] == ["action"] assert "query is required" in parameters["properties"]["action"]["description"]
class FakeProcess: class FakeProcess: