simplify init

Author: ihrpr

examples/snippets/servers/__init__.py

@@ -4,9 +4,22 @@
 Each server demonstrates a single feature and can be run as a standalone server.
 """
 
+import importlib
 import sys
 from typing import Literal
 
+# Available snippet modules
+SNIPPET_MODULES = [
+    "basic_tool",
+    "basic_resource",
+    "basic_prompt",
+    "tool_progress",
+    "sampling",
+    "elicitation",
+    "completion",
+    "notifications",
+]
+
 
 def run_server(module_name: str, transport: Literal["stdio", "sse", "streamable-http"] | None = None) -> None:
     """Run a snippet server with the specified transport.
@@ -16,41 +29,13 @@ def run_server(module_name: str, transport: Literal["stdio", "sse", "streamable-
         transport: Transport to use (stdio, sse, streamable-http).
                   If None, uses first command line arg or defaults to stdio.
     """
-    # Import the specific module based on name
-    if module_name == "basic_tool":
-        from . import basic_tool
-
-        mcp = basic_tool.mcp
-    elif module_name == "basic_resource":
-        from . import basic_resource
-
-        mcp = basic_resource.mcp
-    elif module_name == "basic_prompt":
-        from . import basic_prompt
-
-        mcp = basic_prompt.mcp
-    elif module_name == "tool_progress":
-        from . import tool_progress
-
-        mcp = tool_progress.mcp
-    elif module_name == "sampling":
-        from . import sampling
+    # Validate module name
+    if module_name not in SNIPPET_MODULES:
+        raise ValueError(f"Unknown module: {module_name}. Available modules: {', '.join(SNIPPET_MODULES)}")
 
-        mcp = sampling.mcp
-    elif module_name == "elicitation":
-        from . import elicitation
-
-        mcp = elicitation.mcp
-    elif module_name == "completion":
-        from . import completion
-
-        mcp = completion.mcp
-    elif module_name == "notifications":
-        from . import notifications
-
-        mcp = notifications.mcp
-    else:
-        raise ValueError(f"Unknown module: {module_name}")
+    # Import the module dynamically
+    module = importlib.import_module(f".{module_name}", package=__name__)
+    mcp = module.mcp
 
     # Determine transport
     if transport is None:
@@ -65,42 +50,18 @@ def run_server(module_name: str, transport: Literal["stdio", "sse", "streamable-
     mcp.run(transport=transport)  # type: ignore
 
 
-# Entry points for each snippet
-def run_basic_tool():
-    """Run the basic tool example server."""
-    run_server("basic_tool")
-
-
-def run_basic_resource():
-    """Run the basic resource example server."""
-    run_server("basic_resource")
-
-
-def run_basic_prompt():
-    """Run the basic prompt example server."""
-    run_server("basic_prompt")
-
-
-def run_tool_progress():
-    """Run the tool progress example server."""
-    run_server("tool_progress")
-
-
-def run_sampling():
-    """Run the sampling example server."""
-    run_server("sampling")
-
-
-def run_elicitation():
-    """Run the elicitation example server."""
-    run_server("elicitation")
+# Create entry point functions dynamically
+def _create_run_function(module_name: str):
+    """Create a run function for a specific module."""
 
+    def run_function():
+        f"""Run the {module_name.replace('_', ' ')} example server."""
+        run_server(module_name)
 
-def run_completion():
-    """Run the completion example server."""
-    run_server("completion")
+    return run_function
 
 
-def run_notifications():
-    """Run the notifications example server."""
-    run_server("notifications")
+# Generate entry points for each snippet
+for module in SNIPPET_MODULES:
+    func_name = f"run_{module}"
+    globals()[func_name] = _create_run_function(module)