SupplyAgent/agents/demand_belong_category_agent/run.py

#!/usr/bin/env python3
"""Run demand_belong_category_agent."""

from agents.demand_belong_category_agent import create_demand_belong_category_agent


def main(word_list: list[str]) -> None:
    agent = create_demand_belong_category_agent()
    print(f"demand_belong_category_agent ready | model={agent.model}")
    print(f"tools: {agent.tools.list_tools()}")
    print()
    words_str = ",".join(word_list)
    user_input = f'''
    以下是待分类的词语 : {words_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(['1','2'])