hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-16 06:16:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "feat: support pinning, including, and excluding for Model Providers and Tools" (#7324)

Browse Source
This commit is contained in:
Xiyuan Chen
2024-08-15 11:51:00 -04:00
committed by GitHub
parent fcb6921b57
commit c7df6783df
8 changed files with 12 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
api/tests/unit_tests/utils/position_helper/test_position_helper.py
View File

@@ -2,7 +2,7 @@ from textwrap import dedent
import pytest
from core.helper.position_helper import get_position_map, sort_and_filter_position_map
from core.helper.position_helper import get_position_map
@pytest.fixture
@@ -53,47 +53,3 @@ def test_position_helper_with_all_commented(prepare_empty_commented_positions_ya
folder_path=prepare_empty_commented_positions_yaml,
file_name='example_positions_all_commented.yaml')
assert position_map == {}
def test_excluded_position_map(prepare_example_positions_yaml):
position_map = get_position_map(
folder_path=prepare_example_positions_yaml,
file_name='example_positions.yaml'
)
pin_list = ['forth', 'first']
include_list = []
exclude_list = ['9999999999999']
sorted_filtered_position_map = sort_and_filter_position_map(
original_position_map=position_map,
pin_list=pin_list,
include_list=include_list,
exclude_list=exclude_list
)
assert sorted_filtered_position_map == {
'forth': 0,
'first': 1,
'second': 2,
'third': 3,
}
def test_included_position_map(prepare_example_positions_yaml):
position_map = get_position_map(
folder_path=prepare_example_positions_yaml,
file_name='example_positions.yaml'
)
pin_list = ['second', 'first']
include_list = ['first', 'second', 'third', 'forth']
exclude_list = []
sorted_filtered_position_map = sort_and_filter_position_map(
original_position_map=position_map,
pin_list=pin_list,
include_list=include_list,
exclude_list=exclude_list
)
assert sorted_filtered_position_map == {
'second': 0,
'first': 1,
'third': 2,
'forth': 3,
}