awslabs.openapi-mcp-server 0.2.10__py3-none-any.whl → 0.2.12__py3-none-any.whl

awslabs/openapi_mcp_server/__init__.py

@@ -15,7 +15,7 @@
 OpenAPI MCP Server - A server that dynamically creates MCP tools and resources from OpenAPI specifications.
 """
 
-__version__ = '0.2.10'
+__version__ = '0.2.12'
 
 
 import inspect
@@ -34,7 +34,7 @@ def get_format():
 # Set up enhanced logging format to include function name, line number, and logger name
 # Fixed the whitespace issue after log level by removing padding
 logger.add(
-    sys.stdout,
+    sys.stderr,
     format=get_format(),
     level='INFO',
 )

awslabs/openapi_mcp_server/server.py

@@ -33,7 +33,7 @@ from fastmcp.server.openapi import FastMCPOpenAPI, MCPType, RouteMap
 from typing import Any, Dict
 
 
-def create_mcp_server(config: Config) -> FastMCP:
+async def create_mcp_server_async(config: Config) -> FastMCP:
     """Create and configure the FastMCP server.
 
     Args:
@@ -221,7 +221,7 @@ def create_mcp_server(config: Config) -> FastMCP:
             prompt_manager = MCPPromptManager()
 
             # Generate prompts
-            asyncio.run(prompt_manager.generate_prompts(server, config.api_name, openapi_spec))
+            await prompt_manager.generate_prompts(server, config.api_name, openapi_spec)
 
             # Register resource handler
             prompt_manager.register_api_resource_handler(server, config.api_name, client)
@@ -292,7 +292,7 @@ def create_mcp_server(config: Config) -> FastMCP:
     if hasattr(server, 'list_tools'):
         try:
             # Use asyncio to run the async method in a synchronous context
-            tools = asyncio.run(server.list_tools())  # type: ignore
+            tools = await server.list_tools()  # type: ignore
             tool_count = len(tools)
             tool_names = [tool.get('name') for tool in tools]
 
@@ -344,6 +344,45 @@ def create_mcp_server(config: Config) -> FastMCP:
     return server
 
 
+def create_mcp_server(config: Config) -> FastMCP:
+    """Create and configure the FastMCP server (synchronous wrapper).
+
+    This is a synchronous convenience wrapper that calls
+    :func:`create_mcp_server_async` using ``asyncio.run``.
+    For asynchronous contexts, use :func:`create_mcp_server_async`
+    directly instead of this function.
+
+    Args:
+        config: Server configuration.
+
+    Returns:
+        FastMCP: The configured FastMCP server.
+
+    """
+    return asyncio.run(create_mcp_server_async(config))
+
+
+async def get_all_counts(server: FastMCP) -> tuple[int, int, int, int]:
+    """Get counts of prompts, tools, resources, and resource templates."""
+    prompts = await server.get_prompts()
+    tools = await server.get_tools()
+    resources = await server.get_resources()
+
+    # Get resource templates if available
+    resource_templates = []
+    if hasattr(server, 'get_resource_templates'):
+        try:
+            resource_templates = await server.get_resource_templates()
+        except AttributeError as e:
+            # This is expected if the method exists but is not implemented
+            logger.debug(f'get_resource_templates exists but not implemented: {e}')
+        except Exception as e:
+            # Log other unexpected errors
+            logger.warning(f'Error retrieving resource templates: {e}')
+
+    return len(prompts), len(tools), len(resources), len(resource_templates)
+
+
 def setup_signal_handlers():
     """Set up signal handlers for graceful shutdown."""
     # Store original SIGINT handler
@@ -449,7 +488,7 @@ def main():
 
     # Set up logging with loguru at specified level
     logger.remove()
-    logger.add(lambda msg: print(msg, end=''), level=args.log_level)
+    logger.add(lambda msg: print(msg, end='', file=sys.stderr), level=args.log_level)
     logger.info(f'Starting server with logging level: {args.log_level}')
 
     # Load configuration
@@ -466,25 +505,6 @@ def main():
 
     try:
         # Get counts of prompts, tools, resources, and resource templates
-        async def get_all_counts(server):
-            prompts = await server.get_prompts()
-            tools = await server.get_tools()
-            resources = await server.get_resources()
-
-            # Get resource templates if available
-            resource_templates = []
-            if hasattr(server, 'get_resource_templates'):
-                try:
-                    resource_templates = await server.get_resource_templates()
-                except AttributeError as e:
-                    # This is expected if the method exists but is not implemented
-                    logger.debug(f'get_resource_templates exists but not implemented: {e}')
-                except Exception as e:
-                    # Log other unexpected errors
-                    logger.warning(f'Error retrieving resource templates: {e}')
-
-            return len(prompts), len(tools), len(resources), len(resource_templates)
-
         prompt_count, tool_count, resource_count, resource_template_count = asyncio.run(
             get_all_counts(mcp_server)
         )

