estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: Replace GenericProviderID with ToolProviderID (#21064)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN-
2025-06-16 17:35:46 +08:00
committed by GitHub
parent 41cb39eb5d
commit 2440ac43b1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/services/tools/builtin_tools_manage_service.py
View File

@@ -7,7 +7,7 @@ from sqlalchemy.orm import Session
from configs import dify_config from configs import dify_config
from core.helper.position_helper import is_filtered from core.helper.position_helper import is_filtered
from core.model_runtime.utils.encoders import jsonable_encoder from core.model_runtime.utils.encoders import jsonable_encoder
from core.plugin.entities.plugin import GenericProviderID, ToolProviderID from core.plugin.entities.plugin import ToolProviderID
from core.plugin.impl.exc import PluginDaemonClientSideError from core.plugin.impl.exc import PluginDaemonClientSideError
from core.tools.builtin_tool.providers._positions import BuiltinToolProviderSort from core.tools.builtin_tool.providers._positions import BuiltinToolProviderSort
from core.tools.entities.api_entities import ToolApiEntity, ToolProviderApiEntity from core.tools.entities.api_entities import ToolApiEntity, ToolProviderApiEntity
@@ -290,7 +290,7 @@ class BuiltinToolManageService:
def _fetch_builtin_provider(provider_name: str, tenant_id: str) -> BuiltinToolProvider | None: def _fetch_builtin_provider(provider_name: str, tenant_id: str) -> BuiltinToolProvider | None:
try: try:
full_provider_name = provider_name full_provider_name = provider_name
provider_id_entity = GenericProviderID(provider_name) provider_id_entity = ToolProviderID(provider_name)
provider_name = provider_id_entity.provider_name provider_name = provider_id_entity.provider_name
if provider_id_entity.organization != "langgenius": if provider_id_entity.organization != "langgenius":
provider_obj = ( provider_obj = (
@@ -315,7 +315,7 @@ class BuiltinToolManageService:
if provider_obj is None: if provider_obj is None:
return None return None
provider_obj.provider = GenericProviderID(provider_obj.provider).to_string() provider_obj.provider = ToolProviderID(provider_obj.provider).to_string()
return provider_obj return provider_obj
except Exception: except Exception:
# it's an old provider without organization # it's an old provider without organization