context.go

package reference

import (
	"fmt"
	"strings"
)

// ContextFrame represents an engineered context payload for an agent step.
type ContextFrame struct {
	SystemRules     string   `json:"system_rules"`
	ActiveFile      string   `json:"active_file"`
	WorkspaceState  string   `json:"workspace_state"`
	KnowledgeItems  []string `json:"knowledge_items"`
	ToolHistory     []string `json:"tool_history"`
	MaxHistoryItems int      `json:"max_history_items"`
}

// NewContextFrame creates a new ContextFrame with default limits.
func NewContextFrame(systemRules string) *ContextFrame {
	return &ContextFrame{
		SystemRules:     systemRules,
		MaxHistoryItems: 10,
	}
}

// AddToolObservation appends a condensed tool observation to execution history.
func (c *ContextFrame) AddToolObservation(toolName, summary string) {
	entry := fmt.Sprintf("[%s] %s", toolName, summary)
	c.ToolHistory = append(c.ToolHistory, entry)

	if len(c.ToolHistory) > c.MaxHistoryItems {
		c.ToolHistory = c.ToolHistory[len(c.ToolHistory)-c.MaxHistoryItems:]
	}
}

// AssemblePayload compiles all context elements into a clean, token-dense prompt payload.
func (c *ContextFrame) AssemblePayload() string {
	var sb strings.Builder

	sb.WriteString("=== SYSTEM RULES ===\n")
	sb.WriteString(c.SystemRules)
	sb.WriteString("\n\n")

	if c.WorkspaceState != "" {
		sb.WriteString("=== WORKSPACE STATE ===\n")
		sb.WriteString(c.WorkspaceState)
		sb.WriteString("\n\n")
	}

	if len(c.KnowledgeItems) > 0 {
		sb.WriteString("=== KNOWLEDGE ITEMS ===\n")
		for _, ki := range c.KnowledgeItems {
			sb.WriteString("- ")
			sb.WriteString(ki)
			sb.WriteString("\n")
		}
		sb.WriteString("\n")
	}

	if len(c.ToolHistory) > 0 {
		sb.WriteString("=== EXECUTION HISTORY (CONDENSED) ===\n")
		for _, step := range c.ToolHistory {
			sb.WriteString(step)
			sb.WriteString("\n")
		}
	}

	return sb.String()
}