mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-05 23:47:41 +08:00
refactor: SendMessageTool message_type 改为消息标题
This commit is contained in:
@@ -20,8 +20,8 @@ class SendMessageInput(BaseModel):
|
|||||||
description="The message content to send to the user (should be clear and informative)",
|
description="The message content to send to the user (should be clear and informative)",
|
||||||
)
|
)
|
||||||
message_type: Optional[str] = Field(
|
message_type: Optional[str] = Field(
|
||||||
"info",
|
None,
|
||||||
description="Type of message: 'info' for general information, 'success' for successful operations, 'warning' for warnings, 'error' for error messages",
|
description="Title of the message, a short summary of the message content",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@@ -34,30 +34,23 @@ class SendMessageTool(MoviePilotTool):
|
|||||||
def get_tool_message(self, **kwargs) -> Optional[str]:
|
def get_tool_message(self, **kwargs) -> Optional[str]:
|
||||||
"""根据消息参数生成友好的提示消息"""
|
"""根据消息参数生成友好的提示消息"""
|
||||||
message = kwargs.get("message", "")
|
message = kwargs.get("message", "")
|
||||||
message_type = kwargs.get("message_type", "info")
|
title = kwargs.get("message_type") or ""
|
||||||
|
|
||||||
type_map = {
|
|
||||||
"info": "信息",
|
|
||||||
"success": "成功",
|
|
||||||
"warning": "警告",
|
|
||||||
"error": "错误",
|
|
||||||
}
|
|
||||||
type_desc = type_map.get(message_type, message_type)
|
|
||||||
|
|
||||||
# 截断过长的消息
|
# 截断过长的消息
|
||||||
if len(message) > 50:
|
if len(message) > 50:
|
||||||
message = message[:50] + "..."
|
message = message[:50] + "..."
|
||||||
|
|
||||||
return f"正在发送{type_desc}消息: {message}"
|
if title:
|
||||||
|
return f"正在发送消息: [{title}] {message}"
|
||||||
|
return f"正在发送消息: {message}"
|
||||||
|
|
||||||
async def run(
|
async def run(
|
||||||
self, message: str, message_type: Optional[str] = None, **kwargs
|
self, message: str, message_type: Optional[str] = None, **kwargs
|
||||||
) -> str:
|
) -> str:
|
||||||
logger.info(
|
title = message_type or ""
|
||||||
f"执行工具: {self.name}, 参数: message={message}, message_type={message_type}"
|
logger.info(f"执行工具: {self.name}, 参数: title={title}, message={message}")
|
||||||
)
|
|
||||||
try:
|
try:
|
||||||
await self.send_tool_message(message, title=message_type)
|
await self.send_tool_message(message, title=title)
|
||||||
return "消息已发送"
|
return "消息已发送"
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"发送消息失败: {e}")
|
logger.error(f"发送消息失败: {e}")
|
||||||
|
|||||||
Reference in New Issue
Block a user