nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-03-19 19:46:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix agent tools

Browse Source
This commit is contained in:
jxxghp
2025-11-17 09:28:18 +08:00
parent 6d5a85b144
commit b7d121c58f
7 changed files with 19 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
app/agent/tools/impl/add_download.py
View File

@@ -37,14 +37,9 @@ class AddDownloadTool(MoviePilotTool):
logger.info(
f"执行工具: {self.name}, 参数: torrent_title={torrent_title}, torrent_url={torrent_url}, downloader={downloader}, save_path={save_path}, labels={labels}")
# 发送工具执行说明
await self.send_tool_message(f"正在添加下载任务: {torrent_title}", title="添加下载")
try:
if not torrent_title or not torrent_url:
error_message = "错误：必须提供种子标题和下载链接"
await self.send_tool_message(error_message, title="下载失败")
return error_message
return "错误：必须提供种子标题和下载链接"
# 使用DownloadChain添加下载
download_chain = DownloadChain()
@@ -67,15 +62,9 @@ class AddDownloadTool(MoviePilotTool):
label=labels
)
if did:
success_message = f"成功添加下载任务：{torrent_title}"
await self.send_tool_message(success_message, title="下载成功")
return success_message
return f"成功添加下载任务：{torrent_title}"
else:
error_message = "添加下载任务失败"
await self.send_tool_message(error_message, title="下载失败")
return error_message
return "添加下载任务失败"
except Exception as e:
error_message = f"添加下载任务时发生错误: {str(e)}"
logger.error(f"添加下载任务失败: {e}", exc_info=True)
await self.send_tool_message(error_message, title="下载失败")
return error_message
return f"添加下载任务时发生错误: {str(e)}"