Greg Gauthier
6066e65af8
- Implement `grokkit mcp` command to run as MCP server over stdio - Export analysis, docs, and testgen functions for MCP tools - Add tools: lint_code, analyze_code, generate_docs, generate_tests, generate_commit_msg, run_recipe - Register resources: recipes://local, recipes://global, prompts://language - Update README and add user guide for MCP - Add MCP config options and dependencies
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package docs
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// BuildDocsMessages builds the messages for documentation generation.
|
|
// Exported for use by MCP server.
|
|
func BuildDocsMessages(language, code string) []map[string]string {
|
|
style := docStyle(language)
|
|
systemPrompt := fmt.Sprintf(
|
|
"You are a documentation expert. Add %s documentation comments to the provided code. "+
|
|
"Return ONLY the documented code with no explanations, markdown, or extra text. "+
|
|
"Do NOT include markdown code fences. Document all public functions, methods, types, and constants.",
|
|
style,
|
|
)
|
|
|
|
userPrompt := fmt.Sprintf("Add documentation comments to the following %s code:\n\n%s", language, code)
|
|
|
|
return []map[string]string{
|
|
{"role": "system", "content": systemPrompt},
|
|
{"role": "user", "content": userPrompt},
|
|
}
|
|
}
|
|
|
|
func docStyle(language string) string {
|
|
switch strings.ToLower(language) {
|
|
case "go":
|
|
return "godoc"
|
|
case "python":
|
|
return "PEP 257 docstring"
|
|
case "c", "c++":
|
|
return "doxygen style"
|
|
case "javascript", "typescript":
|
|
return "JSDoc"
|
|
case "rust":
|
|
return "rustdoc"
|
|
case "ruby":
|
|
return "YARD"
|
|
case "java":
|
|
return "javadoc"
|
|
case "shell":
|
|
return "shell style comments"
|
|
default:
|
|
return "standard"
|
|
}
|
|
}
|