fix agent tools

This commit is contained in:
jxxghp
2025-11-01 12:01:48 +08:00
parent 247208b8a9
commit b45b603b97
12 changed files with 145 additions and 331 deletions

View File

@@ -1,16 +1,26 @@
"""发送消息工具"""
from typing import Optional
from typing import Optional, Type
from pydantic import BaseModel, Field
from app.agent.tools.base import MoviePilotTool
from app.log import logger
class SendMessageInput(BaseModel):
"""发送消息工具的输入参数模型"""
explanation: str = Field(..., description="Clear explanation of why this tool is being used in the current context")
message: str = Field(..., description="The message content to send to the user (should be clear and informative)")
message_type: Optional[str] = Field("info", description="Type of message: 'info' for general information, 'success' for successful operations, 'warning' for warnings, 'error' for error messages")
class SendMessageTool(MoviePilotTool):
name: str = "send_message"
description: str = "发送消息通知,向用户发送操作结果或重要信息。"
description: str = "Send notification message to the user through configured notification channels (Telegram, Slack, WeChat, etc.). Used to inform users about operation results, errors, or important updates."
args_schema: Type[BaseModel] = SendMessageInput
async def _arun(self, message: str, explanation: str, message_type: Optional[str] = "info", **kwargs) -> str:
async def _arun(self, message: str, message_type: Optional[str] = None, **kwargs) -> str:
logger.info(f"执行工具: {self.name}, 参数: message={message}, message_type={message_type}")
try:
self.send_tool_message(message, title=message_type)