tjdify/api/core/tools/provider/builtin/_positions.py

import os.path

from core.helper.position_helper import get_tool_position_map, sort_by_position_map
from core.tools.entities.api_entities import UserToolProvider


class BuiltinToolProviderSort:
    _position: dict[str, int] = {}

    @classmethod
    def sort(cls, providers: list[UserToolProvider]) -> list[UserToolProvider]:
        if not cls._position:
            cls._position = get_tool_position_map(os.path.join(os.path.dirname(__file__), ".."))

        def name_func(provider: UserToolProvider) -> str:
            return provider.name

        sorted_providers = sort_by_position_map(cls._position, providers, name_func)

        return sorted_providers