update sse arg

Author: sherlin

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xiyan_mcp_server"
-version = "0.1.4"
+version = "0.1.5.dev0"
 description = "A Model Context Protocol (MCP) server that utilizes XiyanSQL with databases. This server enables AI assistants to list tables, read data, and execute natural language queries"
 readme = "README.md"
 requires-python = ">=3.11"

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name='xiyan_mcp_server',  # 包的名字
-    version='0.1.2',  # 版本号
+    version='0.1.5.dev0',  # 版本号
     packages=find_packages(),  # 自动找到项目中的包
     install_requires=[  # 必要的包依赖
         # 'numpy',  # 示例：若有依赖包，将其列在此

src/xiyan_mcp_server/__init__.py

@@ -1,8 +1,3 @@
-from . import server
-
-def main():
-    """Main entry point for the package."""
-    server.main()  # 修改为调用server.py中的main函数
 
 # Expose important items at package level
-__all__ = ['main', 'server']
+__all__ = ['server']

src/xiyan_mcp_server/__main__.py

@@ -1,4 +1,15 @@
-from . import main
+import argparse
+import sys
+
+from .server import run_mcp
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Run MCP server.")
+    parser.add_argument("-transport", type=str, default="stdio",
+                        help="Transport method: 'stdio' or 'sse'")
+    args = parser.parse_args(sys.argv[1:])  # 忽略第一个参数（模块名）
+    run_mcp(transport=args.transport)
 
 if __name__ == "__main__":
     main()

src/xiyan_mcp_server/server.py

@@ -1,5 +1,8 @@
+import argparse
 import logging
 import os
+from typing import Literal
+
 import yaml  # 添加yaml库导入
 
 from mysql.connector import connect, Error
@@ -24,6 +27,7 @@
 
 def get_yml_config():
     config_path = os.getenv("YML", os.path.join(os.path.dirname(__file__), "config_demo.yml"))
+    print(f"Load configuration from {config_path}")
     try:
         with open(config_path, 'r') as file:
             config = yaml.safe_load(file)
@@ -76,7 +80,7 @@ async def read_resource(table_name) -> str:
         raise RuntimeError(f"Database error: {str(e)}")
 
 
-def sql_gen_and_execute(db_env, query: str):
+def sql_gen_and_execute(db_env: DataBaseEnv, query: str):
     """
     Transfers the input natural language question to sql query (known as Text-to-sql) and executes it on the database.
      Args:
@@ -180,11 +184,12 @@ def get_data(query: str)-> list[TextContent]:
     res=call_xiyan(query)
     return [TextContent(type="text", text=res)]
 
-def main():
-    mcp.run()
-
 
-if __name__ == "__main__":
+def run_mcp(transport: Literal["stdio", "sse"] = "stdio"):
+    """Start the MCP server with the specified transport method.
 
-    main()
+    Args:
+        transport (str): The transport method to use, either "stdio" or "sse".
+    """
+    mcp.run(transport=transport)