Bugfixes #2 (#25)

Author: alvinunreal

internal/ai_client.go

@@ -117,6 +117,9 @@ func (c *AiClient) ChatCompletion(ctx context.Context, messages []Message, model
 	req.Header.Set("HTTP-Referer", "https://github.com/alvinunreal/tmuxai")
 	req.Header.Set("X-Title", "TmuxAI")
 
+	// Log the request details for debugging before sending
+	logger.Debug("Sending API request to: %s with model: %s", url, model)
+
 	// Send the request
 	resp, err := c.client.Do(req)
 	if err != nil {
@@ -135,6 +138,9 @@ func (c *AiClient) ChatCompletion(ctx context.Context, messages []Message, model
 		return "", fmt.Errorf("failed to read response: %w", err)
 	}
 
+	// Log the raw response for debugging
+	logger.Debug("API response status: %d, response size: %d bytes", resp.StatusCode, len(body))
+
 	// Check for errors
 	if resp.StatusCode != http.StatusOK {
 		logger.Error("API returned error: %s", body)
@@ -144,7 +150,7 @@ func (c *AiClient) ChatCompletion(ctx context.Context, messages []Message, model
 	// Parse the response
 	var completionResp ChatCompletionResponse
 	if err := json.Unmarshal(body, &completionResp); err != nil {
-		logger.Error("Failed to unmarshal response: %v", err)
+		logger.Error("Failed to unmarshal response: %v, body: %s", err, body)
 		return "", fmt.Errorf("failed to unmarshal response: %w", err)
 	}
 
@@ -155,8 +161,9 @@ func (c *AiClient) ChatCompletion(ctx context.Context, messages []Message, model
 		return responseContent, nil
 	}
 
-	logger.Error("No completion choices returned")
-	return "", fmt.Errorf("no completion choices returned")
+	// Enhanced error for no completion choices
+	logger.Error("No completion choices returned. Raw response: %s", string(body))
+	return "", fmt.Errorf("no completion choices returned (model: %s, status: %d)", model, resp.StatusCode)
 }
 
 func debugChatMessages(chatMessages []ChatMessage, response string) {

internal/confirm.go

@@ -92,6 +92,8 @@ func (m *Manager) confirmedToExec(command string, prompt string, edit bool) (boo
 			// empty command
 			return false, ""
 		}
+	case "n", "no", "cancel":
+		return false, ""
 	default:
 		// any other input is retry confirmation
 		return m.confirmedToExec(command, prompt, edit)

internal/process_message.go

@@ -63,7 +63,15 @@ func (m *Manager) ProcessUserMessage(ctx context.Context, message string) bool {
 			return false
 		}
 
-		fmt.Println("Failed to get response from AI: " + err.Error())
+		// Log both to console and debug file to capture error context
+		errMsg := "Failed to get response from AI: " + err.Error()
+		fmt.Println(errMsg)
+
+		// Debug the failed request even when there's an error
+		if m.Config.Debug {
+			debugChatMessages(append(history, currentMessage), "ERROR: "+err.Error())
+		}
+
 		return false
 	}
 
@@ -77,7 +85,16 @@ func (m *Manager) ProcessUserMessage(ctx context.Context, message string) bool {
 	if err != nil {
 		s.Stop()
 		m.Status = ""
-		fmt.Println("Failed to parse AI response: " + err.Error())
+
+		// Log both to console and debug file
+		errMsg := "Failed to parse AI response: " + err.Error()
+		fmt.Println(errMsg)
+
+		// Debug the failed parsing even when there's an error
+		if m.Config.Debug {
+			debugChatMessages(append(history, currentMessage), "PARSE ERROR: "+response)
+		}
+
 		return false
 	}
 
@@ -141,24 +158,42 @@ func (m *Manager) ProcessUserMessage(ctx context.Context, message string) bool {
 		}
 	}
 
-	for _, sendKey := range r.SendKeys {
-		code, _ := system.HighlightCode("txt", sendKey)
-		m.Println(code)
+	// Process SendKeys
+	if len(r.SendKeys) > 0 {
+		// Show preview of all keys
+		keysPreview := "Keys to send:\n"
+		for i, sendKey := range r.SendKeys {
+			code, _ := system.HighlightCode("txt", sendKey)
+			if i == len(r.SendKeys)-1 {
+				keysPreview += code
+			} else {
+				keysPreview += code + "\n"
+			}
+		}
 
-		isSafe := false
-		command := sendKey
+		m.Println(keysPreview)
+
+		// Determine confirmation message based on number of keys
+		confirmMessage := "Send this key?"
+		if len(r.SendKeys) > 1 {
+			confirmMessage = "Send all these keys?"
+		}
+
+		// Get confirmation if required
+		allConfirmed := true
 		if m.GetSendKeysConfirm() {
-			isSafe, command = m.confirmedToExec(sendKey, "Send this key(s)?", true)
-		} else {
-			isSafe = true
+			allConfirmed, _ = m.confirmedToExec("keys shown above", confirmMessage, true)
+			if !allConfirmed {
+				m.Status = ""
+				return false
+			}
 		}
-		if isSafe {
-			m.Println("Sending keys: " + command)
-			system.TmuxSendCommandToPane(m.ExecPane.Id, command, false)
+
+		// Send each key with delay
+		for _, sendKey := range r.SendKeys {
+			m.Println("Sending keys: " + sendKey)
+			system.TmuxSendCommandToPane(m.ExecPane.Id, sendKey, false)
 			time.Sleep(1 * time.Second)
-		} else {
-			m.Status = ""
-			return false
 		}
 	}