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

refactor: enhance tool message handling and improve error logging

Browse Source 
- Updated _send_tool_message to accept a title parameter for better message context.
- Modified various tool implementations to utilize the new title parameter for clearer messaging.
- Improved error logging across multiple tools to include exception details for better debugging.
This commit is contained in:
jxxghp
2025-10-31 09:16:53 +08:00
parent c6baf43986
commit 055117d83d
9 changed files with 115 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
app/agent/tools/impl/search_torrents.py
View File

@@ -19,36 +19,44 @@ class SearchTorrentsTool(MoviePilotTool):
logger.info(f"执行工具: {self.name}, 参数: title={title}, year={year}, media_type={media_type}, season={season}, sites={sites}")
# 发送工具执行说明
self._send_tool_message(f"正在搜索种子资源: {title}" + (f" ({year})" if year else ""), "info")
self._send_tool_message(f"正在搜索种子资源: {title}" + (f" ({year})" if year else ""), title="搜索种子")
try:
search_chain = SearchChain()
torrents = search_chain.search_by_title(title=title, sites=sites)
filtered_torrents = []
for torrent in torrents:
if year and torrent.meta_info.year != year:
# torrent 是 Context 对象，需要通过 meta_info 和 media_info 访问属性
if year and torrent.meta_info and torrent.meta_info.year != year:
continue
if media_type and torrent.media_info and torrent.media_info.type != MediaType(media_type):
continue
if season and torrent.meta_info.begin_season != season:
if media_type and torrent.media_info:
try:
if torrent.media_info.type != MediaType(media_type):
continue
except:
pass
if season and torrent.meta_info and torrent.meta_info.begin_season != season:
continue
filtered_torrents.append(torrent)
if filtered_torrents:
result_message = f"找到 {len(filtered_torrents)} 个相关种子资源"
self._send_tool_message(result_message, "success")
self._send_tool_message(result_message, title="搜索成功")
# 发送详细结果
for i, torrent in enumerate(filtered_torrents[:5]): # 只显示前5个结果
torrent_info = f"{i+1}. {torrent.title} - {torrent.site_name}"
self._send_tool_message(torrent_info, "info")
torrent_title = torrent.torrent_info.title if torrent.torrent_info else torrent.meta_info.title if torrent.meta_info else "未知"
site_name = torrent.torrent_info.site_name if torrent.torrent_info else "未知站点"
torrent_info = f"{i+1}. {torrent_title} - {site_name}"
self._send_tool_message(torrent_info, title="搜索结果")
return json.dumps([t.dict() for t in filtered_torrents], ensure_ascii=False, indent=2)
return json.dumps([t.to_dict() for t in filtered_torrents], ensure_ascii=False, indent=2)
else:
error_message = f"未找到相关种子资源: {title}"
self._send_tool_message(error_message, "warning")
self._send_tool_message(error_message, title="搜索完成")
return error_message
except Exception as e:
error_message = f"搜索种子时发生错误: {str(e)}"
self._send_tool_message(error_message, "error")
logger.error(f"搜索种子失败: {e}", exc_info=True)
self._send_tool_message(error_message, title="搜索失败")
return error_message