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

api/tests/unit_tests/utils/position_helper/test_position_helper.py

@@ -2,7 +2,7 @@ from textwrap import dedent
 
 import pytest
 
-from core.helper.position_helper import get_position_map
+from core.helper.position_helper import get_position_map, sort_and_filter_position_map
 
 
 @pytest.fixture
@@ -53,3 +53,47 @@ 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,
+    }