@marcelo-ochoa/server-postgres 1.0.7 → 1.0.8

package/dist/server.js

@@ -1,19 +1,13 @@
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { McpServer, ResourceTemplate } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListResourcesRequestSchema, ListResourceTemplatesRequestSchema, ListToolsRequestSchema, ReadResourceRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
 import { z } from "zod";
 import { initializePool } from "./db.js";
 import { listResourcesHandler, readResourceHandler, callToolHandler } from "./handlers.js";
 import { tools } from "./tools.js";
-const server = new Server({
+// Create server instance
+const server = new McpServer({
     name: "postgres-server",
-    version: "1.0.7",
-}, {
-    capabilities: {
-        resources: {},
-        tools: {},
-        prompts: {}, // Add this line to indicate support for prompts
-    },
+    version: "1.0.8",
 });
 const prompts = [
     { name: "pg-query: Execute Query", description: "pg-query select * from test" },
@@ -22,30 +16,58 @@ const prompts = [
     { name: "pg-connect: Database Connection", description: "pg-connect to PostgreSQL using a connection string like host.docker.internal:5432/dbname with user name and password" },
     { name: "pg-awr: Performance Report", description: "pg-awr to generate a PostgreSQL performance report (requires pg_stat_statements extension)" }
 ];
-const PromptsListRequestSchema = z.object({
-    method: z.literal("prompts/list"),
-    params: z.object({}),
+// Register Prompts
+server.registerPrompt("postgres-prompts", {
+    description: "List available Postgres prompts"
+}, async () => ({
+    messages: [
+        {
+            role: "assistant",
+            content: {
+                type: "text",
+                text: "Available Postgres prompts:\n" + prompts.map(p => `- ${p.name}: ${p.description}`).join("\n")
+            }
+        }
+    ]
+}));
+// Register Resource Templates
+const resourceTemplate = new ResourceTemplate("postgres://{schema}/{table}/schema", {
+    list: async () => listResourcesHandler({})
 });
-server.setRequestHandler(PromptsListRequestSchema, async () => {
-    return {
-        prompts,
-    };
+server.registerResource("Table Schema", resourceTemplate, { description: "Schema information for a PostgreSQL database table including column names and data types" }, async (uri) => {
+    return readResourceHandler({ params: { uri: uri.toString() } });
 });
-server.setRequestHandler(ListResourceTemplatesRequestSchema, async () => {
-    return {
-        resourceTemplates: [
-            {
-                uriTemplate: "postgres://{schema}/{table_name}/schema",
-                name: "Table Schema",
-                description: "Schema information for a PostgreSQL database table including column names and data types",
-            },
-        ],
-    };
+// Register Tools
+tools.forEach(tool => {
+    // Basic mapping of JSON schema to Zod for simple cases
+    let inputSchema = z.object({});
+    if (tool.inputSchema && tool.inputSchema.properties) {
+        const shape = {};
+        for (const [key, prop] of Object.entries(tool.inputSchema.properties)) {
+            let field = z.any();
+            if (prop.type === "string") {
+                field = z.string();
+            }
+            if (prop.description) {
+                field = field.describe(prop.description);
+            }
+            if (tool.inputSchema.required && !tool.inputSchema.required.includes(key)) {
+                field = field.optional();
+            }
+            else if (!tool.inputSchema.required) {
+                field = field.optional();
+            }
+            shape[key] = field;
+        }
+        inputSchema = z.object(shape);
+    }
+    server.registerTool(tool.name, {
+        description: tool.description,
+        inputSchema: inputSchema
+    }, async (args) => {
+        return callToolHandler({ params: { name: tool.name, arguments: args } });
+    });
 });
-server.setRequestHandler(ListResourcesRequestSchema, listResourcesHandler);
-server.setRequestHandler(ReadResourceRequestSchema, readResourceHandler);
-server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools }));
-server.setRequestHandler(CallToolRequestSchema, callToolHandler);
 export async function runServer() {
     const args = process.argv.slice(2);
     const databaseUrl = args[0];

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@marcelo-ochoa/server-postgres",
   "mcpName": "io.github.marcelo-ochoa/postgres",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "repository": {
     "type": "git",
     "url": "https://github.com/marcelo-ochoa/servers.git",