fix pydantic

This commit is contained in:
jxxghp
2025-11-01 09:51:23 +08:00
parent 09a19e94d5
commit d523c7c916
47 changed files with 160 additions and 135 deletions

View File

@@ -27,7 +27,7 @@ class QueryDownloadsTool(MoviePilotTool):
continue
filtered_downloads.append(dl)
if filtered_downloads:
return json.dumps([d.dict() for d in filtered_downloads])
return json.dumps([d.model_dump() for d in filtered_downloads])
return "未找到相关下载任务。"
except Exception as e:
logger.error(f"查询下载失败: {e}", exc_info=True)

View File

@@ -3,7 +3,7 @@
import json
from typing import Optional
from app.db.media_oper import MediaOper
from app.db.mediaserver_oper import MediaServerOper
from app.log import logger
from app.agent.tools.base import MoviePilotTool
@@ -16,8 +16,8 @@ class QueryMediaLibraryTool(MoviePilotTool):
title: Optional[str] = None) -> str:
logger.info(f"执行工具: {self.name}, 参数: media_type={media_type}, title={title}")
try:
media_oper = MediaOper()
medias = media_oper.list()
media_server_oper = MediaServerOper()
medias = media_server_oper.list()
filtered_medias = []
for media in medias:
if media_type != "all" and media.type != media_type: