重构七天训练营老师

image_agent/agent.py

@@ -0,0 +1,33 @@
+from agent_base.agent_base import HuiYuBaseAgent
+from common.models import minimax_model
+from .image_tools import search_images, get_image_detail, list_images
+
+# ============================================================
+# 定义 Agent
+# ============================================================
+image_agent = HuiYuBaseAgent(
+    name="image_agent",
+    model=minimax_model,
+    description="图片查询助手，能够搜索图片库、查看图片详情和图片列表。",
+    instruction=(
+        "你是一个图片查询助手，帮助用户在图片知识库中查找需要的图片。\n\n"
+        "你的能力包括：\n"
+        "1. 搜索图片：根据用户的自然语言描述，用 AI 语义搜索匹配的图片。"
+        "当用户描述一个场景、主题或关键词时，调用 search_images 搜索。\n"
+        "2. 查看图片详情：当用户想了解某张图片的完整 OCR 内容时，"
+        "调用 get_image_detail 获取详情。\n"
+        "3. 浏览图片列表：当用户想看图片库有哪些图片时，"
+        "调用 list_images 获取列表。\n\n"
+        "使用规则：\n"
+        "- 用户问有没有关于某主题的图片时，用 search_images 搜索\n"
+        "- 调用 search_images 时，如果用户说了要几张图片，就把用户说的数量传给 limit 参数；"
+        "如果用户没说，默认传 5\n"
+        "- 用户想看某张图片的完整内容时，用 get_image_detail 查详情\n"
+        "- 用户想浏览图片库时，用 list_images 列表\n"
+        "- 搜索结果返回的是预览，如果用户想看完整内容，引导用户用 get_image_detail\n"
+        "- 回复时使用中文"
+    ),
+    tools=[search_images, get_image_detail, list_images],
+)
+
+root_agent = image_agent