@neverinfamous/postgres-mcp 1.1.0 → 1.3.0

package/dist/adapters/postgresql/tools/core/query.js

@@ -5,6 +5,7 @@
  */
 import { readOnly, write } from "../../../../utils/annotations.js";
 import { getToolIcons } from "../../../../utils/icons.js";
+import { parsePostgresError } from "./error-helpers.js";
 import { ReadQuerySchemaBase, ReadQuerySchema, WriteQuerySchemaBase, WriteQuerySchema, } from "../../schemas/index.js";
 import { ReadQueryOutputSchema, WriteQueryOutputSchema } from "./schemas.js";
 /**
@@ -20,28 +21,55 @@ export function createReadQueryTool(adapter) {
         annotations: readOnly("Read Query"),
         icons: getToolIcons("core", readOnly("Read Query")),
         handler: async (params, _context) => {
-            const { sql, params: queryParams, transactionId, } = ReadQuerySchema.parse(params);
-            let result;
-            if (transactionId !== undefined) {
-                const client = adapter.getTransactionConnection(transactionId);
-                if (!client) {
-                    throw new Error(`Invalid or expired transactionId: ${transactionId}. Use pg_transaction_begin to start a new transaction.`);
+            try {
+                const { sql, params: queryParams, transactionId, } = ReadQuerySchema.parse(params);
+                let result;
+                if (transactionId !== undefined) {
+                    const client = adapter.getTransactionConnection(transactionId);
+                    if (!client) {
+                        return {
+                            success: false,
+                            error: `Invalid or expired transactionId: ${transactionId}. Use pg_transaction_begin to start a new transaction.`,
+                        };
+                    }
+                    try {
+                        result = await adapter.executeOnConnection(client, sql, queryParams);
+                    }
+                    catch (error) {
+                        const parsed = parsePostgresError(error, {
+                            tool: "pg_read_query",
+                            sql,
+                        });
+                        return { success: false, error: parsed.message };
+                    }
                 }
-                result = await adapter.executeOnConnection(client, sql, queryParams);
+                else {
+                    try {
+                        result = await adapter.executeReadQuery(sql, queryParams);
+                    }
+                    catch (error) {
+                        const parsed = parsePostgresError(error, {
+                            tool: "pg_read_query",
+                            sql,
+                        });
+                        return { success: false, error: parsed.message };
+                    }
+                }
+                return {
+                    rows: result.rows,
+                    rowCount: result.rows?.length ?? 0,
+                    // Include column metadata if available
+                    fields: result.fields?.map((f) => ({
+                        name: f.name,
+                        dataTypeID: f.dataTypeID,
+                    })),
+                    executionTimeMs: result.executionTimeMs,
+                };
             }
-            else {
-                result = await adapter.executeReadQuery(sql, queryParams);
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                return { success: false, error: message };
             }
-            return {
-                rows: result.rows,
-                rowCount: result.rows?.length ?? 0,
-                // Include column metadata if available
-                fields: result.fields?.map((f) => ({
-                    name: f.name,
-                    dataTypeID: f.dataTypeID,
-                })),
-                executionTimeMs: result.executionTimeMs,
-            };
         },
     };
 }
@@ -58,32 +86,62 @@ export function createWriteQueryTool(adapter) {
         annotations: write("Write Query"),
         icons: getToolIcons("core", write("Write Query")),
         handler: async (params, _context) => {
-            const { sql, params: queryParams, transactionId, } = WriteQuerySchema.parse(params);
-            // Block SELECT statements - use pg_read_query instead
-            const trimmedUpper = sql.trim().toUpperCase();
-            if (trimmedUpper.startsWith("SELECT")) {
-                throw new Error("pg_write_query is for INSERT/UPDATE/DELETE only. Use pg_read_query for SELECT statements.");
-            }
-            let result;
-            if (transactionId !== undefined) {
-                const client = adapter.getTransactionConnection(transactionId);
-                if (!client) {
-                    throw new Error(`Invalid or expired transactionId: ${transactionId}. Use pg_transaction_begin to start a new transaction.`);
+            try {
+                const { sql, params: queryParams, transactionId, } = WriteQuerySchema.parse(params);
+                // Block SELECT statements - use pg_read_query instead
+                const trimmedUpper = sql.trim().toUpperCase();
+                if (trimmedUpper.startsWith("SELECT")) {
+                    return {
+                        success: false,
+                        error: "pg_write_query is for INSERT/UPDATE/DELETE only. Use pg_read_query for SELECT statements.",
+                    };
+                }
+                let result;
+                if (transactionId !== undefined) {
+                    const client = adapter.getTransactionConnection(transactionId);
+                    if (!client) {
+                        return {
+                            success: false,
+                            error: `Invalid or expired transactionId: ${transactionId}. Use pg_transaction_begin to start a new transaction.`,
+                        };
+                    }
+                    try {
+                        result = await adapter.executeOnConnection(client, sql, queryParams);
+                    }
+                    catch (error) {
+                        const parsed = parsePostgresError(error, {
+                            tool: "pg_write_query",
+                            sql,
+                        });
+                        return { success: false, error: parsed.message };
+                    }
+                }
+                else {
+                    try {
+                        result = await adapter.executeWriteQuery(sql, queryParams);
+                    }
+                    catch (error) {
+                        const parsed = parsePostgresError(error, {
+                            tool: "pg_write_query",
+                            sql,
+                        });
+                        return { success: false, error: parsed.message };
+                    }
                 }
-                result = await adapter.executeOnConnection(client, sql, queryParams);
+                return {
+                    rowsAffected: result.rowsAffected ?? 0,
+                    affectedRows: result.rowsAffected ?? 0, // Alias for common API naming
+                    rowCount: result.rowsAffected ?? 0, // Alias for consistency
+                    command: result.command,
+                    executionTimeMs: result.executionTimeMs,
+                    // Include returned rows when using RETURNING clause
+                    ...(result.rows && result.rows.length > 0 && { rows: result.rows }),
+                };
             }
-            else {
-                result = await adapter.executeWriteQuery(sql, queryParams);
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                return { success: false, error: message };
             }
-            return {
-                rowsAffected: result.rowsAffected,
-                affectedRows: result.rowsAffected, // Alias for common API naming
-                rowCount: result.rowsAffected, // Alias for consistency
-                command: result.command,
-                executionTimeMs: result.executionTimeMs,
-                // Include returned rows when using RETURNING clause
-                ...(result.rows && result.rows.length > 0 && { rows: result.rows }),
-            };
         },
     };
 }

package/dist/adapters/postgresql/tools/core/query.js.map

@@ -1 +1 @@
-{"version":3,"file":"query.js","sourceRoot":"","sources":["../../../../../src/adapters/postgresql/tools/core/query.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAE7E;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAwB;IAC1D,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,WAAW,EACT,yHAAyH;QAC3H,KAAK,EAAE,MAAM;QACb,WAAW,EAAE,mBAAmB,EAAE,gDAAgD;QAClF,YAAY,EAAE,qBAAqB;QACnC,WAAW,EAAE,QAAQ,CAAC,YAAY,CAAC;QACnC,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,MAAM,EACJ,GAAG,EACH,MAAM,EAAE,WAAW,EACnB,aAAa,GACd,GAAG,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAElC,IAAI,MAAM,CAAC;YACX,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,MAAM,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;gBAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CACb,qCAAqC,aAAa,wDAAwD,CAC3G,CAAC;gBACJ,CAAC;gBACD,MAAM,GAAG,MAAM,OAAO,CAAC,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,WAAW,CAAC,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAC5D,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,MAAM,CAAC,IAAI;gBACjB,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;gBAClC,uCAAuC;gBACvC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACjC,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;iBACzB,CAAC,CAAC;gBACH,eAAe,EAAE,MAAM,CAAC,eAAe;aACxC,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAAwB;IAC3D,OAAO;QACL,IAAI,EAAE,gBAAgB;QACtB,WAAW,EACT,qIAAqI;QACvI,KAAK,EAAE,MAAM;QACb,WAAW,EAAE,oBAAoB,EAAE,gDAAgD;QACnF,YAAY,EAAE,sBAAsB;QACpC,WAAW,EAAE,KAAK,CAAC,aAAa,CAAC;QACjC,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;QACjD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,MAAM,EACJ,GAAG,EACH,MAAM,EAAE,WAAW,EACnB,aAAa,GACd,GAAG,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAEnC,sDAAsD;YACtD,MAAM,YAAY,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC9C,IAAI,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACtC,MAAM,IAAI,KAAK,CACb,2FAA2F,CAC5F,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC;YACX,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,MAAM,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;gBAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CACb,qCAAqC,aAAa,wDAAwD,CAC3G,CAAC;gBACJ,CAAC;gBACD,MAAM,GAAG,MAAM,OAAO,CAAC,mBAAmB,CAAC,MAAM,EAAE,GAAG,EAAE,WAAW,CAAC,CAAC;YACvE,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,MAAM,OAAO,CAAC,iBAAiB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO;gBACL,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,8BAA8B;gBACjE,QAAQ,EAAE,MAAM,CAAC,YAAY,EAAE,wBAAwB;gBACvD,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,eAAe,EAAE,MAAM,CAAC,eAAe;gBACvC,oDAAoD;gBACpD,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;aACpE,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"query.js","sourceRoot":"","sources":["../../../../../src/adapters/postgresql/tools/core/query.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,kCAAkC,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,oBAAoB,EACpB,gBAAgB,GACjB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,qBAAqB,EAAE,sBAAsB,EAAE,MAAM,cAAc,CAAC;AAE7E;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,OAAwB;IAC1D,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,WAAW,EACT,yHAAyH;QAC3H,KAAK,EAAE,MAAM;QACb,WAAW,EAAE,mBAAmB,EAAE,gDAAgD;QAClF,YAAY,EAAE,qBAAqB;QACnC,WAAW,EAAE,QAAQ,CAAC,YAAY,CAAC;QACnC,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,IAAI,CAAC;gBACH,MAAM,EACJ,GAAG,EACH,MAAM,EAAE,WAAW,EACnB,aAAa,GACd,GAAG,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAElC,IAAI,MAAM,CAAC;gBACX,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;oBAChC,MAAM,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;oBAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,OAAO;4BACL,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,qCAAqC,aAAa,wDAAwD;yBAClH,CAAC;oBACJ,CAAC;oBACD,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,OAAO,CAAC,mBAAmB,CACxC,MAAM,EACN,GAAG,EACH,WAAW,CACZ,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAc,EAAE,CAAC;wBACxB,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE;4BACvC,IAAI,EAAE,eAAe;4BACrB,GAAG;yBACJ,CAAC,CAAC;wBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAgB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;oBAC5D,CAAC;oBAAC,OAAO,KAAc,EAAE,CAAC;wBACxB,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE;4BACvC,IAAI,EAAE,eAAe;4BACrB,GAAG;yBACJ,CAAC,CAAC;wBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnD,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC;oBAClC,uCAAuC;oBACvC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBACjC,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB,CAAC,CAAC;oBACH,eAAe,EAAE,MAAM,CAAC,eAAe;iBACxC,CAAC;YACJ,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC5C,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAAwB;IAC3D,OAAO;QACL,IAAI,EAAE,gBAAgB;QACtB,WAAW,EACT,qIAAqI;QACvI,KAAK,EAAE,MAAM;QACb,WAAW,EAAE,oBAAoB,EAAE,gDAAgD;QACnF,YAAY,EAAE,sBAAsB;QACpC,WAAW,EAAE,KAAK,CAAC,aAAa,CAAC;QACjC,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;QACjD,OAAO,EAAE,KAAK,EAAE,MAAe,EAAE,QAAwB,EAAE,EAAE;YAC3D,IAAI,CAAC;gBACH,MAAM,EACJ,GAAG,EACH,MAAM,EAAE,WAAW,EACnB,aAAa,GACd,GAAG,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEnC,sDAAsD;gBACtD,MAAM,YAAY,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;gBAC9C,IAAI,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACtC,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EACH,2FAA2F;qBAC9F,CAAC;gBACJ,CAAC;gBAED,IAAI,MAAM,CAAC;gBACX,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;oBAChC,MAAM,MAAM,GAAG,OAAO,CAAC,wBAAwB,CAAC,aAAa,CAAC,CAAC;oBAC/D,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,OAAO;4BACL,OAAO,EAAE,KAAK;4BACd,KAAK,EAAE,qCAAqC,aAAa,wDAAwD;yBAClH,CAAC;oBACJ,CAAC;oBACD,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,OAAO,CAAC,mBAAmB,CACxC,MAAM,EACN,GAAG,EACH,WAAW,CACZ,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAc,EAAE,CAAC;wBACxB,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE;4BACvC,IAAI,EAAE,gBAAgB;4BACtB,GAAG;yBACJ,CAAC,CAAC;wBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnD,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,OAAO,CAAC,iBAAiB,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;oBAC7D,CAAC;oBAAC,OAAO,KAAc,EAAE,CAAC;wBACxB,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,EAAE;4BACvC,IAAI,EAAE,gBAAgB;4BACtB,GAAG;yBACJ,CAAC,CAAC;wBACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnD,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC;oBACtC,YAAY,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC,EAAE,8BAA8B;oBACtE,QAAQ,EAAE,MAAM,CAAC,YAAY,IAAI,CAAC,EAAE,wBAAwB;oBAC5D,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,eAAe,EAAE,MAAM,CAAC,eAAe;oBACvC,oDAAoD;oBACpD,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;iBACpE,CAAC;YACJ,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;YAC5C,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/adapters/postgresql/tools/core/schemas.d.ts

@@ -96,14 +96,25 @@ export declare const ObjectDetailsSchema: z.ZodPipe<z.ZodPipe<z.ZodTransform<unk
     type?: "function" | "table" | "view" | "materialized_view" | "partitioned_table" | "index" | "sequence" | undefined;
     objectType?: "function" | "table" | "view" | "materialized_view" | "partitioned_table" | "index" | "sequence" | undefined;
 }>>;
+export declare const AnalyzeDbHealthSchemaBase: z.ZodObject<{
+    includeIndexes: z.ZodOptional<z.ZodBoolean>;
+    includeVacuum: z.ZodOptional<z.ZodBoolean>;
+    includeConnections: z.ZodOptional<z.ZodBoolean>;
+}, z.core.$strip>;
 export declare const AnalyzeDbHealthSchema: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodObject<{
     includeIndexes: z.ZodOptional<z.ZodBoolean>;
     includeVacuum: z.ZodOptional<z.ZodBoolean>;
     includeConnections: z.ZodOptional<z.ZodBoolean>;
 }, z.core.$strip>>;
+export declare const AnalyzeWorkloadIndexesSchemaBase: z.ZodObject<{
+    topQueries: z.ZodOptional<z.ZodNumber>;
+    minCalls: z.ZodOptional<z.ZodNumber>;
+    queryPreviewLength: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
 export declare const AnalyzeWorkloadIndexesSchema: z.ZodPipe<z.ZodTransform<unknown, unknown>, z.ZodObject<{
     topQueries: z.ZodOptional<z.ZodNumber>;
     minCalls: z.ZodOptional<z.ZodNumber>;
+    queryPreviewLength: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>>;
 export declare const AnalyzeQueryIndexesSchemaBase: z.ZodObject<{
     sql: z.ZodOptional<z.ZodString>;
@@ -133,18 +144,20 @@ export declare const AnalyzeQueryIndexesSchema: z.ZodPipe<z.ZodObject<{
     verbosity?: "full" | "summary" | undefined;
 }>>;
 export declare const ReadQueryOutputSchema: z.ZodObject<{
-    rows: z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    rowCount: z.ZodNumber;
+    success: z.ZodOptional<z.ZodBoolean>;
+    rows: z.ZodOptional<z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
+    rowCount: z.ZodOptional<z.ZodNumber>;
     fields: z.ZodOptional<z.ZodArray<z.ZodObject<{
         name: z.ZodString;
         dataTypeID: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>>>;
     executionTimeMs: z.ZodOptional<z.ZodNumber>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const WriteQueryOutputSchema: z.ZodObject<{
     success: z.ZodOptional<z.ZodBoolean>;
     operation: z.ZodOptional<z.ZodString>;
-    rowsAffected: z.ZodNumber;
+    rowsAffected: z.ZodOptional<z.ZodNumber>;
     affectedRows: z.ZodOptional<z.ZodNumber>;
     rowCount: z.ZodOptional<z.ZodNumber>;
     insertedCount: z.ZodOptional<z.ZodNumber>;
@@ -153,6 +166,7 @@ export declare const WriteQueryOutputSchema: z.ZodObject<{
     rows: z.ZodOptional<z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     sql: z.ZodOptional<z.ZodString>;
     hint: z.ZodOptional<z.ZodString>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const TableListOutputSchema: z.ZodObject<{
     tables: z.ZodArray<z.ZodObject<{
@@ -168,37 +182,42 @@ export declare const TableListOutputSchema: z.ZodObject<{
     hint: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const TableDescribeOutputSchema: z.ZodObject<{
-    name: z.ZodString;
-    schema: z.ZodString;
-    type: z.ZodString;
-    columns: z.ZodArray<z.ZodObject<{
+    success: z.ZodOptional<z.ZodBoolean>;
+    name: z.ZodOptional<z.ZodString>;
+    schema: z.ZodOptional<z.ZodString>;
+    type: z.ZodOptional<z.ZodString>;
+    columns: z.ZodOptional<z.ZodArray<z.ZodObject<{
         name: z.ZodString;
         type: z.ZodString;
         nullable: z.ZodBoolean;
         default: z.ZodOptional<z.ZodString>;
         primaryKey: z.ZodOptional<z.ZodBoolean>;
-    }, z.core.$strip>>;
-    primaryKey: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    }, z.core.$strip>>>;
+    primaryKey: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodString>>>;
     foreignKeys: z.ZodOptional<z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     indexes: z.ZodOptional<z.ZodArray<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     rowCount: z.ZodOptional<z.ZodNumber>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const TableOperationOutputSchema: z.ZodObject<{
-    success: z.ZodBoolean;
+    success: z.ZodOptional<z.ZodBoolean>;
     table: z.ZodOptional<z.ZodString>;
     dropped: z.ZodOptional<z.ZodString>;
     existed: z.ZodOptional<z.ZodBoolean>;
     sql: z.ZodOptional<z.ZodString>;
     compositePrimaryKey: z.ZodOptional<z.ZodArray<z.ZodString>>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const TruncateOutputSchema: z.ZodObject<{
     success: z.ZodBoolean;
-    table: z.ZodString;
-    cascade: z.ZodBoolean;
-    restartIdentity: z.ZodBoolean;
+    table: z.ZodOptional<z.ZodString>;
+    cascade: z.ZodOptional<z.ZodBoolean>;
+    restartIdentity: z.ZodOptional<z.ZodBoolean>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const IndexListOutputSchema: z.ZodObject<{
-    indexes: z.ZodArray<z.ZodObject<{
+    success: z.ZodOptional<z.ZodBoolean>;
+    indexes: z.ZodOptional<z.ZodArray<z.ZodObject<{
         name: z.ZodString;
         table: z.ZodOptional<z.ZodString>;
         tableName: z.ZodOptional<z.ZodString>;
@@ -208,19 +227,21 @@ export declare const IndexListOutputSchema: z.ZodObject<{
         type: z.ZodOptional<z.ZodString>;
         unique: z.ZodOptional<z.ZodBoolean>;
         columns: z.ZodOptional<z.ZodArray<z.ZodString>>;
-    }, z.core.$strip>>;
-    count: z.ZodNumber;
+    }, z.core.$strip>>>;
+    count: z.ZodOptional<z.ZodNumber>;
     totalCount: z.ZodOptional<z.ZodNumber>;
     truncated: z.ZodOptional<z.ZodBoolean>;
     hint: z.ZodOptional<z.ZodString>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const IndexOperationOutputSchema: z.ZodObject<{
-    success: z.ZodBoolean;
+    success: z.ZodOptional<z.ZodBoolean>;
     message: z.ZodOptional<z.ZodString>;
     index: z.ZodOptional<z.ZodString>;
     table: z.ZodOptional<z.ZodString>;
     sql: z.ZodOptional<z.ZodString>;
     hint: z.ZodOptional<z.ZodString>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const ObjectListOutputSchema: z.ZodObject<{
     objects: z.ZodArray<z.ZodObject<{
@@ -235,11 +256,13 @@ export declare const ObjectListOutputSchema: z.ZodObject<{
     hint: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const ObjectDetailsOutputSchema: z.ZodObject<{
-    name: z.ZodString;
-    schema: z.ZodString;
-    type: z.ZodString;
+    success: z.ZodOptional<z.ZodBoolean>;
+    name: z.ZodOptional<z.ZodString>;
+    schema: z.ZodOptional<z.ZodString>;
+    type: z.ZodOptional<z.ZodString>;
     owner: z.ZodOptional<z.ZodString>;
     details: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const ExtensionListOutputSchema: z.ZodObject<{
     extensions: z.ZodArray<z.ZodObject<{
@@ -284,16 +307,20 @@ export declare const QueryIndexAnalysisOutputSchema: z.ZodObject<{
     error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const CountOutputSchema: z.ZodObject<{
-    count: z.ZodNumber;
+    success: z.ZodOptional<z.ZodBoolean>;
+    count: z.ZodOptional<z.ZodNumber>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 export declare const ExistsOutputSchema: z.ZodObject<{
-    exists: z.ZodBoolean;
-    table: z.ZodString;
-    mode: z.ZodEnum<{
+    success: z.ZodOptional<z.ZodBoolean>;
+    exists: z.ZodOptional<z.ZodBoolean>;
+    table: z.ZodOptional<z.ZodString>;
+    mode: z.ZodOptional<z.ZodEnum<{
         filtered: "filtered";
         any_rows: "any_rows";
-    }>;
+    }>>;
     where: z.ZodOptional<z.ZodString>;
     hint: z.ZodOptional<z.ZodString>;
+    error: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
 //# sourceMappingURL=schemas.d.ts.map

package/dist/adapters/postgresql/tools/core/schemas.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../../../../../src/adapters/postgresql/tools/core/schemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAkCxB,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;iBA4BhC,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;kBAG7B,CAAC;AAuEF,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;iBAiClC,CAAC;AAGH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAS5B,CAAC;AAEL,eAAO,MAAM,qBAAqB;;;;kBAgBjC,CAAC;AAEF,eAAO,MAAM,4BAA4B;;;kBASxC,CAAC;AAGF,eAAO,MAAM,6BAA6B;;;;;;;;iBAaxC,CAAC;AAGH,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;GAOlC,CAAC;AAWL,eAAO,MAAM,qBAAqB;;;;;;;;iBAOhC,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;iBAejC,CAAC;AAYH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;iBAMhC,CAAC;AAYH,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;iBAepC,CAAC;AAGH,eAAO,MAAM,0BAA0B;;;;;;;iBAUrC,CAAC;AAGH,eAAO,MAAM,oBAAoB;;;;;iBAK/B,CAAC;AAgBH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;iBAMhC,CAAC;AAGH,eAAO,MAAM,0BAA0B;;;;;;;iBAOrC,CAAC;AAWH,eAAO,MAAM,sBAAsB;;;;;;;;;;;iBAMjC,CAAC;AAGH,eAAO,MAAM,yBAAyB;;;;;;iBASpC,CAAC;AAWH,eAAO,MAAM,yBAAyB;;;;;;;;iBAGpC,CAAC;AAWH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;iBA+BrC,CAAC;AAGH,eAAO,MAAM,gCAAgC;;;;iBAM3C,CAAC;AAGH,eAAO,MAAM,8BAA8B;;;;;;;;;;iBAmBzC,CAAC;AAGH,eAAO,MAAM,iBAAiB;;iBAE5B,CAAC;AAGH,eAAO,MAAM,kBAAkB;;;;;;;;;iBAM7B,CAAC"}
+{"version":3,"file":"schemas.d.ts","sourceRoot":"","sources":["../../../../../src/adapters/postgresql/tools/core/schemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAkCxB,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;iBA4BhC,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;kBAG7B,CAAC;AAuEF,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;iBAiClC,CAAC;AAGH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAS5B,CAAC;AAEL,eAAO,MAAM,yBAAyB;;;;iBAapC,CAAC;AAEH,eAAO,MAAM,qBAAqB;;;;kBAGjC,CAAC;AAEF,eAAO,MAAM,gCAAgC;;;;iBAY3C,CAAC;AAEH,eAAO,MAAM,4BAA4B;;;;kBAGxC,CAAC;AAGF,eAAO,MAAM,6BAA6B;;;;;;;;iBAaxC,CAAC;AAGH,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;GAOlC,CAAC;AAWL,eAAO,MAAM,qBAAqB;;;;;;;;;;iBAUhC,CAAC;AAGH,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;iBAgBjC,CAAC;AAYH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;iBAMhC,CAAC;AAYH,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;iBAqBpC,CAAC;AAGH,eAAO,MAAM,0BAA0B;;;;;;;;iBAWrC,CAAC;AAGH,eAAO,MAAM,oBAAoB;;;;;;iBAS/B,CAAC;AAgBH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;iBAQhC,CAAC;AAGH,eAAO,MAAM,0BAA0B;;;;;;;;iBAQrC,CAAC;AAWH,eAAO,MAAM,sBAAsB;;;;;;;;;;;iBAMjC,CAAC;AAGH,eAAO,MAAM,yBAAyB;;;;;;;;iBAWpC,CAAC;AAWH,eAAO,MAAM,yBAAyB;;;;;;;;iBAGpC,CAAC;AAWH,eAAO,MAAM,0BAA0B;;;;;;;;;;;;;;;;iBA+BrC,CAAC;AAGH,eAAO,MAAM,gCAAgC;;;;iBAM3C,CAAC;AAGH,eAAO,MAAM,8BAA8B;;;;;;;;;;iBAmBzC,CAAC;AAGH,eAAO,MAAM,iBAAiB;;;;iBAI5B,CAAC;AAGH,eAAO,MAAM,kBAAkB;;;;;;;;;;;iBAQ7B,CAAC"}

package/dist/adapters/postgresql/tools/core/schemas.js

@@ -176,7 +176,7 @@ export const ObjectDetailsSchema = z
     .refine((data) => data.name !== "", {
     message: "name (or object/objectName/table alias) is required",
 });
-export const AnalyzeDbHealthSchema = z.preprocess(defaultToEmpty, z.object({
+export const AnalyzeDbHealthSchemaBase = z.object({
     includeIndexes: z
         .boolean()
         .optional()
@@ -189,14 +189,20 @@ export const AnalyzeDbHealthSchema = z.preprocess(defaultToEmpty, z.object({
         .boolean()
         .optional()
         .describe("Include connection stats (default: true)"),
-}));
-export const AnalyzeWorkloadIndexesSchema = z.preprocess(defaultToEmpty, z.object({
+});
+export const AnalyzeDbHealthSchema = z.preprocess(defaultToEmpty, AnalyzeDbHealthSchemaBase);
+export const AnalyzeWorkloadIndexesSchemaBase = z.object({
     topQueries: z
         .number()
         .optional()
         .describe("Number of top queries to analyze (default: 20)"),
     minCalls: z.number().optional().describe("Minimum call count threshold"),
-}));
+    queryPreviewLength: z
+        .number()
+        .optional()
+        .describe("Maximum characters for query preview (default: 200). Truncated queries end with '…'"),
+});
+export const AnalyzeWorkloadIndexesSchema = z.preprocess(defaultToEmpty, AnalyzeWorkloadIndexesSchemaBase);
 // Base schema for MCP visibility - exported so tool can use it for inputSchema
 export const AnalyzeQueryIndexesSchemaBase = z.object({
     sql: z
@@ -226,18 +232,21 @@ const FieldSchema = z.object({
 });
 // Output schema for pg_read_query
 export const ReadQueryOutputSchema = z.object({
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
     rows: z
         .array(z.record(z.string(), z.unknown()))
+        .optional()
         .describe("Query result rows"),
-    rowCount: z.number().describe("Number of rows returned"),
+    rowCount: z.number().optional().describe("Number of rows returned"),
     fields: z.array(FieldSchema).optional().describe("Column metadata"),
     executionTimeMs: z.number().optional().describe("Query execution time in ms"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Output schema for pg_write_query, pg_upsert, pg_batch_insert
 export const WriteQueryOutputSchema = z.object({
     success: z.boolean().optional().describe("Whether the operation succeeded"),
     operation: z.string().optional().describe("Operation type (insert/update)"),
-    rowsAffected: z.number().describe("Number of rows affected"),
+    rowsAffected: z.number().optional().describe("Number of rows affected"),
     affectedRows: z.number().optional().describe("Alias for rowsAffected"),
     rowCount: z.number().optional().describe("Alias for rowsAffected"),
     insertedCount: z.number().optional().describe("Rows inserted (batch insert)"),
@@ -249,6 +258,7 @@ export const WriteQueryOutputSchema = z.object({
         .describe("Returned rows (RETURNING clause)"),
     sql: z.string().optional().describe("Generated SQL statement"),
     hint: z.string().optional().describe("Additional information"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Table info schema for list tables
 const TableInfoSchema = z.object({
@@ -276,11 +286,16 @@ const ColumnInfoSchema = z.object({
 });
 // Output schema for pg_describe_table
 export const TableDescribeOutputSchema = z.object({
-    name: z.string().describe("Table name"),
-    schema: z.string().describe("Schema name"),
-    type: z.string().describe("Object type"),
-    columns: z.array(ColumnInfoSchema).describe("Column definitions"),
-    primaryKey: z.array(z.string()).optional().describe("Primary key columns"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
+    name: z.string().optional().describe("Table name"),
+    schema: z.string().optional().describe("Schema name"),
+    type: z.string().optional().describe("Object type"),
+    columns: z.array(ColumnInfoSchema).optional().describe("Column definitions"),
+    primaryKey: z
+        .array(z.string())
+        .nullable()
+        .optional()
+        .describe("Primary key columns"),
     foreignKeys: z
         .array(z.record(z.string(), z.unknown()))
         .optional()
@@ -290,10 +305,11 @@ export const TableDescribeOutputSchema = z.object({
         .optional()
         .describe("Index definitions"),
     rowCount: z.number().optional().describe("Estimated row count"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Output schema for pg_create_table, pg_drop_table
 export const TableOperationOutputSchema = z.object({
-    success: z.boolean().describe("Whether the operation succeeded"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
     table: z.string().optional().describe("Qualified table name"),
     dropped: z.string().optional().describe("Dropped table name (drop only)"),
     existed: z.boolean().optional().describe("Whether table existed before drop"),
@@ -302,13 +318,18 @@ export const TableOperationOutputSchema = z.object({
         .array(z.string())
         .optional()
         .describe("Composite PK columns"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Output schema for pg_truncate
 export const TruncateOutputSchema = z.object({
     success: z.boolean().describe("Whether the operation succeeded"),
-    table: z.string().describe("Truncated table"),
-    cascade: z.boolean().describe("Whether CASCADE was used"),
-    restartIdentity: z.boolean().describe("Whether identity was restarted"),
+    table: z.string().optional().describe("Truncated table"),
+    cascade: z.boolean().optional().describe("Whether CASCADE was used"),
+    restartIdentity: z
+        .boolean()
+        .optional()
+        .describe("Whether identity was restarted"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Index info schema
 const IndexInfoSchema = z.object({
@@ -324,20 +345,23 @@ const IndexInfoSchema = z.object({
 });
 // Output schema for pg_get_indexes
 export const IndexListOutputSchema = z.object({
-    indexes: z.array(IndexInfoSchema).describe("List of indexes"),
-    count: z.number().describe("Number of indexes"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
+    indexes: z.array(IndexInfoSchema).optional().describe("List of indexes"),
+    count: z.number().optional().describe("Number of indexes"),
     totalCount: z.number().optional().describe("Total count before truncation"),
     truncated: z.boolean().optional().describe("Whether results were truncated"),
     hint: z.string().optional().describe("Additional information"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Output schema for pg_create_index, pg_drop_index
 export const IndexOperationOutputSchema = z.object({
-    success: z.boolean().describe("Whether the operation succeeded"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
     message: z.string().optional().describe("Result message"),
     index: z.string().optional().describe("Index name"),
     table: z.string().optional().describe("Table name"),
     sql: z.string().optional().describe("Generated SQL"),
     hint: z.string().optional().describe("Additional information"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Database object schema
 const DatabaseObjectSchema = z.object({
@@ -356,14 +380,16 @@ export const ObjectListOutputSchema = z.object({
 });
 // Output schema for pg_object_details - flexible due to different object types
 export const ObjectDetailsOutputSchema = z.object({
-    name: z.string().describe("Object name"),
-    schema: z.string().describe("Schema name"),
-    type: z.string().describe("Object type"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
+    name: z.string().optional().describe("Object name"),
+    schema: z.string().optional().describe("Schema name"),
+    type: z.string().optional().describe("Object type"),
     owner: z.string().optional().describe("Object owner"),
     details: z
         .record(z.string(), z.unknown())
         .optional()
         .describe("Type-specific details"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Extension info schema
 const ExtensionInfoSchema = z.object({
@@ -446,14 +472,18 @@ export const QueryIndexAnalysisOutputSchema = z.object({
 });
 // Output schema for pg_count
 export const CountOutputSchema = z.object({
-    count: z.number().describe("Row count"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
+    count: z.number().optional().describe("Row count"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 // Output schema for pg_exists
 export const ExistsOutputSchema = z.object({
-    exists: z.boolean().describe("Whether rows exist"),
-    table: z.string().describe("Table checked"),
-    mode: z.enum(["filtered", "any_rows"]).describe("Check mode"),
+    success: z.boolean().optional().describe("Whether the operation succeeded"),
+    exists: z.boolean().optional().describe("Whether rows exist"),
+    table: z.string().optional().describe("Table checked"),
+    mode: z.enum(["filtered", "any_rows"]).optional().describe("Check mode"),
     where: z.string().optional().describe("WHERE clause used (filtered mode)"),
     hint: z.string().optional().describe("Clarifying hint (any_rows mode)"),
+    error: z.string().optional().describe("Error message if operation failed"),
 });
 //# sourceMappingURL=schemas.js.map