dify/api/services/tools/tools_manage_service.py

27 lines
800 B
Python
Raw Normal View History

2024-05-27 22:01:11 +08:00
import logging
2024-09-23 18:06:16 +08:00
from core.tools.entities.api_entities import ToolProviderTypeApiLiteral
2024-05-27 22:01:11 +08:00
from core.tools.tool_manager import ToolManager
from services.tools.tools_transform_service import ToolTransformService
logger = logging.getLogger(__name__)
class ToolCommonService:
@staticmethod
2024-09-23 18:06:16 +08:00
def list_tool_providers(user_id: str, tenant_id: str, typ: ToolProviderTypeApiLiteral = None):
2024-05-27 22:01:11 +08:00
"""
list tool providers
2024-05-27 22:01:11 +08:00
:return: the list of tool providers
2024-05-27 22:01:11 +08:00
"""
2024-09-24 16:35:01 +08:00
providers = ToolManager.list_providers_from_api(user_id, tenant_id, typ)
2024-05-27 22:01:11 +08:00
# add icon
for provider in providers:
2024-10-30 16:09:17 +08:00
ToolTransformService.repack_provider(tenant_id=tenant_id, provider=provider)
2024-05-27 22:01:11 +08:00
result = [provider.to_dict() for provider in providers]
return result