自定义输出格式

默认情况下，代理返回纯文本输出。但你可以定义一个结构化的输出格式，以便简化后续处理。

自定义输出格式

通过这个示例，你可以定义代理应返回的输出格式。

代码示例

  1. from pydantic import BaseModel
  3. # 定义输出格式，使用 Pydantic 模型
  4. class Post(BaseModel):
  5.     post_title: str
  6.     post_url: str
  7.     num_comments: int
  8.     hours_since_post: int
  11. class Posts(BaseModel):
  12.     posts: List[Post]
  15. controller = Controller(output_model=Posts)
  18. async def main():
  19.     task = '访问 Hacker News 的 Show HN 页面，并获取前 5 篇帖子'
  20.     model = ChatOpenAI(model='gpt-4o')
  21.     agent = Agent(task=task, llm=model, controller=controller)
  23.     history = await agent.run()
  25.     result = history.final_result()
  26.     if result:
  27.         parsed: Posts = Posts.model_validate_json(result)
  29.         for post in parsed.posts:
  30.             print('\n--------------------------------')
  31.             print(f'标题:            {post.post_title}')
  32.             print(f'链接:            {post.post_url}')
  33.             print(f'评论数:          {post.num_comments}')
  34.             print(f'发布时长 (小时): {post.hours_since_post}')
  35.     else:
  36.         print('无结果')
  39. if __name__ == '__main__':
  40.     asyncio.run(main())

说明

  • Post 类定义了每个帖子的结构，包括标题、链接、评论数和发布时间（小时数）。
  • Posts 类是 Post 的列表，用于存储多个帖子的信息。
  • Controller 通过 output_model=Posts 指定了自定义输出格式。
  • 在 main() 函数中，代理执行任务并返回结果，然后使用 Posts.model_validate_json(result) 解析 JSON 数据。
  • 最后，遍历解析后的数据并打印结果。