增加299老师分身agent

This commit is contained in:
2026-04-09 21:36:35 +08:00
parent 143b609924
commit ea2e30f568
7 changed files with 117 additions and 66 deletions

View File

@@ -1,25 +1,7 @@
import os
from agent_base.agent_base import HuiYuBaseAgent
from google.adk.models.lite_llm import LiteLlm
from common.models import minimax_model
from .note_formatter import NoteFormatter
# ============================================================
# MiniMax 模型配置(从 .env 文件读取)
# ============================================================
MINIMAX_API_KEY = os.getenv("MINIMAX_API_KEY")
MINIMAX_API_BASE = os.getenv("MINIMAX_API_BASE")
MINIMAX_MODEL = os.getenv("MINIMAX_MODEL")
# ============================================================
# 创建 LiteLlm 模型适配器
# ============================================================
model = LiteLlm(
model=MINIMAX_MODEL,
api_base=MINIMAX_API_BASE,
api_key=MINIMAX_API_KEY,
)
# ============================================================
# 笔记格式化器
# ============================================================
@@ -57,7 +39,7 @@ def generate_image_note(title: str, description: str) -> str:
# ============================================================
note_agent = HuiYuBaseAgent(
name="note_agent",
model=model,
model=minimax_model,
description="一个多模态笔记助手,能够将图片、文字等内容转换为结构化的 Markdown 笔记。",
instruction=(
"你是一个专业的笔记助手,帮助用户将各种内容整理成 Markdown 笔记。\n"