SupplyAgent/agents/demand_grade_agent/run.py

#!/usr/bin/env python3
"""Run demand_grade_agent on one batch of demand names.

批次选择（哪些词、多少个）由调用方（通常是调度任务）负责，本入口只处理
传入的这一批，不做分页/自行遍历需求池。
"""
from __future__ import annotations

from agents.demand_grade_agent import create_demand_grade_agent


def main(demand_names: list[str], biz_dt: str | None = None) -> None:
    agent = create_demand_grade_agent()
    print(f"demand_grade_agent ready | model={agent.model}")
    print(f"tools: {agent.tools.list_tools()}")
    print()

    names_str = "\n".join(f"- {name}" for name in demand_names)
    biz_dt_note = f"biz_dt={biz_dt}" if biz_dt else "未指定 biz_dt，请先调用 query_latest_biz_dt() 确定"
    user_input = f"""
    请对以下 {len(demand_names)} 个需求词逐一评级（S/A/B/C/D），{biz_dt_note}。
    只处理这一批，不要尝试查找或列举更多需求词。判级完成后调用 batch_save_demand_grades 落库。

    需求词列表:
    {names_str}
    """
    result = agent.run(user_input)
    print(result.content)
    print(f"\n[iterations={result.iterations}, tool_calls={result.tool_calls_made}]")


if __name__ == "__main__":
    main(["因果报应", "降半旗"], biz_dt="20260714")