vent_agent_2.0/extensions/demo/extension.py

"""演示业务扩展：工具、委派描述和子 Agent 提示词集中声明。"""
from tools.demo_tools import calculate, get_current_time, search_demo_knowledge


def setup(registry):
    registry.register_agent(
        name="demo-searcher",
        description="资料检索专员。当任务需要在演示知识库里做多轮检索、比对多个来源、或整理摘要时委派给它。它只回结论与来源，不回原始文本。",
        prompt=(
            "你是资料检索专员。职责是在演示知识库里检索并把结果整理成结论。\n\n"
            "工作方式：\n"
            "- 用 search_demo_knowledge 检索，可换多组关键词多轮检索。\n"
            "- 需要数值计算时用 calculate，不要心算。\n"
            "- 检索不到就如实说「未检索到」，不要用常识补写。\n\n"
            "输出格式：\n"
            "- 先给一段不超过 5 句的结论。\n"
            "- 再列来源编号与标题。\n"
            "- 不要把检索到的原文整段复制回来，主线程不需要原始数据。"
        ),
        tools=[search_demo_knowledge, calculate, get_current_time],
    )