This commit is contained in:
jxxghp
2025-10-18 21:32:46 +08:00
parent bdd5f1231e
commit ee71bafc96
2 changed files with 5 additions and 7 deletions
+3 -4
View File
@@ -231,7 +231,7 @@ class MoviePilotAgent:
# 执行Agent # 执行Agent
logger.info(f"Agent执行推理: session_id={self.session_id}, input={message}") logger.info(f"Agent执行推理: session_id={self.session_id}, input={message}")
result = await self._execute_agent(input_context) await self._execute_agent(input_context)
# 获取Agent回复 # 获取Agent回复
agent_message = await self.callback_handler.get_message() agent_message = await self.callback_handler.get_message()
@@ -239,8 +239,7 @@ class MoviePilotAgent:
# 发送Agent回复给用户 # 发送Agent回复给用户
self.message_helper.put( self.message_helper.put(
message=agent_message, message=agent_message,
role="system", role="system"
title="AI助手回复"
) )
# 添加Agent回复到记忆 # 添加Agent回复到记忆
@@ -260,7 +259,7 @@ class MoviePilotAgent:
self.message_helper.put( self.message_helper.put(
message=error_message, message=error_message,
role="system", role="system",
title="AI助手错误" title="MoviePilot助手错误"
) )
return error_message return error_message
+2 -3
View File
@@ -26,13 +26,12 @@ class MoviePilotTool(BaseTool):
async def _arun(self, **kwargs) -> str: async def _arun(self, **kwargs) -> str:
raise NotImplementedError raise NotImplementedError
def _send_tool_message(self, message: str, message_type: str = "info"): def _send_tool_message(self, message: str, **kwargs):
"""发送工具执行消息""" """发送工具执行消息"""
try: try:
self._message_helper.put( self._message_helper.put(
message=message, message=message,
role="system", role="system"
title=f"AI工具执行 ({message_type})"
) )
except Exception as e: except Exception as e:
logger.error(f"发送工具消息失败: {e}") logger.error(f"发送工具消息失败: {e}")