Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 19:59:50 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fix: remove magic number

Browse Source
This commit is contained in:
GareArc 2024-08-15 14:41:39 -04:00
parent 68bd78701f
commit fab4570463
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/constants/position.py Normal file
View File

@ -0,0 +1,3 @@
# Used to mark a position as excluded.
# See api/core/helper/position_helper.py for more details.
POSITION_EXCLUDED = -999

6
api/core/helper/position_helper.py
View File

@ -4,6 +4,7 @@ from collections.abc import Callable
from typing import Any
from configs import dify_config
from constants.position import POSITION_EXCLUDED
from core.tools.utils.yaml_utils import load_yaml_file
@ -73,7 +74,7 @@ def sort_and_filter_position_map(original_position_map: dict[str, int], pin_list
start_idx = len(position_map)
for name in positions:
if name in exclude_set:
position_map[name] = -1
position_map[name] = POSITION_EXCLUDED
elif name in include_set and name not in position_map:
position_map[name] = start_idx
start_idx += 1
@ -96,8 +97,7 @@ def sort_by_position_map(
if not position_map or not data:
return data
filtered_data = [item for item in data if position_map.get(name_func(item), 0) >= 0]
filtered_data = [item for item in data if position_map.get(name_func(item), POSITION_EXCLUDED+1) != POSITION_EXCLUDED]
return sorted(filtered_data, key=lambda x: position_map.get(name_func(x), float('inf')))