Spaces:
Running on Zero
Running on Zero
File size: 1,763 Bytes
6d5047c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | """CLI entrypoint for Qwen planner adapter."""
from __future__ import annotations
import argparse
import json
from typing import Sequence
from kimodo.planner import QwenPlannerAdapter
from kimodo.schemas import CharacterDefinition, PlannerRequest
def build_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(description="Generate planner scripts from a story prompt using Qwen.")
parser.add_argument("--scene-id", required=True, help="Scene identifier.")
parser.add_argument("--prompt", required=True, help="User story prompt.")
parser.add_argument(
"--character",
action="append",
required=True,
help="Character definition in form id[:skeleton[:description]]. Repeat for multiple characters.",
)
parser.add_argument("--duration-limit-sec", type=float, default=60.0)
return parser
def _parse_character_arg(raw: str) -> CharacterDefinition:
parts = raw.split(":", 2)
character_id = parts[0]
skeleton_type = parts[1] if len(parts) >= 2 and parts[1] else "soma"
description = parts[2] if len(parts) == 3 and parts[2] else None
return CharacterDefinition(character_id=character_id, skeleton_type=skeleton_type, description=description)
def main(argv: Sequence[str] | None = None) -> int:
args = build_parser().parse_args(argv)
request = PlannerRequest(
scene_id=args.scene_id,
user_prompt=args.prompt,
duration_limit_sec=args.duration_limit_sec,
characters=[_parse_character_arg(item) for item in args.character],
)
adapter = QwenPlannerAdapter()
response = adapter.plan(request)
print(json.dumps(response.model_dump(), indent=2))
return 0
if __name__ == "__main__":
raise SystemExit(main())
|