feat: 更新agent应用入口和依赖

- 更新 agent/app/main.py 入口文件
- 更新 agent/requirements.txt 依赖

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-03-12 15:23:29 +08:00
parent 414147911a
commit 963666b8bb
2 changed files with 8 additions and 0 deletions

View File

@@ -127,6 +127,9 @@ class ChatRequest(BaseModel):
model_provider: Optional[str] = None model_provider: Optional[str] = None
api_key: Optional[str] = None api_key: Optional[str] = None
base_url: Optional[str] = None base_url: Optional[str] = None
# Embedding 模型(可选)
embedding_model: Optional[str] = None
embedding_base_url: Optional[str] = None
class TeamChatRequest(BaseModel): class TeamChatRequest(BaseModel):
@@ -254,6 +257,8 @@ async def chat(request: ChatRequest):
model=config.model_name, model=config.model_name,
api_key=request.api_key or config.api_key, api_key=request.api_key or config.api_key,
base_url=request.base_url or config.base_url, 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) result = await xbot.run(request.message, session_id)
response_content = result["content"] response_content = result["content"]
@@ -334,6 +339,8 @@ async def chat_stream(request: ChatRequest):
model=config.model_name, model=config.model_name,
api_key=request.api_key or config.api_key, api_key=request.api_key or config.api_key,
base_url=request.base_url or config.base_url, 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(): async def event_generator():

View File

@@ -8,3 +8,4 @@ aiohttp>=3.8.0
redis>=5.0.0 redis>=5.0.0
loguru>=0.7.0 loguru>=0.7.0
tiktoken>=0.12.0 tiktoken>=0.12.0
simplemem>=0.1.0