Method Assist

server-source-code/internal/handler/ai.go:243–306

Assist handles terminal assistance requests.

(w http.ResponseWriter, r *http.Request)

// Assist handles terminal assistance requests.
func (h *AIHandler) Assist(w http.ResponseWriter, r *http.Request) {
	userID, _ := r.Context().Value(middleware.UserIDKey).(string)
	if !h.checkRateLimit(r.Context(), userID) {
		Error(w, http.StatusTooManyRequests, "Rate limit exceeded. Please wait a moment.")
		return
	}

	s, err := h.settings.GetFirst(r.Context())
	if err != nil {
		Error(w, http.StatusInternalServerError, "Failed to load settings")
		return
	}
	if !s.AiEnabled {
		Error(w, http.StatusBadRequest, "AI assistant is not enabled")
		return
	}
	if s.AiAPIKey == nil || *s.AiAPIKey == "" {
		Error(w, http.StatusBadRequest, "AI API key not configured")
		return
	}

	var req AssistRequest
	if err := decodeJSON(r, &req); err != nil {
		Error(w, http.StatusBadRequest, "Invalid request body")
		return
	}
	req.Question = strings.TrimSpace(req.Question)
	if len(req.Question) < 1 || len(req.Question) > 2000 {
		Error(w, http.StatusBadRequest, "question must be 1-2000 characters")
		return
	}
	if len(req.Context) > 10000 {
		req.Context = req.Context[:10000]
	}

	// Sanitize history: last 10 messages, role+content required, max 2000 chars each
	sanitized := make([]ai.Message, 0, 10)
	start := len(req.History) - 10
	if start < 0 {
		start = 0
	}
	for i := start; i < len(req.History) && len(sanitized) < 10; i++ {
		m := req.History[i]
		if m.Role == "" || m.Content == "" {
			continue
		}
		role := "user"
		if m.Role == "assistant" {
			role = "assistant"
		}
		content := m.Content
		if len(content) > 2000 {
			content = content[:2000]
		}
		sanitized = append(sanitized, ai.Message{Role: role, Content: content})
	}

	response, err := h.aiSvc.GetAssistance(s, req.Question, req.Context, sanitized)