DemonGatanjieuandClaude Opus 5.5
1b03553946
A value of the right type under an unknown name may mean something
else (set_prompt {style: "anime"} is no prompt), so the correction now
takes only the other names light models were seen to write in the tool
exams (name/workflow_name, file/filename/image_name, text/positive,
model/id, node/id); any other name is answered with what the tool takes.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
38 lines
1.7 KiB
Python
38 lines
1.7 KiB
Python
"""Normalize an unambiguous argument name before approval, execution and task recording.
|
|
|
|
Only a known other name counts (ALIASES, names light models were seen to write in the tool exams,
|
|
docs/architecture/tool-exam.md): a value of the right type under any other name may mean something
|
|
else (set_prompt {style: "anime"} is not a prompt), and that call is answered with what the tool takes.
|
|
"""
|
|
|
|
ALIASES = { # needed argument -> other names that mean it
|
|
"workflow": {"name", "workflow_name"},
|
|
"image": {"file", "filename", "image_name"},
|
|
"prompt": {"text", "positive"},
|
|
"model_id": {"model", "id"},
|
|
"node_id": {"node", "id"},
|
|
}
|
|
JSON_TYPES = {"string": (str,), "integer": (int,), "number": (int, float), "boolean": (bool,), "array": (list,), "object": (dict,)}
|
|
|
|
|
|
def prepare_arguments(spec, arguments):
|
|
"""Return (arguments, (wrong, right) or None), without changing the caller's dictionary."""
|
|
unknown = set(arguments) - set(spec["properties"])
|
|
missing = [key for key in spec.get("required", []) if key not in arguments]
|
|
if len(unknown) != 1 or len(missing) != 1:
|
|
return arguments, None
|
|
wrong, right = next(iter(unknown)), missing[0]
|
|
if wrong not in ALIASES.get(right, ()):
|
|
return arguments, None
|
|
kinds = JSON_TYPES.get(spec["properties"][right].get("type"))
|
|
value = arguments[wrong]
|
|
if not kinds or not isinstance(value, kinds) or (isinstance(value, bool) and bool not in kinds):
|
|
return arguments, None
|
|
corrected = {key: value for key, value in arguments.items() if key != wrong}
|
|
corrected[right] = value
|
|
return corrected, (wrong, right)
|
|
|
|
|
|
def renaming_note(name, renamed):
|
|
return f"Took {renamed[0]} as {renamed[1]}: {name} calls it {renamed[1]}."
|