mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
055117d83d
- 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.
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
"""查询订阅工具"""
|
|
|
|
import json
|
|
from typing import Optional
|
|
|
|
from app.db.subscribe_oper import SubscribeOper
|
|
from app.log import logger
|
|
from app.agent.tools.base import MoviePilotTool
|
|
|
|
|
|
class QuerySubscribesTool(MoviePilotTool):
|
|
name: str = "query_subscribes"
|
|
description: str = "查询订阅状态，查看用户的订阅列表和状态。"
|
|
|
|
async def _arun(self, explanation: str, status: Optional[str] = "all",
|
|
media_type: Optional[str] = "all") -> str:
|
|
logger.info(f"执行工具: {self.name}, 参数: status={status}, media_type={media_type}")
|
|
try:
|
|
subscribe_oper = SubscribeOper()
|
|
subscribes = subscribe_oper.list()
|
|
filtered_subscribes = []
|
|
for sub in subscribes:
|
|
if status != "all" and sub.state != status:
|
|
continue
|
|
if media_type != "all" and sub.type != media_type:
|
|
continue
|
|
filtered_subscribes.append(sub)
|
|
if filtered_subscribes:
|
|
return json.dumps([s.to_dict() for s in filtered_subscribes], ensure_ascii=False, indent=2)
|
|
return "未找到相关订阅。"
|
|
except Exception as e:
|
|
logger.error(f"查询订阅失败: {e}", exc_info=True)
|
|
return f"查询订阅时发生错误: {str(e)}"
|