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

fix search tools

Browse Source
This commit is contained in:
jxxghp
2025-11-17 09:14:36 +08:00
parent 78121917c6
commit 6d5a85b144
2 changed files with 3 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app/agent/tools/impl/search_media.py
View File

@@ -63,25 +63,12 @@ class SearchMediaTool(MoviePilotTool):
filtered_results.append(result)
if filtered_results:
result_message = f"找到 {len(filtered_results)} 个相关媒体资源"
await self.send_tool_message(result_message, title="搜索成功")
# 发送详细结果
for i, result in enumerate(filtered_results[:5]): # 只显示前5个结果
media_info = f"{i + 1}. {result.title} ({result.year}) - {result.type.value if result.type else '未知'}"
await self.send_tool_message(media_info, title="搜索结果")
return json.dumps([r.to_dict() for r in filtered_results], ensure_ascii=False, indent=2)
else:
error_message = f"未找到符合条件的媒体资源: {title}"
await self.send_tool_message(error_message, title="搜索完成")
return error_message
return f"未找到符合条件的媒体资源: {title}"
else:
error_message = f"未找到相关媒体资源: {title}"
await self.send_tool_message(error_message, title="搜索完成")
return error_message
return f"未找到相关媒体资源: {title}"
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

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

@@ -55,23 +55,10 @@ class SearchTorrentsTool(MoviePilotTool):
filtered_torrents.append(torrent)
if filtered_torrents:
result_message = f"找到 {len(filtered_torrents)} 个相关种子资源"
await self.send_tool_message(result_message, title="搜索成功")
# 发送详细结果
for i, torrent in enumerate(filtered_torrents[:5]): # 只显示前5个结果
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}"
await self.send_tool_message(torrent_info, title="搜索结果")
return json.dumps([t.to_dict() for t in filtered_torrents], ensure_ascii=False, indent=2)
else:
error_message = f"未找到相关种子资源: {title}"
await self.send_tool_message(error_message, title="搜索完成")
return error_message
return f"未找到相关种子资源: {title}"
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