From 07ec576dfb76fa45bea3772685f7ed8a8a48fd52 Mon Sep 17 00:00:00 2001
From: Harshit28j <Harshitjain0562@gmail.com>
Date: Fri, 6 Mar 2026 23:44:04 +0530
Subject: [PATCH] fix: add missing MCP mock fields and document new providers

- Fix test_add_update_server_without_alias: set tool_name_to_display_name,
  tool_name_to_description, byok_api_key_help_url to None to avoid
  MagicMock Pydantic validation errors
- Add bedrock_mantle and searchapi to provider_endpoints_support.json
  to fix check_code_and_doc_quality CI failure

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 provider_endpoints_support.json    | 35 ++++++++++++++++++++++++++++++
 tests/mcp_tests/test_mcp_server.py |  3 +++
 2 files changed, 38 insertions(+)

diff --git a/provider_endpoints_support.json b/provider_endpoints_support.json
index 8834d8b19c0..eefe69a67c1 100644
--- a/provider_endpoints_support.json
+++ b/provider_endpoints_support.json
@@ -229,6 +229,24 @@
         "interactions": true
       }
     },
+    "bedrock_mantle": {
+      "display_name": "AWS - Bedrock Mantle (`bedrock_mantle`)",
+      "url": "https://docs.litellm.ai/docs/providers/bedrock",
+      "endpoints": {
+        "chat_completions": true,
+        "messages": false,
+        "responses": false,
+        "embeddings": false,
+        "image_generations": false,
+        "audio_transcriptions": false,
+        "audio_speech": false,
+        "moderations": false,
+        "batches": false,
+        "rerank": false,
+        "a2a": false,
+        "interactions": false
+      }
+    },
     "bedrock": {
       "display_name": "AWS - Bedrock (`bedrock`)",
       "url": "https://docs.litellm.ai/docs/providers/bedrock",
@@ -1888,6 +1906,23 @@
         "interactions": true
       }
     },
+    "searchapi": {
+      "display_name": "SearchAPI.io (`searchapi`)",
+      "url": "https://docs.litellm.ai/docs/providers/searchapi",
+      "endpoints": {
+        "chat_completions": false,
+        "messages": false,
+        "responses": false,
+        "embeddings": false,
+        "image_generations": false,
+        "audio_transcriptions": false,
+        "audio_speech": false,
+        "moderations": false,
+        "batches": false,
+        "rerank": false,
+        "search": true
+      }
+    },
     "searxng": {
       "display_name": "SearXNG (`searxng`)",
       "url": "https://docs.litellm.ai/docs/search/searxng",
diff --git a/tests/mcp_tests/test_mcp_server.py b/tests/mcp_tests/test_mcp_server.py
index c71e55b086a..94d2008294a 100644
--- a/tests/mcp_tests/test_mcp_server.py
+++ b/tests/mcp_tests/test_mcp_server.py
@@ -1497,6 +1497,9 @@ async def test_add_update_server_without_alias():
     mock_mcp_server.authorization_url = None
     mock_mcp_server.registration_url = None
     mock_mcp_server.token_url = None
+    mock_mcp_server.tool_name_to_display_name = None
+    mock_mcp_server.tool_name_to_description = None
+    mock_mcp_server.byok_api_key_help_url = None
 
     # Add server to manager
     await test_manager.add_server(mock_mcp_server)