游戏时间:运行完整代理
您的篮球教练代理几乎可以上场了。您已经编写了一个回调函数,用于显示步骤编号和 token 用量。
现在是把一切组装起来的时候:您将配置一个使用该回调的 CodeAgent。
您已经可以使用:
- 一个
basketball_tool,让代理能够搜索球探报告 - 一个语言模型变量
model - 一个回调函数:
action_callback
您的目标是完成代理的连线与配置。
本练习是课程的一部分
使用 Hugging Face smolagents 的 AI Agents
练习说明
- 导入
ActionStep类,以便用于注册动作回调。 - 为
ActionStep类注册action_callback函数,使代理在执行时显示其动作。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Import ActionStep to register a callback for it
from smolagents import ____
coach_agent = CodeAgent(
tools=[basketball_tool],
model=model,
verbosity_level=0,
# Register a callback that runs when an ActionStep is triggered
step_callbacks={ActionStep: ____}
)
# Run a question through the agent
result = coach_agent.run("What defensive strategy should we use to stop their point guard who averages 25 points per game?")
print(result)