feat: dino and eyes

Author: dartpain

.claude/settings.json

@@ -0,0 +1,17 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(go build:*)",
+      "Bash(./docsgpt-cli --version)",
+      "Bash(./docsgpt-cli --help)",
+      "Bash(./docsgpt-cli config:*)",
+      "Bash(./docsgpt-cli chat:*)",
+      "Bash(./docsgpt-cli keys:*)",
+      "Bash(curl -s -X POST http://localhost:7091/v1/chat/completions -H 'Authorization: Bearer 46ded8a6-b308-421d-8ed3-839175fe9ad3' -H 'Content-Type: application/json' -d '{:*)",
+      "Bash(curl -s -N -X POST http://localhost:7091/v1/chat/completions -H 'Authorization: Bearer 46ded8a6-b308-421d-8ed3-839175fe9ad3' -H 'Content-Type: application/json' -d '{:*)",
+      "Bash(timeout 30 ./docsgpt-cli ask --no-stream \"Create a file called TESTFILE123.md with the text 'hello from docsgpt'\")",
+      "Bash(./docsgpt-cli ask:*)",
+      "Bash(grep -v grep echo \"---\")"
+    ]
+  }
+}

cmd/ask.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"context"
 	"fmt"
+	"os"
 	"strings"
 	"time"
 
@@ -12,7 +13,6 @@ import (
 	"docsgpt-cli/internal/display"
 	"docsgpt-cli/internal/tools"
 
-	"github.com/fatih/color"
 	"github.com/spf13/cobra"
 )
 
@@ -51,39 +51,41 @@ This command will provide a contextual answer and, if applicable, copy a relevan
 			{Role: "user", Content: fullQuestion},
 		}
 
-		green := color.New(color.FgGreen).SprintFunc()
-		fmt.Printf(green("Key: %s\n"), keyName)
-		fmt.Printf(green(" ❯ "))
+		cwd, _ := os.Getwd()
+		fmt.Println(display.RenderHeader(keyName, baseURL, cwd))
+		fmt.Print(display.Prompt("❯ "))
 
 		ctx := context.Background()
-		toolDefs := tools.ToolDefinitions()
+		var toolDefs []api.Tool
+		if !globalNoContext {
+			toolDefs = tools.ToolDefinitions()
+		}
 		timeout := time.Duration(globalTimeout) * time.Second
 
+		renderer := display.NewStreamRenderer()
+
 		onDelta := func(delta api.Delta, finishReason string) {
-			display.StreamDelta(delta)
+			renderer.Delta(delta)
 		}
 
 		onToolCall := func(tc api.ToolCall) string {
 			return handleToolCall(tc, timeout)
 		}
 
