瀏覽代碼

remove useless Tool class attributes (#23389)

tags/1.7.2
Will 2 月之前
父節點
當前提交
7fe23a0ca6
No account linked to committer's email address

+ 0
- 3
api/core/tools/__base/tool.py 查看文件

@@ -20,9 +20,6 @@ class Tool(ABC):
The base class of a tool
"""

entity: ToolEntity
runtime: ToolRuntime

def __init__(self, entity: ToolEntity, runtime: ToolRuntime) -> None:
self.entity = entity
self.runtime = runtime

+ 0
- 2
api/core/tools/builtin_tool/tool.py 查看文件

@@ -20,8 +20,6 @@ class BuiltinTool(Tool):
:param meta: the meta data of a tool call processing
"""

provider: str

def __init__(self, provider: str, **kwargs):
super().__init__(**kwargs)
self.provider = provider

+ 0
- 3
api/core/tools/custom_tool/tool.py 查看文件

@@ -21,9 +21,6 @@ API_TOOL_DEFAULT_TIMEOUT = (


class ApiTool(Tool):
api_bundle: ApiToolBundle
provider_id: str

"""
Api tool
"""

+ 1
- 8
api/core/tools/mcp_tool/tool.py 查看文件

@@ -8,23 +8,16 @@ from core.mcp.mcp_client import MCPClient
from core.mcp.types import ImageContent, TextContent
from core.tools.__base.tool import Tool
from core.tools.__base.tool_runtime import ToolRuntime
from core.tools.entities.tool_entities import ToolEntity, ToolInvokeMessage, ToolParameter, ToolProviderType
from core.tools.entities.tool_entities import ToolEntity, ToolInvokeMessage, ToolProviderType


class MCPTool(Tool):
tenant_id: str
icon: str
runtime_parameters: Optional[list[ToolParameter]]
server_url: str
provider_id: str

def __init__(
self, entity: ToolEntity, runtime: ToolRuntime, tenant_id: str, icon: str, server_url: str, provider_id: str
) -> None:
super().__init__(entity, runtime)
self.tenant_id = tenant_id
self.icon = icon
self.runtime_parameters = None
self.server_url = server_url
self.provider_id = provider_id


+ 1
- 6
api/core/tools/plugin_tool/tool.py 查看文件

@@ -9,11 +9,6 @@ from core.tools.entities.tool_entities import ToolEntity, ToolInvokeMessage, Too


class PluginTool(Tool):
tenant_id: str
icon: str
plugin_unique_identifier: str
runtime_parameters: Optional[list[ToolParameter]]

def __init__(
self, entity: ToolEntity, runtime: ToolRuntime, tenant_id: str, icon: str, plugin_unique_identifier: str
) -> None:
@@ -21,7 +16,7 @@ class PluginTool(Tool):
self.tenant_id = tenant_id
self.icon = icon
self.plugin_unique_identifier = plugin_unique_identifier
self.runtime_parameters = None
self.runtime_parameters: Optional[list[ToolParameter]] = None

def tool_provider_type(self) -> ToolProviderType:
return ToolProviderType.PLUGIN

+ 0
- 2
api/core/tools/utils/dataset_retriever_tool.py 查看文件

@@ -20,8 +20,6 @@ from core.tools.utils.dataset_retriever.dataset_retriever_base_tool import Datas


class DatasetRetrieverTool(Tool):
retrieval_tool: DatasetRetrieverBaseTool

def __init__(self, entity: ToolEntity, runtime: ToolRuntime, retrieval_tool: DatasetRetrieverBaseTool) -> None:
super().__init__(entity, runtime)
self.retrieval_tool = retrieval_tool

+ 0
- 9
api/core/tools/workflow_as_tool/tool.py 查看文件

@@ -25,15 +25,6 @@ logger = logging.getLogger(__name__)


class WorkflowTool(Tool):
workflow_app_id: str
version: str
workflow_entities: dict[str, Any]
workflow_call_depth: int
thread_pool_id: Optional[str] = None
workflow_as_tool_id: str

label: str

"""
Workflow tool.
"""

Loading…
取消
儲存