From 963666b8bb555efd98f0b80f844e91c181c5fcab Mon Sep 17 00:00:00 2001 From: "DESKTOP-72TV0V4\\caoxiaozhu" Date: Thu, 12 Mar 2026 15:23:29 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9B=B4=E6=96=B0agent=E5=BA=94?= =?UTF-8?q?=E7=94=A8=E5=85=A5=E5=8F=A3=E5=92=8C=E4=BE=9D=E8=B5=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 更新 agent/app/main.py 入口文件 - 更新 agent/requirements.txt 依赖 Co-Authored-By: Claude Opus 4.6 --- agent/app/main.py | 7 +++++++ agent/requirements.txt | 1 + 2 files changed, 8 insertions(+) diff --git a/agent/app/main.py b/agent/app/main.py index dffabce..e77dd41 100644 --- a/agent/app/main.py +++ b/agent/app/main.py @@ -127,6 +127,9 @@ class ChatRequest(BaseModel): model_provider: Optional[str] = None api_key: Optional[str] = None base_url: Optional[str] = None + # Embedding 模型（可选） + embedding_model: Optional[str] = None + embedding_base_url: Optional[str] = None class TeamChatRequest(BaseModel): @@ -254,6 +257,8 @@ async def chat(request: ChatRequest): model=config.model_name, api_key=request.api_key or config.api_key, base_url=request.base_url or config.base_url, + embedding_model=request.embedding_model, + embedding_base_url=request.embedding_base_url, ) result = await xbot.run(request.message, session_id) response_content = result["content"] @@ -334,6 +339,8 @@ async def chat_stream(request: ChatRequest): model=config.model_name, api_key=request.api_key or config.api_key, base_url=request.base_url or config.base_url, + embedding_model=request.embedding_model, + embedding_base_url=request.embedding_base_url, ) async def event_generator(): diff --git a/agent/requirements.txt b/agent/requirements.txt index ed14232..97f7f47 100644 --- a/agent/requirements.txt +++ b/agent/requirements.txt @@ -8,3 +8,4 @@ aiohttp>=3.8.0 redis>=5.0.0 loguru>=0.7.0 tiktoken>=0.12.0 +simplemem>=0.1.0