6eeb919013
- Simplify model selection and defaults in config - Enhance system prompt and history management - Update CLI output with colors and exit commands - Remove placeholder tool handling for cleaner agent mode
100 lines
3.1 KiB
Go
100 lines
3.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/fatih/color"
|
|
"github.com/spf13/cobra"
|
|
"gmgauthier.com/grokkit/config"
|
|
"gmgauthier.com/grokkit/internal/grok"
|
|
)
|
|
|
|
var chatCmd = &cobra.Command{
|
|
Use: "chat",
|
|
Short: "Interactive chat with Grok (use --agent for tool-enabled mode)",
|
|
Long: `Start a persistent conversation with Grok.
|
|
|
|
Normal mode: coherent, reasoning-focused chat (uses full model).
|
|
Agent mode (--agent): Grok can call tools with safety previews (uses fast model).`,
|
|
Run: runChat,
|
|
}
|
|
|
|
func init() {
|
|
chatCmd.Flags().Bool("agent", false, "Enable agent mode with tool calling (uses fast non-reasoning model)")
|
|
chatCmd.Flags().String("model", "", "Override model")
|
|
rootCmd.AddCommand(chatCmd)
|
|
}
|
|
|
|
func runChat(cmd *cobra.Command, args []string) {
|
|
agentMode, _ := cmd.Flags().GetBool("agent")
|
|
modelFlag, _ := cmd.Flags().GetString("model")
|
|
|
|
// Model switching logic
|
|
var model string
|
|
if modelFlag != "" {
|
|
model = modelFlag
|
|
} else if agentMode {
|
|
model = config.GetModel("chat-agent", "") // we'll add this default next
|
|
} else {
|
|
model = config.GetModel("chat", "")
|
|
}
|
|
|
|
client := grok.NewClient()
|
|
|
|
systemPrompt := map[string]string{
|
|
"role": "system",
|
|
"content": fmt.Sprintf("You are Grok 4, the latest and most powerful model from xAI (2026). You are currently running as `%s`. Be helpful, truthful, and a little irreverent.", model),
|
|
}
|
|
|
|
if agentMode {
|
|
systemPrompt["content"] = `You are Grok in Agent Mode.
|
|
You have access to tools: edit, scaffold, testgen, lint, commit, etc.
|
|
Always use tools when the user wants you to change code, generate tests, lint or commit.
|
|
Be concise and action-oriented. After every tool call, wait for the result.`
|
|
}
|
|
|
|
history := loadChatHistory()
|
|
if history == nil || len(history) == 0 {
|
|
history = []map[string]string{systemPrompt}
|
|
} else if history[0]["role"] != "system" {
|
|
history = append([]map[string]string{systemPrompt}, history...)
|
|
} else {
|
|
history[0] = systemPrompt // refresh system prompt
|
|
}
|
|
|
|
color.Cyan("┌──────────────────────────────────────────────────────────────┐")
|
|
color.Cyan("│ Grokkit Chat %s — Model: %s │", map[bool]string{true: "(Agent Mode)", false: ""}[agentMode], model)
|
|
color.Cyan("│ Type /quit to exit │")
|
|
color.Cyan("└──────────────────────────────────────────────────────────────┘\n")
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
for {
|
|
color.Yellow("You > ")
|
|
if !scanner.Scan() {
|
|
break
|
|
}
|
|
|
|
input := strings.TrimSpace(scanner.Text())
|
|
if input == "" {
|
|
continue
|
|
}
|
|
if input == "/quit" || input == "/q" || input == "exit" {
|
|
color.Cyan("\nGoodbye 👋\n")
|
|
break
|
|
}
|
|
|
|
history = append(history, map[string]string{"role": "user", "content": input})
|
|
|
|
color.Green("Grok > ")
|
|
reply := client.Stream(history, model)
|
|
|
|
history = append(history, map[string]string{"role": "assistant", "content": reply})
|
|
_ = saveChatHistory(history)
|
|
|
|
fmt.Println()
|
|
}
|
|
}
|