jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: support pinning, including, and excluding for Model Providers and Tools (#7283)

Browse source
This commit is contained in:
Xiyuan Chen 2024-08-15 00:58:38 -04:00 • committed by GitHub
commit 7619850855
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 204 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/core/tools/provider/builtin/_positions.py
View file

@ -1,6 +1,6 @@
import os.path
from core.helper.position_helper import get_position_map, sort_by_position_map
from core.helper.position_helper import get_tool_position_map, sort_by_position_map
from core.tools.entities.api_entities import UserToolProvider
@ -10,11 +10,11 @@ class BuiltinToolProviderSort:
@classmethod
def sort(cls, providers: list[UserToolProvider]) -> list[UserToolProvider]:
if not cls._position:
cls._position = get_position_map(os.path.join(os.path.dirname(__file__), '..'))
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
return sorted_providers