nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-04-27 12:12:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Update agent stream processing to support 'v2' chunk format and prevent emitting empty content.

Browse Source
This commit is contained in:
jxxghp
2026-03-23 17:23:44 +08:00
parent 565e10b6a5
commit 9b00a5f3f1
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/agent/__init__.py
View File

@@ -166,12 +166,16 @@ class MoviePilotAgent:
async for chunk in agent.astream( async for chunk in agent.astream(
{"messages": messages}, {"messages": messages},
stream_mode="messages", stream_mode="messages",
config=agent_config config=agent_config,
version="v2"
): ):
token, metadata = chunk
# 处理流式token（过滤工具调用token，只保留模型生成的内容） # 处理流式token（过滤工具调用token，只保留模型生成的内容）
if token and hasattr(token, "tool_call_chunks") and not token.tool_call_chunks: if chunk["type"] == "messages":
self.stream_handler.emit(token.content) token, metadata = chunk["data"]
if (token and hasattr(token, "tool_call_chunks")
and not token.tool_call_chunks):
if token.content:
self.stream_handler.emit(token.content)
# 发送最终消息给用户 # 发送最终消息给用户
await self.send_agent_message( await self.send_agent_message(