nofx/agent/history.go

package agent

import (
	"strings"
	"sync"
	"time"
)

// chatMessage represents a single message in conversation history.
type chatMessage struct {
	Role      string    `json:"role"` // "user" or "assistant"
	Content   string    `json:"content"`
	Timestamp time.Time `json:"timestamp"`
}

// chatHistory stores conversation history per user.
type chatHistory struct {
	mu       sync.RWMutex
	sessions map[int64][]chatMessage
	maxTurns int // hard safety cap in messages per user
}

func newChatHistory(maxTurns int) *chatHistory {
	if maxTurns <= 0 {
		maxTurns = 100 // default hard cap; recent-window trimming is handled separately
	}
	return &chatHistory{
		sessions: make(map[int64][]chatMessage),
		maxTurns: maxTurns,
	}
}

// Add appends a message to the user's history.
func (h *chatHistory) Add(userID int64, role, content string) {
	h.mu.Lock()
	defer h.mu.Unlock()

	h.sessions[userID] = append(h.sessions[userID], chatMessage{
		Role:      role,
		Content:   content,
		Timestamp: time.Now(),
	})

	// Hard safety cap in case summarization is unavailable.
	msgs := h.sessions[userID]
	if len(msgs) > h.maxTurns {
		h.sessions[userID] = msgs[len(msgs)-h.maxTurns:]
	}
}

// Get returns the conversation history for a user.
func (h *chatHistory) Get(userID int64) []chatMessage {
	h.mu.RLock()
	defer h.mu.RUnlock()

	msgs := h.sessions[userID]
	if msgs == nil {
		return nil
	}
	// Return a copy
	result := make([]chatMessage, len(msgs))
	copy(result, msgs)
	return result
}

func (h *chatHistory) Replace(userID int64, msgs []chatMessage) {
	h.mu.Lock()
	defer h.mu.Unlock()

	if len(msgs) == 0 {
		delete(h.sessions, userID)
		return
	}

	if len(msgs) > h.maxTurns {
		msgs = msgs[len(msgs)-h.maxTurns:]
	}
	cloned := make([]chatMessage, len(msgs))
	copy(cloned, msgs)
	h.sessions[userID] = cloned
}

// Clear resets conversation history for a user.
func (h *chatHistory) Clear(userID int64) {
	h.mu.Lock()
	defer h.mu.Unlock()
	delete(h.sessions, userID)
}

// CleanOld removes sessions older than the given duration.
func (h *chatHistory) CleanOld(maxAge time.Duration) {
	h.mu.Lock()
	defer h.mu.Unlock()

	now := time.Now()
	for uid, msgs := range h.sessions {
		if len(msgs) > 0 {
			lastMsg := msgs[len(msgs)-1]
			if now.Sub(lastMsg.Timestamp) > maxAge {
				delete(h.sessions, uid)
			}
		}
	}
}

func (a *Agent) getLastAssistantReply(userID int64) string {
	if a == nil || a.history == nil {
		return ""
	}
	msgs := a.history.Get(userID)
	for i := len(msgs) - 1; i >= 0; i-- {
		if strings.EqualFold(strings.TrimSpace(msgs[i].Role), "assistant") {
			return strings.TrimSpace(msgs[i].Content)
		}
	}
	return ""
}