update sqlite's mcp version to 1.6.0

Author: reticenceji

src/sqlite/pyproject.toml

@@ -4,7 +4,7 @@ version = "0.6.2"
 description = "A simple SQLite MCP server"
 readme = "README.md"
 requires-python = ">=3.10"
-dependencies = ["mcp>=1.0.0"]
+dependencies = ["mcp>=1.6.0"]
 
 [build-system]
 requires = ["hatchling"]

src/sqlite/src/mcp_server_sqlite/server.py

@@ -4,13 +4,14 @@
 import logging
 from contextlib import closing
 from pathlib import Path
-from mcp.server.models import InitializationOptions
-import mcp.types as types
-from mcp.server import NotificationOptions, Server
-import mcp.server.stdio
 from pydantic import AnyUrl
 from typing import Any
 
+from mcp.server import InitializationOptions
+from mcp.server.lowlevel import Server, NotificationOptions
+from mcp.server.stdio import stdio_server
+import mcp.types as types
+
 # reconfigure UnicodeEncodeError prone default (i.e. windows-1252) to utf-8
 if sys.platform == "win32" and os.environ.get('PYTHONIOENCODING') is None:
     sys.stdin.reconfigure(encoding="utf-8")
@@ -101,6 +102,7 @@
 Start your first message fully in character with something like "Oh, Hey there! I see you've chosen the topic {topic}. Let's get started! 🚀"
 """
 
+
 class SqliteDatabase:
     def __init__(self, db_path: str):
         self.db_path = str(Path(db_path).expanduser())
@@ -159,6 +161,7 @@ def _execute_query(self, query: str, params: dict[str, Any] | None = None) -> li
             logger.error(f"Database error executing query: {e}")
             raise
 
+
 async def main(db_path: str):
     logger.info(f"Starting SQLite MCP Server with DB path: {db_path}")
 
@@ -213,7 +216,8 @@ async def handle_list_prompts() -> list[types.Prompt]:
 
     @server.get_prompt()
     async def handle_get_prompt(name: str, arguments: dict[str, str] | None) -> types.GetPromptResult:
-        logger.debug(f"Handling get_prompt request for {name} with args {arguments}")
+        logger.debug(
+            f"Handling get_prompt request for {name} with args {arguments}")
         if name != "mcp-demo":
             logger.error(f"Unknown prompt: {name}")
             raise ValueError(f"Unknown prompt: {name}")
@@ -231,7 +235,8 @@ async def handle_get_prompt(name: str, arguments: dict[str, str] | None) -> type
             messages=[
                 types.PromptMessage(
                     role="user",
-                    content=types.TextContent(type="text", text=prompt.strip()),
+                    content=types.TextContent(
+                        type="text", text=prompt.strip()),
                 )
             ],
         )
@@ -342,19 +347,22 @@ async def handle_call_tool(
 
             if name == "read_query":
                 if not arguments["query"].strip().upper().startswith("SELECT"):
-                    raise ValueError("Only SELECT queries are allowed for read_query")
+                    raise ValueError(
+                        "Only SELECT queries are allowed for read_query")
                 results = db._execute_query(arguments["query"])
                 return [types.TextContent(type="text", text=str(results))]
 
             elif name == "write_query":
                 if arguments["query"].strip().upper().startswith("SELECT"):
-                    raise ValueError("SELECT queries are not allowed for write_query")
+                    raise ValueError(
+                        "SELECT queries are not allowed for write_query")
                 results = db._execute_query(arguments["query"])
                 return [types.TextContent(type="text", text=str(results))]
 
             elif name == "create_table":
                 if not arguments["query"].strip().upper().startswith("CREATE TABLE"):
-                    raise ValueError("Only CREATE TABLE statements are allowed")
+                    raise ValueError(
+                        "Only CREATE TABLE statements are allowed")
                 db._execute_query(arguments["query"])
                 return [types.TextContent(type="text", text="Table created successfully")]
 
@@ -366,7 +374,7 @@ async def handle_call_tool(
         except Exception as e:
             return [types.TextContent(type="text", text=f"Error: {str(e)}")]
 
-    async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):
+    async with stdio_server() as (read_stream, write_stream):
         logger.info("Server running with stdio transport")
         await server.run(
             read_stream,
@@ -380,3 +388,14 @@ async def handle_call_tool(
                 ),
             ),
         )
+
+
+class ServerWrapper():
+    """A helper class which allows you to go with `mcp run` or `mcp dev`"""
+
+    async def run(self):
+        import asyncio
+        asyncio.run(main("test.db"))
+
+
+server = ServerWrapper()