jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[Chore/Refactor] Improve type annotations in models module (#25281)

Browse source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
This commit is contained in:
-LAN- 2025-09-08 09:42:27 +08:00 • committed by GitHub
commit 9b8a03b53b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
23 changed files with 332 additions and 251 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/tools/mcp_tool/provider.py
View file

@ -1,5 +1,5 @@
import json
from typing import Any, Optional
from typing import Any, Optional, Self
from core.mcp.types import Tool as RemoteMCPTool
from core.tools.__base.tool_provider import ToolProviderController
@ -48,7 +48,7 @@ class MCPToolProviderController(ToolProviderController):
return ToolProviderType.MCP
@classmethod
def _from_db(cls, db_provider: MCPToolProvider) -> "MCPToolProviderController":
def from_db(cls, db_provider: MCPToolProvider) -> Self:
"""
from db provider
"""

4
api/core/tools/tool_manager.py
View file

@ -773,7 +773,7 @@ class ToolManager:
if provider is None:
raise ToolProviderNotFoundError(f"mcp provider {provider_id} not found")
controller = MCPToolProviderController._from_db(provider)
controller = MCPToolProviderController.from_db(provider)
return controller
@ -928,7 +928,7 @@ class ToolManager:
tenant_id: str,
provider_type: ToolProviderType,
provider_id: str,
) -> Union[str, dict]:
) -> Union[str, dict[str, Any]]:
"""
get the tool icon