/**
* Prompt generator utilities
*
* Helpers for generating tool documentation and discovery prompts.
*/
import type { ToolDefinition } from '../types/index.js';
/**
* Generate a compact index of all tools grouped by category
*/
export function generateCompactIndex(tools: ToolDefinition[]): string {
// Group tools by category
const groups: Record<string, ToolDefinition[]> = {};
for (const tool of tools) {
const group = tool.group ?? 'other';
groups[group] ??= [];
groups[group].push(tool);
}
// Generate compact format
const lines: string[] = [];
for (const [group, groupTools] of Object.entries(groups)) {
lines.push(`### ${group} (${String(groupTools.length)} tools)`);
lines.push('');
for (const tool of groupTools) {
lines.push(`- **${tool.name}**: ${tool.description}`);
}
lines.push('');
}
return lines.join('\n');
}
/**
* Generate a discovery prompt with common use cases
*/
export function generateDiscoveryPrompt(tools: ToolDefinition[]): string {
// Find commonly used tools based on tags
const queryTools = tools.filter(t =>
t.name.includes('query') || t.name.includes('read') || t.name.includes('write')
).slice(0, 3);
const schemaTools = tools.filter(t =>
t.name.includes('table') || t.name.includes('schema') || t.name.includes('describe')
).slice(0, 3);
const perfTools = tools.filter(t =>
t.name.includes('explain') || t.name.includes('performance') || t.name.includes('stats')
).slice(0, 3);
const lines: string[] = [
'## Quick Start',
'',
'### Query Data',
];
for (const tool of queryTools) {
lines.push(`- \`${tool.name}\`: ${tool.description}`);
}
lines.push('', '### Explore Schema');
for (const tool of schemaTools) {
lines.push(`- \`${tool.name}\`: ${tool.description}`);
}
lines.push('', '### Analyze Performance');
for (const tool of perfTools) {
lines.push(`- \`${tool.name}\`: ${tool.description}`);
}
lines.push('');
return lines.join('\n');
}
