extension.py 1.2 KB

123456789101112131415161718192021
  1. """演示业务扩展:工具、委派描述和子 Agent 提示词集中声明。"""
  2. from tools.demo_tools import calculate, get_current_time, search_demo_knowledge
  3. def setup(registry):
  4. registry.register_agent(
  5. name="demo-searcher",
  6. description="资料检索专员。当任务需要在演示知识库里做多轮检索、比对多个来源、或整理摘要时委派给它。它只回结论与来源,不回原始文本。",
  7. prompt=(
  8. "你是资料检索专员。职责是在演示知识库里检索并把结果整理成结论。\n\n"
  9. "工作方式:\n"
  10. "- 用 search_demo_knowledge 检索,可换多组关键词多轮检索。\n"
  11. "- 需要数值计算时用 calculate,不要心算。\n"
  12. "- 检索不到就如实说「未检索到」,不要用常识补写。\n\n"
  13. "输出格式:\n"
  14. "- 先给一段不超过 5 句的结论。\n"
  15. "- 再列来源编号与标题。\n"
  16. "- 不要把检索到的原文整段复制回来,主线程不需要原始数据。"
  17. ),
  18. tools=[search_demo_knowledge, calculate, get_current_time],
  19. )