fix agent tools

This commit is contained in:
jxxghp
2025-11-01 12:01:48 +08:00
parent 247208b8a9
commit b45b603b97
12 changed files with 145 additions and 331 deletions
+11 -2
View File
@@ -1,6 +1,9 @@
"""查询下载器工具"""
import json
from typing import Type
from pydantic import BaseModel, Field
from app.db.systemconfig_oper import SystemConfigOper
from app.log import logger
@@ -8,11 +11,17 @@ from app.schemas.types import SystemConfigKey
from app.agent.tools.base import MoviePilotTool
class QueryDownloadersInput(BaseModel):
"""查询下载器工具的输入参数模型"""
explanation: str = Field(..., description="Clear explanation of why this tool is being used in the current context")
class QueryDownloadersTool(MoviePilotTool):
name: str = "query_downloaders"
description: str = "查询下载器配置,查看可用的下载器列表和配置信息。"
description: str = "Query downloader configuration and list all available downloaders. Shows downloader status, connection details, and configuration settings."
args_schema: Type[BaseModel] = QueryDownloadersInput
async def _arun(self, explanation: str, **kwargs) -> str:
async def _arun(self, **kwargs) -> str:
logger.info(f"执行工具: {self.name}")
try:
system_config_oper = SystemConfigOper()