Greg Gauthier
8e0d06d8a1
Introduce new `agent` command that scans .go files in the project, generates an AI-driven plan for changes based on user instruction, and applies edits with previews and backups. Includes integration with Grok client for planning and content generation. Update existing files with timestamp comments as part of the agent's editing demonstration. Add agentCmd to root command.
33 lines
961 B
Go
33 lines
961 B
Go
// Current time: 2024-09-07 10:00:00 UTC
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/spf13/cobra"
|
|
"gmgauthier.com/grokkit/config"
|
|
"gmgauthier.com/grokkit/internal/git"
|
|
"gmgauthier.com/grokkit/internal/grok"
|
|
)
|
|
|
|
var reviewCmd = &cobra.Command{
|
|
Use: "review [path]",
|
|
Short: "Review the current repository or directory",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
modelFlag, _ := cmd.Flags().GetString("model")
|
|
model := config.GetModel(modelFlag)
|
|
|
|
client := grok.NewClient()
|
|
diff := git.Run([]string{"diff", "--no-color"})
|
|
status := git.Run([]string{"status", "--short"})
|
|
|
|
messages := []map[string]string{
|
|
{"role": "system", "content": "You are an expert code reviewer. Give a concise summary + 3-5 actionable improvements."},
|
|
{"role": "user", "content": fmt.Sprintf("Git status:\n%s\n\nGit diff:\n%s", status, diff)},
|
|
}
|
|
color.Yellow("Grok is reviewing the repo...")
|
|
client.Stream(messages, model)
|
|
},
|
|
} |