content-find-agent-new/content_agent/cli.py

from __future__ import annotations

import argparse
import json

from content_agent.run_service import RunService
from content_agent.schemas import RunStartRequest


def main() -> None:
    parser = argparse.ArgumentParser(description="Run the Content Agent V1 mock chain.")
    parser.add_argument("--trace-id", default=None)
    parser.add_argument("--source", default=None)
    parser.add_argument("--platform", default="douyin")
    args = parser.parse_args()

    service = RunService()
    state = service.start_run(
        RunStartRequest(trace_id=args.trace_id, source=args.source, platform=args.platform)
    )
    print(json.dumps(service.get_summary(state["trace_id"]), ensure_ascii=False, indent=2))


if __name__ == "__main__":
    main()