fix: async configure_mcp_server (#206)

Author: tstadel

src/deepset_mcp/main.py

@@ -2,6 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
+import asyncio
 import logging
 import os
 from enum import StrEnum
@@ -191,17 +192,19 @@ def main(
         raise typer.Exit(1)
 
     mcp = FastMCP("deepset AI platform MCP server")
-    configure_mcp_server(
-        mcp_server_instance=mcp,
-        deepset_api_key=api_key,
-        deepset_api_url=api_url,
-        deepset_workspace=workspace,
-        tools_to_register=tool_names,
-        deepset_docs_shareable_prototype_url=docs_share_url,
-        get_api_key_from_authorization_header=api_key_from_auth_header,
-        object_store_backend=backend,
-        object_store_redis_url=redis_url,
-        object_store_ttl=ttl,
+    asyncio.run(
+        configure_mcp_server(
+            mcp_server_instance=mcp,
+            deepset_api_key=api_key,
+            deepset_api_url=api_url,
+            deepset_workspace=workspace,
+            tools_to_register=tool_names,
+            deepset_docs_shareable_prototype_url=docs_share_url,
+            get_api_key_from_authorization_header=api_key_from_auth_header,
+            object_store_backend=backend,
+            object_store_redis_url=redis_url,
+            object_store_ttl=ttl,
+        )
     )
     mcp.settings.host = host
     if port is not None:

src/deepset_mcp/mcp/server.py

@@ -2,7 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import asyncio
 from urllib.parse import parse_qs, urlparse
 
 import jwt
@@ -16,7 +15,7 @@
 from deepset_mcp.mcp.tool_registry import TOOL_REGISTRY
 
 
-def configure_mcp_server(
+async def configure_mcp_server(
     mcp_server_instance: FastMCP,
     tools_to_register: set[str] | None = None,
     deepset_api_key: str | None = None,
@@ -56,8 +55,8 @@ def configure_mcp_server(
             "Please provide 'deepset_api_key' or enable 'get_api_key_from_authorization_header'."
         )
 
-    workspace_name, pipeline_name, api_key_docs = asyncio.run(
-        fetch_shared_prototype_details(deepset_docs_shareable_prototype_url)
+    workspace_name, pipeline_name, api_key_docs = await fetch_shared_prototype_details(
+        deepset_docs_shareable_prototype_url
     )
     docs_config = DeepsetDocsConfig(api_key=api_key_docs, workspace_name=workspace_name, pipeline_name=pipeline_name)
 

test/unit/test_server_base_url.py

@@ -6,19 +6,22 @@
 
 from unittest.mock import MagicMock, patch
 
+import pytest
+
 from deepset_mcp.mcp.server import configure_mcp_server
 
 
 class TestConfigureMcpServerBaseUrl:
     """Test the configure_mcp_server function with base_url parameter."""
 
     @patch("deepset_mcp.mcp.server.register_tools")
-    def test_configure_mcp_server_passes_base_url(self, mock_register_tools: MagicMock) -> None:
+    @pytest.mark.asyncio
+    async def test_configure_mcp_server_passes_base_url(self, mock_register_tools: MagicMock) -> None:
         """Test that configure_mcp_server passes base_url to register_tools."""
         mock_server = MagicMock()
         custom_url = "https://custom.api.example.com"
 
-        configure_mcp_server(
+        await configure_mcp_server(
             mcp_server_instance=mock_server,
             tools_to_register={"list_pipelines"},
             deepset_api_key="test-key",
@@ -32,11 +35,12 @@ def test_configure_mcp_server_passes_base_url(self, mock_register_tools: MagicMo
         assert call_args[1]["base_url"] == custom_url
 
     @patch("deepset_mcp.mcp.server.register_tools")
-    def test_configure_mcp_server_without_base_url(self, mock_register_tools: MagicMock) -> None:
+    @pytest.mark.asyncio
+    async def test_configure_mcp_server_without_base_url(self, mock_register_tools: MagicMock) -> None:
         """Test that configure_mcp_server works without base_url."""
         mock_server = MagicMock()
 
-        configure_mcp_server(
+        await configure_mcp_server(
             mcp_server_instance=mock_server,
             tools_to_register={"list_pipelines"},
             deepset_api_key="test-key",