update readme

Author: reticenceji

src/sqlite/README.md

@@ -111,6 +111,12 @@ Docker:
 docker build -t mcp/sqlite .
 ```
 
+## Test with MCP inspector
+
+```bash
+mcp dev main.py:server 
+```
+
 ## License
 
 This MCP server is licensed under the MIT License. This means you are free to use, modify, and distribute the software, subject to the terms and conditions of the MIT License. For more details, please see the LICENSE file in the project repository.

src/sqlite/src/mcp_server_sqlite/server.py

@@ -12,6 +12,7 @@
 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")
@@ -102,7 +103,6 @@
 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())
@@ -161,7 +161,6 @@ 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}")
 
@@ -216,8 +215,7 @@ 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}")
@@ -235,8 +233,7 @@ 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()),
                 )
             ],
         )
@@ -347,22 +344,19 @@ 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")]
 
@@ -374,7 +368,7 @@ async def handle_call_tool(
         except Exception as e:
             return [types.TextContent(type="text", text=f"Error: {str(e)}")]
 
-    async with stdio_server() as (read_stream, write_stream):
+    async with mcp.server.stdio.stdio_server() as (read_stream, write_stream):
         logger.info("Server running with stdio transport")
         await server.run(
             read_stream,
@@ -389,10 +383,8 @@ 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"))