memory

type ReActChain struct {
	Thoughts     []string
	Actions      []ToolAction
	Observations []string
}

type RoundSummary struct {
	RoundID      string                 `json:"round_id"`
	Summary      string                 `json:"summary"`
	KeyPoints    []string               `json:"key_points"`
	Messages     int                    `json:"messages"`
	Thoughts     int                    `json:"thoughts"`
	Actions      int                    `json:"actions"`
	Observations int                    `json:"observations"`
	Timestamp    time.Time              `json:"timestamp"`
	ExpiresAt    time.Time              `json:"expires_at"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}

type ShortTermMemory struct {
	// contains filtered or unexported fields
}

type ShortTermMemoryStats struct {
	StoredRounds int
	MaxRounds    int
	TTL          time.Duration
}

type ToolAction struct {
	ToolName  string
	Input     map[string]interface{}
	Timestamp time.Time
}

type ToolObservation struct {
	ToolName  string
	Output    string
	Error     string
	Timestamp time.Time
}

type WorkingMemory struct {
	// contains filtered or unexported fields
}

type WorkingMemoryContext struct {
	RoundID    string
	Messages   []protocol.Message
	ReActChain ReActChain
	TempVars   map[string]interface{}
	Stats      WorkingMemoryStats
}

type WorkingMemoryStats struct {
	RoundID          string
	MessageCount     int
	ThoughtCount     int
	ActionCount      int
	ObservationCount int
	TempVarCount     int
	Duration         time.Duration
	StartTime        time.Time
}