fix agent

app/agent/tools/impl/query_media_library.py

@@ -1,30 +1,24 @@
 """查询媒体库工具"""
 
 import json
-from typing import Optional
+from typing import Optional, List
 
+from app.agent.tools.base import MoviePilotTool
 from app.db.mediaserver_oper import MediaServerOper
 from app.log import logger
-from app.agent.tools.base import MoviePilotTool
+from app.schemas import MediaServerItem
 
 
 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, **kwargs) -> str:
+    async def _arun(self, explanation: str, media_type: Optional[str] = "all",
+                    title: Optional[str] = None, year: Optional[str] = None, **kwargs) -> str:
         logger.info(f"执行工具: {self.name}, 参数: media_type={media_type}, title={title}")
         try:
             media_server_oper = MediaServerOper()
-            medias = media_server_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)
+            filtered_medias: List[MediaServerItem] = media_server_oper.exists(title=title, year=year, mtype=media_type)
             if filtered_medias:
                 return json.dumps([m.to_dict() for m in filtered_medias])
             return "媒体库中未找到相关媒体。"