Sun-ZhenXing/mcp-template-python
1
0
Fork 0
mirror of https://github.com/Sun-ZhenXing/mcp-template-python.git synced 2026-02-04 10:13:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add /ws

Browse Source
This commit is contained in:
Sun-ZhenXing
2025-07-21 00:09:53 +08:00
parent bcb5994356
commit 48f2efef7a
10 changed files with 347 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/mcp_template_python/__main__.py
View File

@@ -2,11 +2,12 @@ import argparse
import sys
from .__about__ import __module_name__, __version__
from .app import MCP_MAP
from .config import settings
def main():
from .app import MCP_MAP
parser = argparse.ArgumentParser(description="MCP Server")
parser.add_argument(
@@ -14,6 +15,13 @@ def main():
action="store_true",
help="Run the server with STDIO (default: False)",
)
parser.add_argument(
"--mcp",
type=str,
default=settings.default_mcp,
choices=list(MCP_MAP.keys()),
help=f"Select the MCP to run in STDIO mode (default: {settings.default_mcp})",
)
parser.add_argument(
"--host",
default=settings.default_host,
@@ -44,7 +52,10 @@ def main():
sys.exit(0)
if args.stdio:
mcp = MCP_MAP[settings.default_mcp]
mcp = MCP_MAP.get(args.mcp)
if mcp is None:
print(f"Error: MCP '{args.mcp}' not found.")
sys.exit(1)
mcp.run()
else:
import uvicorn