增加299老师分身agent
This commit is contained in:
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user