fix: replace GenericProviderID with ToolProviderID

langgenius · Feb 25, 2025 · f3c4621 · f3c4621
1 parent a0a0985
commit f3c4621
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/core/tools/tool_manager.py b/api/core/tools/tool_manager.py
@@ -9,7 +9,7 @@
 from yarl import URL
 
 import contexts
-from core.plugin.entities.plugin import GenericProviderID, ToolProviderID
+from core.plugin.entities.plugin import ToolProviderID
 from core.plugin.manager.tool import PluginToolManager
 from core.tools.__base.tool_provider import ToolProviderController
 from core.tools.__base.tool_runtime import ToolRuntime
@@ -188,7 +188,7 @@ def get_tool_runtime(
                 )
 
             if isinstance(provider_controller, PluginToolProviderController):
-                provider_id_entity = GenericProviderID(provider_id)
+                provider_id_entity = ToolProviderID(provider_id)
                 # get credentials
                 builtin_provider: BuiltinToolProvider | None = (
                     db.session.query(BuiltinToolProvider)