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