INNER CODE UNIT · Python

_load_prompt

invergent-ai/surogate · surogate/serve/tools/reference/qwen3_5/cli.py:132

def _load_prompt(frontend: Frontend, args: argparse.Namespace) -> PromptInput:
    if args.ids is not None:
        try:
            token_ids = parse_ids(args.ids)
        except ValueError as exc:
            raise ValueError(f"invalid --ids: {exc}") from exc
        batch = None
        rendered = frontend.decode(token_ids, skip_special_tokens=False)
        source = args.ids
    else:
        if args.prompt is not None:
            batch = frontend.process_text(args.prompt, thinking=args.thinking)
            source = args.prompt
        else:
            batch = frontend.process(load_messages(args.messages), thinking=args.thinking)
            source = args.messages
        token_ids = batch.input_ids.tolist()
        rendered = frontend.decode(token_ids, skip_special_tokens=False)

View source record →