[Chore/Refactor] Improve type annotations in models module (#25281)
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
This commit is contained in:
@@ -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
|
||||
|
||||
|
Reference in New Issue
Block a user