-		updatedHistory, err := client.RunWithTools(
+		_, err = client.RunWithTools(
 			ctx, messages, toolDefs, !globalNoStream, onDelta, onToolCall,
 		)
 		if err != nil {
 			return err
 		}
 		fmt.Println()
 
-		// Find the last assistant message for clipboard
-		var answer string
-		for i := len(updatedHistory) - 1; i >= 0; i-- {
-			if updatedHistory[i].Role == "assistant" {
-				answer = updatedHistory[i].Content
-				break
-			}
+		// Render markdown for the final answer if it contains formatting
+		if rendered := renderer.Finish(); rendered != "" {
+			fmt.Print(rendered)
 		}
 
+		answer := renderer.Content()
 		command := extractCommand(answer)
 		if command != "" {
 			copyToClipboard(command)

cmd/chat.go

@@ -1,11 +1,9 @@
 package cmd
 
 import (
-	"bufio"
 	"context"
 	"fmt"
 	"os"
-	"os/signal"
 	"strings"
 	"time"
 
@@ -15,7 +13,7 @@ import (
 	"docsgpt-cli/internal/display"
 	"docsgpt-cli/internal/tools"
 
-	"github.com/fatih/color"
+	prompt "github.com/c-bata/go-prompt"
 	"github.com/spf13/cobra"
 )
 
@@ -27,7 +25,10 @@ var chatCmd = &cobra.Command{
 Special commands:
     /quit   - Exit the chat session
     /clear  - Clear conversation history
-    /copy   - Copy the last code block to clipboard`,
+    /copy   - Copy the last code block to clipboard
+    /think  - Toggle reasoning visibility
+
+Type "/" to see available commands with live autocomplete.`,
 	RunE: func(cmd *cobra.Command, args []string) error {
 		cfg, err := config.Load()
 		if err != nil {
@@ -42,10 +43,11 @@ Special commands:
 		baseURL := cfg.ResolveURL(globalURL)
 		client := api.NewClient(baseURL, apiKey)
 
-		green := color.New(color.FgGreen).SprintFunc()
-		fmt.Printf(green("Connected with key: %s\n"), keyName)
-		fmt.Printf(green("Server: %s\n"), baseURL)
-		fmt.Println("Type /quit to exit, /clear to reset history, /copy to copy last code block.")
+		cwd, _ := os.Getwd()
+		fmt.Println(display.RenderHeader(keyName, baseURL, cwd))
+		if hints := display.RenderHints("chat"); hints != "" {
+			fmt.Println(hints)
+		}
 		fmt.Println()
 
 		var history []api.Message
@@ -65,96 +67,138 @@ Special commands:
 	},
 }
 
-func runChatLoop(client *api.Client, history []api.Message) error {
-	reader := bufio.NewReader(os.Stdin)
-	var lastAnswer string
-
-	// Handle Ctrl+C gracefully
-	sigCh := make(chan os.Signal, 1)
-	signal.Notify(sigCh, os.Interrupt)
-	go func() {
-		<-sigCh
-		fmt.Println("\nGoodbye!")
+// chatSession holds the mutable state for an interactive chat.
+type chatSession struct {
+	client        *api.Client
+	history       []api.Message
+	lastAnswer    string
+	showReasoning bool
+	toolDefs      []api.Tool
+	timeout       time.Duration
+}
+
+func (s *chatSession) executor(input string) {
+	input = strings.TrimSpace(input)
+	if input == "" {
+		return
+	}
+
+	switch input {
+	case "/quit":
+		fmt.Println("Goodbye!")
 		os.Exit(0)
-	}()
+	case "/clear":
+		var newHistory []api.Message
+		if len(s.history) > 0 && s.history[0].Role == "system" {
+			newHistory = append(newHistory, s.history[0])
+		}
+		s.history = newHistory
+		s.lastAnswer = ""
+		fmt.Println("History cleared.")
+		return
+	case "/copy":
+		if s.lastAnswer == "" {
+			printError("No previous response to copy from.")
+			return
+		}
+		command := extractCommand(s.lastAnswer)
+		if command != "" {
+			copyToClipboard(command)
+		} else {
+			printError("No code block found in last response.")
+		}
+		return
+	case "/think":
+		s.showReasoning = !s.showReasoning
+		if s.showReasoning {
+			fmt.Println(display.Muted("Reasoning: visible"))
+		} else {
+			fmt.Println(display.Muted("Reasoning: hidden"))
+		}
+		return
+	}
 
-	toolDefs := tools.ToolDefinitions()
-	timeout := time.Duration(globalTimeout) * time.Second
+	s.history = append(s.history, api.Message{Role: "user", Content: input})
+	ctx := context.Background()
 
-	for {
-		green := color.New(color.FgGreen).SprintFunc()
-		fmt.Print(green("> "))
+	renderer := display.NewStreamRenderer()
+	renderer.ShowReasoning = s.showReasoning
 
-		input, err := reader.ReadString('\n')
-		if err != nil {
-			return nil // EOF
-		}
-		input = strings.TrimSpace(input)
+	onDelta := func(delta api.Delta, finishReason string) {
+		renderer.Delta(delta)
+	}
 
-		if input == "" {
-			continue
-		}
+	onToolCall := func(tc api.ToolCall) string {
+		return handleToolCall(tc, s.timeout)
+	}
 
-		switch input {
-		case "/quit":
-			fmt.Println("Goodbye!")
-			return nil
-		case "/clear":
-			// Keep system message if present
-			var newHistory []api.Message
-			if len(history) > 0 && history[0].Role == "system" {
-				newHistory = append(newHistory, history[0])
-			}
-			history = newHistory
-			lastAnswer = ""
-			fmt.Println("History cleared.")
-			continue
-		case "/copy":
-			if lastAnswer == "" {
-				printError("No previous response to copy from.")
-				continue
-			}
-			command := extractCommand(lastAnswer)
-			if command != "" {
-				copyToClipboard(command)
-			} else {
-				printError("No code block found in last response.")
-			}
-			continue
-		}
+	updatedHistory, err := s.client.RunWithTools(
+		ctx, s.history, s.toolDefs, !globalNoStream, onDelta, onToolCall,
+	)
+	if err != nil {
+		printError(err.Error())
+		return
+	}
+	fmt.Println()
 
-		history = append(history, api.Message{Role: "user", Content: input})
-		ctx := context.Background()
+	if rendered := renderer.Finish(); rendered != "" {
+		fmt.Print(rendered)
+	}
 
-		onDelta := func(delta api.Delta, finishReason string) {
-			display.StreamDelta(delta)
-		}
+	s.history = updatedHistory
+	s.lastAnswer = renderer.Content()
 
-		onToolCall := func(tc api.ToolCall) string {
-			return handleToolCall(tc, timeout)
-		}
+	fmt.Println()
+}
 
-		updatedHistory, err := client.RunWithTools(
-			ctx, history, toolDefs, !globalNoStream, onDelta, onToolCall,
-		)
-		if err != nil {
-			printError(err.Error())
-			continue
-		}
-		fmt.Println()
+func (s *chatSession) completer(d prompt.Document) []prompt.Suggest {
+	text := d.TextBeforeCursor()
+	if !strings.HasPrefix(text, "/") {
+		return nil
+	}
 
-		history = updatedHistory
+	suggestions := []prompt.Suggest{
+		{Text: "/quit", Description: "Exit the chat session"},
+		{Text: "/clear", Description: "Clear conversation history"},
+		{Text: "/copy", Description: "Copy last code block to clipboard"},
+		{Text: "/think", Description: "Toggle reasoning visibility"},
+	}
 
-		// Find the last assistant message for lastAnswer
-		for i := len(history) - 1; i >= 0; i-- {
-			if history[i].Role == "assistant" {
-				lastAnswer = history[i].Content
-				break
-			}
-		}
+	return prompt.FilterHasPrefix(suggestions, text, true)
+}
 
-		fmt.Println()
+func runChatLoop(client *api.Client, history []api.Message) error {
+	var toolDefs []api.Tool
+	if !globalNoContext {
+		toolDefs = tools.ToolDefinitions()
+	}
+
+	session := &chatSession{
+		client:   client,
+		history:  history,
+		toolDefs: toolDefs,
+		timeout:  time.Duration(globalTimeout) * time.Second,
 	}
+
+	p := prompt.New(
+		session.executor,
+		session.completer,
+		prompt.OptionPrefix("> "),
+		prompt.OptionPrefixTextColor(prompt.Purple),
+		prompt.OptionSuggestionBGColor(prompt.DarkGray),
+		prompt.OptionSuggestionTextColor(prompt.White),
+		prompt.OptionSelectedSuggestionBGColor(prompt.Purple),
+		prompt.OptionSelectedSuggestionTextColor(prompt.White),
+		prompt.OptionDescriptionBGColor(prompt.DarkGray),
+		prompt.OptionDescriptionTextColor(prompt.White),
+		prompt.OptionSelectedDescriptionBGColor(prompt.Purple),
+		prompt.OptionSelectedDescriptionTextColor(prompt.White),
+		prompt.OptionScrollbarBGColor(prompt.DarkGray),
+		prompt.OptionScrollbarThumbColor(prompt.Purple),
+		prompt.OptionShowCompletionAtStart(),
+	)
+	p.Run()
+	return nil
 }
 
 func handleToolCall(tc api.ToolCall, timeout time.Duration) string {
@@ -164,8 +208,7 @@ func handleToolCall(tc api.ToolCall, timeout time.Duration) string {
 	if normalizedName == "run_command" {
 		safe, reason := tools.IsSafe(tc.Function.Arguments)
 		if !safe {
-			red := color.New(color.FgRed).SprintFunc()
-			fmt.Printf("\n%s Command blocked: %s\n", red("✗"), reason)
+			fmt.Printf("\n%s Command blocked: %s\n", display.Danger("✗"), reason)
 			return fmt.Sprintf("Command was blocked for safety: %s", reason)
 		}
 	}