luna/twitter-cli-cookiefile
1
0
Fork 0
Code Issues Pull Requests Actions Releases Activity

feat: wrap structured outputs in agent schema

Browse Source
This commit is contained in:
jackwener
2026-03-10 21:10:48 +08:00
parent 642ffe84a8
commit 9b7bdf3b06
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
twitter_cli/output.py
View File

@@ -55,12 +55,13 @@ def emit_structured(data: Any, *, as_json: bool, as_yaml: bool) -> bool:
fmt = default_structured_format(as_json=as_json, as_yaml=as_yaml) fmt = default_structured_format(as_json=as_json, as_yaml=as_yaml)
if not fmt: if not fmt:
return False return False
payload = _normalize_success_payload(data)
if fmt == "json": if fmt == "json":
click.echo(json.dumps(data, ensure_ascii=False, indent=2)) click.echo(json.dumps(payload, ensure_ascii=False, indent=2))
else: else:
click.echo( click.echo(
yaml.safe_dump( yaml.safe_dump(
data, payload,
allow_unicode=True, allow_unicode=True,
sort_keys=False, sort_keys=False,
default_flow_style=False, default_flow_style=False,
@@ -91,3 +92,10 @@ def error_payload(code: str, message: str, *, details: Any | None = None) -> dic
"schema_version": _SCHEMA_VERSION, "schema_version": _SCHEMA_VERSION,
"error": error, "error": error,
} }
def _normalize_success_payload(data: Any) -> Any:
"""Wrap plain structured data in the shared agent success schema."""
if isinstance(data, dict) and data.get("schema_version") == _SCHEMA_VERSION and "ok" in data:
return data
return success_payload(data)