fix agent bug

This commit is contained in:
jxxghp
2026-03-22 22:02:54 +08:00
parent e4451c7e6a
commit 5739ca7f97
+3 -2
View File
@@ -161,7 +161,8 @@ class MoviePilotAgent:
config=agent_config config=agent_config
): ):
token, metadata = chunk token, metadata = chunk
if token: # 处理流式token(过滤工具调用token,只保留模型生成的内容)
if token and hasattr(token, "tool_call_chunks") and not token.tool_call_chunks:
self.stream_handler.emit(token.content) self.stream_handler.emit(token.content)
# 发送最终消息给用户 # 发送最终消息给用户
@@ -173,7 +174,7 @@ class MoviePilotAgent:
memory_manager.save_agent_messages( memory_manager.save_agent_messages(
session_id=self.session_id, session_id=self.session_id,
user_id=self.user_id, user_id=self.user_id,
messages=agent.get_state(agent_config).values("messages") messages=agent.get_state(agent_config).values().get("messages", [])
) )
except asyncio.CancelledError: except asyncio.CancelledError: