/**
* postgres-mcp - Prompt Generator Utilities
*
* Helper functions for generating prompts, especially for lazy hydration.
*/
import type { ToolDefinition } from "../types/index.js";
/**
* Generate a compact tool index for lazy hydration.
* Groups tools by category and provides minimal token overhead.
*/
export function generateCompactIndex(tools: ToolDefinition[]): string {
// Group tools by their group
const grouped = new Map<string, ToolDefinition[]>();
for (const tool of tools) {
const group = tool.group;
const groupTools = grouped.get(group) ?? [];
groupTools.push(tool);
grouped.set(group, groupTools);
}
// Build compact index
const lines: string[] = [
`# PostgreSQL MCP Tools (${String(tools.length)} total)`,
"",
"Use specific tool names when needed. Ask for details about any tool.",
"",
];
for (const [group, groupTools] of grouped) {
lines.push(`## ${group} (${String(groupTools.length)})`);
for (const tool of groupTools) {
// Compact format: name - short description
const shortDesc = tool.description.split(".")[0] ?? tool.description;
lines.push(`- ${tool.name}: ${shortDesc}`);
}
lines.push("");
}
return lines.join("\n");
}
/**
* Generate a discovery prompt that helps the AI find relevant tools.
*/
export function generateDiscoveryPrompt(tools: ToolDefinition[]): string {
const lines: string[] = [
"You have access to PostgreSQL database tools. Here is how to find what you need:",
"",
"**Tool Categories:**",
];
// Group and count
const grouped = new Map<string, number>();
for (const tool of tools) {
grouped.set(tool.group, (grouped.get(tool.group) ?? 0) + 1);
}
for (const [group, count] of grouped) {
lines.push(`- ${group}: ${String(count)} tools`);
}
lines.push("");
lines.push("**Common Tasks:**");
lines.push("- Query data: pg_read_query");
lines.push("- Modify data: pg_write_query");
lines.push("- View tables: pg_list_tables, pg_describe_table");
lines.push("- Analyze performance: pg_explain_analyze, pg_stat_statements");
lines.push("- Manage transactions: pg_transaction_begin/commit/rollback");
lines.push("");
lines.push(
"Ask about any category or task for specific tool recommendations.",
);
return lines.join("\n");
}
/**
* Generate tool tags for search/discovery.
*/
export function generateToolTags(tool: ToolDefinition): string[] {
const tags: string[] = [tool.group];
// Extract keywords from name
const nameParts = tool.name.replace("pg_", "").split("_");
tags.push(...nameParts);
// Extract keywords from description
const descWords = tool.description
.toLowerCase()
.split(/\s+/)
.filter((w) => w.length > 3 && !["the", "and", "for", "with"].includes(w));
tags.push(...descWords.slice(0, 5)); // Limit to 5 keywords
return [...new Set(tags)]; // Deduplicate
}
