@neverinfamous/postgres-mcp 1.3.0 → 2.0.0

package/dist/adapters/postgresql/tools/text.js

@@ -79,15 +79,30 @@ function createTextSearchTool(adapter) {
                 const tsvector = sanitizedCols
                     .map((c) => `coalesce(${c}, '')`)
                     .join(" || ' ' || ");
-                const limitClause = parsed.limit !== undefined && parsed.limit > 0
-                    ? ` LIMIT ${String(parsed.limit)}`
-                    : "";
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 const sql = `SELECT ${selectCols}, ts_rank_cd(to_tsvector('${cfg}', ${tsvector}), plainto_tsquery('${cfg}', $1)) as rank
                         FROM ${tableName}
                         WHERE to_tsvector('${cfg}', ${tsvector}) @@ plainto_tsquery('${cfg}', $1)
                         ORDER BY rank DESC${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.query]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -172,15 +187,30 @@ function createTextRankTool(adapter) {
                 const tsvector = sanitizedCols
                     .map((c) => `coalesce(${c}, '')`)
                     .join(" || ' ' || ");
-                const limitClause = parsed.limit !== undefined && parsed.limit > 0
-                    ? ` LIMIT ${String(parsed.limit)}`
-                    : "";
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 const sql = `SELECT ${selectCols}, ts_rank_cd(to_tsvector('${cfg}', ${tsvector}), plainto_tsquery('${cfg}', $1), ${String(norm)}) as rank
                         FROM ${tableName}
                         WHERE to_tsvector('${cfg}', ${tsvector}) @@ plainto_tsquery('${cfg}', $1)
                         ORDER BY rank DESC${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.query]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -212,8 +242,13 @@ function createTrigramSimilarityTool(adapter) {
             try {
                 const parsed = TrigramSimilaritySchema.parse(params);
                 const thresh = parsed.threshold ?? 0.3;
-                // Default limit to 100 to prevent large payloads
-                const limitVal = parsed.limit !== undefined && parsed.limit > 0 ? parsed.limit : 100;
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 // The preprocessor guarantees table is set (converts tableName → table)
                 const resolvedTable = parsed.table ?? parsed.tableName;
                 if (!resolvedTable) {
@@ -233,9 +268,20 @@ function createTrigramSimilarityTool(adapter) {
                 const sql = `SELECT ${selectCols}, similarity(${columnName}, $1) as similarity
                         FROM ${tableName}
                         WHERE similarity(${columnName}, $1) > ${String(thresh)}${additionalWhere}
-                        ORDER BY similarity DESC LIMIT ${String(limitVal)}`;
+                        ORDER BY similarity DESC${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.value]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -309,8 +355,13 @@ function createFuzzyMatchTool(adapter) {
                 }
                 const method = rawMethod;
                 const maxDist = parsed.maxDistance ?? 3;
-                // Default limit to 100 to prevent large payloads
-                const limitVal = parsed.limit !== undefined && parsed.limit > 0 ? parsed.limit : 100;
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 // The preprocessor guarantees table is set (converts tableName → table)
                 const resolvedTable = parsed.table ?? parsed.tableName;
                 if (!resolvedTable) {
@@ -329,16 +380,27 @@ function createFuzzyMatchTool(adapter) {
                     : "";
                 let sql;
                 if (method === "soundex") {
-                    sql = `SELECT ${selectCols}, soundex(${columnName}) as code FROM ${tableName} WHERE soundex(${columnName}) = soundex($1)${additionalWhere} LIMIT ${String(limitVal)}`;
+                    sql = `SELECT ${selectCols}, soundex(${columnName}) as code FROM ${tableName} WHERE soundex(${columnName}) = soundex($1)${additionalWhere}${limitClause}`;
                 }
                 else if (method === "metaphone") {
-                    sql = `SELECT ${selectCols}, metaphone(${columnName}, 10) as code FROM ${tableName} WHERE metaphone(${columnName}, 10) = metaphone($1, 10)${additionalWhere} LIMIT ${String(limitVal)}`;
+                    sql = `SELECT ${selectCols}, metaphone(${columnName}, 10) as code FROM ${tableName} WHERE metaphone(${columnName}, 10) = metaphone($1, 10)${additionalWhere}${limitClause}`;
                 }
                 else {
-                    sql = `SELECT ${selectCols}, levenshtein(${columnName}, $1) as distance FROM ${tableName} WHERE levenshtein(${columnName}, $1) <= ${String(maxDist)}${additionalWhere} ORDER BY distance LIMIT ${String(limitVal)}`;
+                    sql = `SELECT ${selectCols}, levenshtein(${columnName}, $1) as distance FROM ${tableName} WHERE levenshtein(${columnName}, $1) <= ${String(maxDist)}${additionalWhere} ORDER BY distance${limitClause}`;
                 }
                 const result = await adapter.executeQuery(sql, [parsed.value]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -386,12 +448,27 @@ function createRegexpMatchTool(adapter) {
                 const additionalWhere = parsed.where
                     ? ` AND (${sanitizeWhereClause(parsed.where)})`
                     : "";
-                // Default limit to 100 to prevent large payloads
-                const limitVal = parsed.limit !== undefined && parsed.limit > 0 ? parsed.limit : 100;
-                const limitClause = ` LIMIT ${String(limitVal)}`;
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 const sql = `SELECT ${selectCols} FROM ${tableName} WHERE ${columnName} ${op} $1${additionalWhere}${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.pattern]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -463,12 +540,27 @@ function createLikeSearchTool(adapter) {
                 const additionalWhere = parsed.where
                     ? ` AND (${sanitizeWhereClause(parsed.where)})`
                     : "";
-                // Default limit to 100 to prevent large payloads
-                const limitVal = parsed.limit !== undefined && parsed.limit > 0 ? parsed.limit : 100;
-                const limitClause = ` LIMIT ${String(limitVal)}`;
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 const sql = `SELECT ${selectCols} FROM ${tableName} WHERE ${columnName} ${op} $1${additionalWhere}${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.pattern]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -561,14 +653,29 @@ function createTextHeadlineTool(adapter) {
                 const selectCols = parsed.select !== undefined && parsed.select.length > 0
                     ? sanitizeIdentifiers(parsed.select).join(", ") + ", "
                     : "";
-                const limitClause = parsed.limit !== undefined && parsed.limit > 0
-                    ? ` LIMIT ${String(parsed.limit)}`
-                    : "";
+                // Default limit to 100 to prevent large payloads; limit: 0 means no limit
+                const limitVal = parsed.limit === 0
+                    ? null
+                    : parsed.limit !== undefined && parsed.limit > 0
+                        ? parsed.limit
+                        : 100;
+                const limitClause = limitVal !== null ? ` LIMIT ${String(limitVal)}` : "";
                 const sql = `SELECT ${selectCols}ts_headline('${cfg}', ${columnName}, plainto_tsquery('${cfg}', $1), '${opts}') as headline
                         FROM ${tableName}
                         WHERE to_tsvector('${cfg}', ${columnName}) @@ plainto_tsquery('${cfg}', $1)${limitClause}`;
                 const result = await adapter.executeQuery(sql, [parsed.query]);
-                return { rows: result.rows, count: result.rows?.length ?? 0 };
+                const count = result.rows?.length ?? 0;
+                const truncated = limitVal !== null && count === limitVal;
+                return {
+                    rows: result.rows,
+                    count,
+                    ...(truncated
+                        ? {
+                            truncated: true,
+                            hint: `Results limited to ${String(limitVal)}. Use limit: 0 for all rows.`,
+                        }
+                        : {}),
+                };
             }
             catch (error) {
                 if (error instanceof ZodError) {
@@ -681,11 +788,27 @@ function createTextNormalizeTool(adapter) {
         annotations: readOnly("Text Normalize"),
         icons: getToolIcons("text", readOnly("Text Normalize")),
         handler: async (params, _context) => {
-            const parsed = NormalizeSchema.parse(params ?? {});
-            // Ensure unaccent extension is available
-            await adapter.executeQuery("CREATE EXTENSION IF NOT EXISTS unaccent");
-            const result = await adapter.executeQuery(`SELECT unaccent($1) as normalized`, [parsed.text]);
-            return { normalized: result.rows?.[0]?.["normalized"] };
+            try {
+                const parsed = NormalizeSchema.parse(params ?? {});
+                // Ensure unaccent extension is available
+                await adapter.executeQuery("CREATE EXTENSION IF NOT EXISTS unaccent");
+                const result = await adapter.executeQuery(`SELECT unaccent($1) as normalized`, [parsed.text]);
+                return { normalized: result.rows?.[0]?.["normalized"] };
+            }
+            catch (error) {
+                if (error instanceof ZodError) {
+                    return {
+                        success: false,
+                        error: `pg_text_normalize validation error: ${error.issues.map((e) => e.message).join(", ")}`,
+                    };
+                }
+                return {
+                    success: false,
+                    error: formatPostgresError(error, {
+                        tool: "pg_text_normalize",
+                    }),
+                };
+            }
         },
     };
 }
@@ -709,95 +832,109 @@ function createTextSentimentTool(_adapter) {
         annotations: readOnly("Text Sentiment"),
         icons: getToolIcons("text", readOnly("Text Sentiment")),
         handler: (params, _context) => {
-            const parsed = SentimentSchema.parse(params ?? {});
-            const text = parsed.text.toLowerCase();
-            const positiveWords = [
-                "good",
-                "great",
-                "excellent",
-                "amazing",
-                "wonderful",
-                "fantastic",
-                "love",
-                "happy",
-                "positive",
-                "best",
-                "beautiful",
-                "awesome",
-                "perfect",
-                "nice",
-                "helpful",
-                "thank",
-                "thanks",
-                "pleased",
-                "satisfied",
-                "recommend",
-                "enjoy",
-                "impressive",
-                "brilliant",
-            ];
-            const negativeWords = [
-                "bad",
-                "terrible",
-                "awful",
-                "horrible",
-                "worst",
-                "hate",
-                "angry",
-                "disappointed",
-                "poor",
-                "wrong",
-                "problem",
-                "issue",
-                "fail",
-                "failed",
-                "broken",
-                "useless",
-                "waste",
-                "frustrating",
-                "annoyed",
-                "unhappy",
-                "negative",
-                "complaint",
-                "slow",
-            ];
-            const words = text.split(/\s+/);
-            const matchedPositive = words
-                .map((w) => w.replace(/[^a-z]/g, ""))
-                .filter((w) => positiveWords.includes(w));
-            const matchedNegative = words
-                .map((w) => w.replace(/[^a-z]/g, ""))
-                .filter((w) => negativeWords.includes(w));
-            const positiveScore = matchedPositive.length;
-            const negativeScore = matchedNegative.length;
-            const totalScore = positiveScore - negativeScore;
-            let sentiment;
-            if (totalScore > 2)
-                sentiment = "very_positive";
-            else if (totalScore > 0)
-                sentiment = "positive";
-            else if (totalScore < -2)
-                sentiment = "very_negative";
-            else if (totalScore < 0)
-                sentiment = "negative";
-            else
-                sentiment = "neutral";
-            const result = {
-                sentiment,
-                score: totalScore,
-                positiveCount: positiveScore,
-                negativeCount: negativeScore,
-                confidence: positiveScore + negativeScore > 3
-                    ? "high"
-                    : positiveScore + negativeScore > 1
-                        ? "medium"
-                        : "low",
-            };
-            if (parsed.returnWords) {
-                result.matchedPositive = matchedPositive;
-                result.matchedNegative = matchedNegative;
+            try {
+                const parsed = SentimentSchema.parse(params ?? {});
+                const text = parsed.text.toLowerCase();
+                const positiveWords = [
+                    "good",
+                    "great",
+                    "excellent",
+                    "amazing",
+                    "wonderful",
+                    "fantastic",
+                    "love",
+                    "happy",
+                    "positive",
+                    "best",
+                    "beautiful",
+                    "awesome",
+                    "perfect",
+                    "nice",
+                    "helpful",
+                    "thank",
+                    "thanks",
+                    "pleased",
+                    "satisfied",
+                    "recommend",
+                    "enjoy",
+                    "impressive",
+                    "brilliant",
+                ];
+                const negativeWords = [
+                    "bad",
+                    "terrible",
+                    "awful",
+                    "horrible",
+                    "worst",
+                    "hate",
+                    "angry",
+                    "disappointed",
+                    "poor",
+                    "wrong",
+                    "problem",
+                    "issue",
+                    "fail",
+                    "failed",
+                    "broken",
+                    "useless",
+                    "waste",
+                    "frustrating",
+                    "annoyed",
+                    "unhappy",
+                    "negative",
+                    "complaint",
+                    "slow",
+                ];
+                const words = text.split(/\s+/);
+                const matchedPositive = words
+                    .map((w) => w.replace(/[^a-z]/g, ""))
+                    .filter((w) => positiveWords.includes(w));
+                const matchedNegative = words
+                    .map((w) => w.replace(/[^a-z]/g, ""))
+                    .filter((w) => negativeWords.includes(w));
+                const positiveScore = matchedPositive.length;
+                const negativeScore = matchedNegative.length;
+                const totalScore = positiveScore - negativeScore;
+                let sentiment;
+                if (totalScore > 2)
+                    sentiment = "very_positive";
+                else if (totalScore > 0)
+                    sentiment = "positive";
+                else if (totalScore < -2)
+                    sentiment = "very_negative";
+                else if (totalScore < 0)
+                    sentiment = "negative";
+                else
+                    sentiment = "neutral";
+                const result = {
+                    sentiment,
+                    score: totalScore,
+                    positiveCount: positiveScore,
+                    negativeCount: negativeScore,
+                    confidence: positiveScore + negativeScore > 3
+                        ? "high"
+                        : positiveScore + negativeScore > 1
+                            ? "medium"
+                            : "low",
+                };
+                if (parsed.returnWords) {
+                    result.matchedPositive = matchedPositive;
+                    result.matchedNegative = matchedNegative;
+                }
+                return Promise.resolve(result);
+            }
+            catch (error) {
+                if (error instanceof ZodError) {
+                    return Promise.resolve({
+                        success: false,
+                        error: `pg_text_sentiment validation error: ${error.issues.map((e) => e.message).join(", ")}`,
+                    });
+                }
+                return Promise.resolve({
+                    success: false,
+                    error: error instanceof Error ? error.message : "Unknown error occurred",
+                });
             }
-            return Promise.resolve(result);
         },
     };
 }
@@ -821,10 +958,26 @@ function createTextToVectorTool(adapter) {
         annotations: readOnly("Text to Vector"),
         icons: getToolIcons("text", readOnly("Text to Vector")),
         handler: async (params, _context) => {
-            const parsed = ToVectorSchema.parse(params ?? {});
-            const cfg = parsed.config ?? "english";
-            const result = await adapter.executeQuery(`SELECT to_tsvector($1, $2) as vector`, [cfg, parsed.text]);
-            return { vector: result.rows?.[0]?.["vector"] };
+            try {
+                const parsed = ToVectorSchema.parse(params ?? {});
+                const cfg = parsed.config ?? "english";
+                const result = await adapter.executeQuery(`SELECT to_tsvector($1, $2) as vector`, [cfg, parsed.text]);
+                return { vector: result.rows?.[0]?.["vector"] };
+            }
+            catch (error) {
+                if (error instanceof ZodError) {
+                    return {
+                        success: false,
+                        error: `pg_text_to_vector validation error: ${error.issues.map((e) => e.message).join(", ")}`,
+                    };
+                }
+                return {
+                    success: false,
+                    error: formatPostgresError(error, {
+                        tool: "pg_text_to_vector",
+                    }),
+                };
+            }
         },
     };
 }
@@ -852,22 +1005,38 @@ function createTextToQueryTool(adapter) {
         annotations: readOnly("Text to Query"),
         icons: getToolIcons("text", readOnly("Text to Query")),
         handler: async (params, _context) => {
-            const parsed = ToQuerySchema.parse(params ?? {});
-            const cfg = parsed.config ?? "english";
-            const mode = parsed.mode ?? "plain";
-            let fn;
-            switch (mode) {
-                case "phrase":
-                    fn = "phraseto_tsquery";
-                    break;
-                case "websearch":
-                    fn = "websearch_to_tsquery";
-                    break;
-                default:
-                    fn = "plainto_tsquery";
+            try {
+                const parsed = ToQuerySchema.parse(params ?? {});
+                const cfg = parsed.config ?? "english";
+                const mode = parsed.mode ?? "plain";
+                let fn;
+                switch (mode) {
+                    case "phrase":
+                        fn = "phraseto_tsquery";
+                        break;
+                    case "websearch":
+                        fn = "websearch_to_tsquery";
+                        break;
+                    default:
+                        fn = "plainto_tsquery";
+                }
+                const result = await adapter.executeQuery(`SELECT ${fn}($1, $2) as query`, [cfg, parsed.text]);
+                return { query: result.rows?.[0]?.["query"], mode };
+            }
+            catch (error) {
+                if (error instanceof ZodError) {
+                    return {
+                        success: false,
+                        error: `pg_text_to_query validation error: ${error.issues.map((e) => e.message).join(", ")}`,
+                    };
+                }
+                return {
+                    success: false,
+                    error: formatPostgresError(error, {
+                        tool: "pg_text_to_query",
+                    }),
+                };
             }
-            const result = await adapter.executeQuery(`SELECT ${fn}($1, $2) as query`, [cfg, parsed.text]);
-            return { query: result.rows?.[0]?.["query"], mode };
         },
     };
 }
@@ -884,8 +1053,9 @@ function createTextSearchConfigTool(adapter) {
         annotations: readOnly("Search Configurations"),
         icons: getToolIcons("text", readOnly("Search Configurations")),
         handler: async (_params, _context) => {
-            const result = await adapter.executeQuery(`
-                SELECT 
+            try {
+                const result = await adapter.executeQuery(`
+                SELECT
                     c.cfgname as name,
                     n.nspname as schema,
                     obj_description(c.oid, 'pg_ts_config') as description
@@ -893,10 +1063,19 @@ function createTextSearchConfigTool(adapter) {
                 JOIN pg_namespace n ON n.oid = c.cfgnamespace
                 ORDER BY c.cfgname
             `);
-            return {
-                configs: result.rows ?? [],
-                count: result.rows?.length ?? 0,
-            };
+                return {
+                    configs: result.rows ?? [],
+                    count: result.rows?.length ?? 0,
+                };
+            }
+            catch (error) {
+                return {
+                    success: false,
+                    error: formatPostgresError(error, {
+                        tool: "pg_text_search_config",
+                    }),
+                };
+            }
         },
     };
 }