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.4 KiB
Python
34 lines
1.4 KiB
Python
"""查询媒体库工具"""
|
|
|
|
import json
|
|
from typing import Optional
|
|
|
|
from app.db.media_oper import MediaOper
|
|
from app.log import logger
|
|
from app.agent.tools.base import MoviePilotTool
|
|
|
|
|
|
class QueryMediaLibraryTool(MoviePilotTool):
|
|
name: str = "query_media_library"
|
|
description: str = "查询媒体库状态，查看已入库的媒体文件情况。"
|
|
|
|
async def _arun(self, explanation: str, media_type: Optional[str] = "all",
|
|
title: Optional[str] = None) -> str:
|
|
logger.info(f"执行工具: {self.name}, 参数: media_type={media_type}, title={title}")
|
|
try:
|
|
media_oper = MediaOper()
|
|
medias = media_oper.list()
|
|
filtered_medias = []
|
|
for media in medias:
|
|
if media_type != "all" and media.type != media_type:
|
|
continue
|
|
if title and title.lower() not in media.title.lower():
|
|
continue
|
|
filtered_medias.append(media)
|
|
if filtered_medias:
|
|
return json.dumps([m.to_dict() if hasattr(m, 'to_dict') else m.dict() if hasattr(m, 'dict') else m.model_dump() if hasattr(m, 'model_dump') else m for m in filtered_medias], ensure_ascii=False, indent=2)
|
|
return "媒体库中未找到相关媒体。"
|
|
except Exception as e:
|
|
logger.error(f"查询媒体库失败: {e}", exc_info=True)
|
|
return f"查询媒体库时发生错误: {str(e)}"
|