MCP TOOLS UNIT TEST

Author: NeptuneHub

test/unit/test_mcp_server.py

@@ -607,6 +607,21 @@ def test_unknown_tool_returns_error(self):
 
 
 class TestToolSurface:
+    def test_shared_schemas_omit_ollama_only_unique_items(self):
+        ai_mod = _import_ai_mcp_client()
+
+        def contains_unique_items(value):
+            if isinstance(value, dict):
+                return 'uniqueItems' in value or any(
+                    contains_unique_items(child) for child in value.values()
+                )
+            if isinstance(value, list):
+                return any(contains_unique_items(child) for child in value)
+            return False
+
+        for tool in ai_mod.get_mcp_tools():
+            assert not contains_unique_items(tool['inputSchema'])
+
     def test_no_llm_facing_get_songs_or_dead_params(self):
         ai_mod = _import_ai_mcp_client()
         for tool in ai_mod.get_mcp_tools():

test/unit/test_tool_plan.py

@@ -593,6 +593,39 @@ def test_source_schema_not_mutated(self):
         pr.build_tool_calls_schema(tools)
         assert 'additionalProperties' not in tools[0]['inputSchema']
 
+    def test_unique_items_are_added_only_to_derived_argument_arrays(self):
+        pr = _prompts()
+        tools = _tools_fixture()
+
+        schema = pr.build_tool_calls_schema(tools)
+        branches = schema['properties']['tool_calls']['items']['oneOf']
+        argument_schemas = [branch['properties']['arguments'] for branch in branches]
+
+        def array_schemas(value):
+            if isinstance(value, dict):
+                if value.get('type') == 'array':
+                    yield value
+                for child in value.values():
+                    yield from array_schemas(child)
+            elif isinstance(value, list):
+                for child in value:
+                    yield from array_schemas(child)
+
+        derived_arrays = [
+            array_schema
+            for argument_schema in argument_schemas
+            for array_schema in array_schemas(argument_schema)
+        ]
+        source_arrays = [
+            array_schema
+            for tool in tools
+            for array_schema in array_schemas(tool['inputSchema'])
+        ]
+
+        assert derived_arrays
+        assert all(array_schema['uniqueItems'] is True for array_schema in derived_arrays)
+        assert all('uniqueItems' not in array_schema for array_schema in source_arrays)
+
 
 class TestPromptRendering:
     def test_system_prompt_derives_tools_and_rules(self):