tools_manage_service.py 800 B

123456789101112131415161718192021222324252627
  1. import logging
  2. from core.tools.entities.api_entities import ToolProviderTypeApiLiteral
  3. from core.tools.tool_manager import ToolManager
  4. from services.tools.tools_transform_service import ToolTransformService
  5. logger = logging.getLogger(__name__)
  6. class ToolCommonService:
  7. @staticmethod
  8. def list_tool_providers(user_id: str, tenant_id: str, typ: ToolProviderTypeApiLiteral = None):
  9. """
  10. list tool providers
  11. :return: the list of tool providers
  12. """
  13. providers = ToolManager.list_providers_from_api(user_id, tenant_id, typ)
  14. # add icon
  15. for provider in providers:
  16. ToolTransformService.repack_provider(tenant_id=tenant_id, provider=provider)
  17. result = [provider.to_dict() for provider in providers]
  18. return result