{awslabs_openapi_mcp_server-0.2.10.dist-info → awslabs_openapi_mcp_server-0.2.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.openapi-mcp-server
-Version: 0.2.10
+Version: 0.2.12
 Summary: An AWS Labs Model Context Protocol (MCP) server for OpenAPI
 Project-URL: Homepage, https://awslabs.github.io/mcp/
 Project-URL: Documentation, https://awslabs.github.io/mcp/servers/openapi-mcp-server/

{awslabs_openapi_mcp_server-0.2.10.dist-info → awslabs_openapi_mcp_server-0.2.12.dist-info}/RECORD

@@ -1,6 +1,6 @@
 awslabs/__init__.py,sha256=S-EDeDHGZzllMFhAeuJx_47fv85t2OUMOcsOisD5d_8,796
-awslabs/openapi_mcp_server/__init__.py,sha256=KvZTootEq0yM2pvTuKzQxatRs-QrhMrcmocQMEU4VlQ,2061
-awslabs/openapi_mcp_server/server.py,sha256=NVe2uoqEsP2N30gIo6WihLaWyTMFvcgq8_LA0LFCUhY,21633
+awslabs/openapi_mcp_server/__init__.py,sha256=bJ6KlrgdqcCVRFtCyjuV5FTQnnZTiqrWQk8NrHagsNY,2061
+awslabs/openapi_mcp_server/server.py,sha256=Hx7Gldi5_lhTW9PMkiPKTu3IEw9bldmIVHJ4YBp_zTo,22155
 awslabs/openapi_mcp_server/api/__init__.py,sha256=KWcmd1bH1vact1QJBfR0zFX7knWhFrBgaMEe9Tu9qi0,774
 awslabs/openapi_mcp_server/api/config.py,sha256=WT9ELzBOtZkd_DrzrkL_uuVoDibw2k94tU28CvHL32Q,9503
 awslabs/openapi_mcp_server/auth/__init__.py,sha256=wDFPpe2PmaDVvlYSdzR4saSKthgQmVPr9lwaKe3Z2RE,1109
@@ -30,9 +30,9 @@ awslabs/openapi_mcp_server/utils/http_client.py,sha256=a2wu7lfyuq1O_6ilaFQ9geFJy
 awslabs/openapi_mcp_server/utils/metrics_provider.py,sha256=1R__ZUxUrrB5KoKx4q6WYWHhaDTpFnr0UWTgZO28HVI,17731
 awslabs/openapi_mcp_server/utils/openapi.py,sha256=eRQUC5AidkyR1ALQNsR0kLXlr0Mq8ZXCduLVWUXXhuA,8802
 awslabs/openapi_mcp_server/utils/openapi_validator.py,sha256=MVJj84a8vw0BUn21pfkav6oJagDF1dSotcLK9otXzE4,9614
-awslabs_openapi_mcp_server-0.2.10.dist-info/METADATA,sha256=v6HokDxOcB0ehygaFvG6oP102rSk-zrcFbZA9bP7J28,19955
-awslabs_openapi_mcp_server-0.2.10.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-awslabs_openapi_mcp_server-0.2.10.dist-info/entry_points.txt,sha256=0BwvRNOdGm62ExFDks-9tSTWgtOdI3qmg-aemCMXQkM,86
-awslabs_openapi_mcp_server-0.2.10.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
-awslabs_openapi_mcp_server-0.2.10.dist-info/licenses/NOTICE,sha256=fG29aqEG3L4KHNXheKdyD5TdsgFh7eaaFlXu-okbf5o,94
-awslabs_openapi_mcp_server-0.2.10.dist-info/RECORD,,
+awslabs_openapi_mcp_server-0.2.12.dist-info/METADATA,sha256=Eur0k9WDVet6qouW4Lo0_LqUCsOgN9sOpLvkWk1FEJ0,19955
+awslabs_openapi_mcp_server-0.2.12.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+awslabs_openapi_mcp_server-0.2.12.dist-info/entry_points.txt,sha256=0BwvRNOdGm62ExFDks-9tSTWgtOdI3qmg-aemCMXQkM,86
+awslabs_openapi_mcp_server-0.2.12.dist-info/licenses/LICENSE,sha256=CeipvOyAZxBGUsFoaFqwkx54aPnIKEtm9a5u2uXxEws,10142
+awslabs_openapi_mcp_server-0.2.12.dist-info/licenses/NOTICE,sha256=fG29aqEG3L4KHNXheKdyD5TdsgFh7eaaFlXu-okbf5o,94
+awslabs_openapi_mcp_server-0.2.12.dist-info/RECORD,,