memory

type Attachment struct {
	Name string `json:"name,omitempty" yaml:"name"`
	URL  string `json:"url,omitempty"  yaml:"url"`
	Size int64  `json:"size,omitempty" yaml:"size"` // bytes
	// MediaType allows UI to decide how to display or download.
	MediaType string `json:"mediaType,omitempty" yaml:"mediaType,omitempty"`
}

type Attachments []Attachment

type CombinedPolicy struct {
	Policies []Policy
}

type ConversationMeta struct {
	ID         string    `json:"id"`
	ParentID   string    `json:"parentId,omitempty"`
	Title      string    `json:"title,omitempty"`
	Visibility string    `json:"visibility,omitempty"` // full|summary|none
	CreatedAt  time.Time `json:"createdAt"`

	// Model stores the last LLM model explicitly chosen by the user within
	// this conversation. When a subsequent turn omits the model override the
	// orchestration can fall back to this value so that the user does not
	// have to repeat the flag every time.
	Model string `json:"model,omitempty"`

	// Tools keeps the last explicit per-turn tool allow-list requested by the
	// user. When a subsequent turn sends an empty tools slice, orchestration
	// falls back to this stored list so the preference persists.
	Tools []string `json:"tools,omitempty"`

	// Agent records the last agent configuration reference (path or name)
	// explicitly used in the conversation so that subsequent requests can
	// omit the field and still continue the thread with the same agent.
	Agent string `json:"agent,omitempty"`

	// Context holds the latest accepted elicitation payload so that the user
	// does not have to resend the same data every turn when the same agent
	// schema still applies.
	Context map[string]interface{} `json:"context,omitempty"`
}

type EmbedFunc func(ctx context.Context, texts []string) ([][]float32, error)

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

type ExecutionTrace struct {
	// Auto-incremented per-conversation identifier.
	ID int `json:"id" yaml:"id"`

	// Parent assistant message ID that triggered this tool call.
	ParentMsgID string `json:"parentId" yaml:"parentId"`

	// Canonical service method, e.g. "file.count".
	Name string `json:"name" yaml:"name"`

	// Marshalled JSON argument object supplied to the tool.
	Request json.RawMessage `json:"request" yaml:"request"`

	// Whether the call succeeded.
	Success bool `json:"success" yaml:"success"`

	// Result (when Success==true) or nil.
	Result json.RawMessage `json:"result,omitempty" yaml:"result,omitempty"`

	// Error message when Success==false.
	Error string `json:"error,omitempty" yaml:"error,omitempty"`

	StartedAt time.Time `json:"startedAt" yaml:"startedAt"`
	EndedAt   time.Time `json:"endedAt" yaml:"endedAt"`

	// Optional plan context – allows UI to tie tool invocation back to the
	// generating plan step.
	PlanID    string     `json:"planId,omitempty" yaml:"planId,omitempty"`
	StepIndex int        `json:"stepIndex,omitempty" yaml:"stepIndex,omitempty"`
	Step      *plan.Step `json:"step,omitempty" yaml:"step,omitempty"`
	Plan      *plan.Plan `json:"plan,omitempty" yaml:"plan,omitempty"`
	// When the step requested additional user input, surface the elicitation so
	// that callers can render an appropriate prompt.
	Elicitation *plan.Elicitation `json:"elicitation,omitempty" yaml:"elicitation,omitempty"`
}

type History interface {
	AddMessage(ctx context.Context, msg Message) error
	GetMessages(ctx context.Context, convID string) ([]Message, error)
	Retrieve(ctx context.Context, convID string, policy Policy) ([]Message, error)

	// UpdateMessage finds message by id within convID and applies mutator.
	UpdateMessage(ctx context.Context, messageId string, mutate func(*Message)) error

	// LookupMessage searches all conversations and returns the first message
	LookupMessage(ctx context.Context, messageID string) (*Message, error)

	LatestMessage(ctx context.Context) (msg *Message, err error)

	// --- Conversation meta management ------------------------------
	CreateMeta(ctx context.Context, id, parentID, title, visibility string)
	Meta(ctx context.Context, id string) (*ConversationMeta, bool)
	Children(ctx context.Context, parentID string) ([]ConversationMeta, bool)
}

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

type LastNPolicy struct {
	N int
}

type Message struct {
	ID             string  `json:"id"`
	ConversationID string  `json:"conversationId"`
	ParentID       string  `json:"parentId,omitempty"`
	Role           string  `json:"role"`
	Actor          string  `json:"actor,omitempty" yaml:"actor,omitempty"`
	Content        string  `json:"content"`
	ToolName       *string `json:"toolName,omitempty"` // Optional tool name, can be nil
	// When messages include file uploads the Attachments slice describes each
	// uploaded asset (or generated/downloadable asset on assistant side).
	Attachments Attachments     `json:"attachments,omitempty" yaml:"attachments,omitempty" sqlx:"-"`
	Executions  []*plan.Outcome `json:"executions,omitempty" yaml:"executions,omitempty" sqlx:"-"`
	CreatedAt   time.Time       `json:"createdAt" yaml:"createdAt"`

	// Elicitation carries a structured schema-driven prompt when the assistant
	// needs additional user input. When non-nil the UI can render an
	// interactive form instead of plain text. It is omitted for all other
	// message kinds.
	Elicitation *plan.Elicitation `json:"elicitation,omitempty" yaml:"elicitation,omitempty"`

	// CallbackURL is set when the message requires a user action through a
	// dedicated REST callback (e.g. MCP elicitation). Empty for normal chat
	// messages.
	CallbackURL string `json:"callbackURL,omitempty" yaml:"callbackURL,omitempty"`

	// Status indicates the resolution state of interactive MCP prompts.
	// "open"   – waiting for user
	// "done"   – accepted and finished
	// "declined" – user declined
	Status string `json:"status,omitempty" yaml:"status,omitempty"`

	// Interaction contains details for an MCP user-interaction request (e.g.
	// "open the following URL and confirm when done"). When non-nil the UI
	// should render an approval card with a link and Accept/Decline buttons.
	Interaction *UserInteraction `json:"interaction,omitempty" yaml:"interaction,omitempty"`

	// PolicyApproval is non-nil when the system requires explicit user
	// approval before executing a potentially sensitive action (e.g. running
	// an external tool).  The UI should show the approval dialog when the
	// message role == "policyapproval" and Status == "open".
	PolicyApproval *PolicyApproval `json:"policyApproval,omitempty" yaml:"policyApproval,omitempty"`
}

type NextTokenPolicy struct {
	Threshold  int
	Keep       int
	Summarizer SummarizerFunc
	Estimator  TokenEstimatorFunc
}

type Policy interface {
	// Apply processes messages, potentially invoking summarization, returning filtered messages or an error.
	Apply(ctx context.Context, messages []Message) ([]Message, error)
}

type PolicyApproval struct {
	Tool   string                 `json:"tool" yaml:"tool"`                     // tool/function name such as "system.exec"
	Args   map[string]interface{} `json:"args,omitempty" yaml:"args,omitempty"` // flattened argument map
	Reason string                 `json:"reason,omitempty" yaml:"reason,omitempty"`
}

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

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

type SummarizerFunc func(ctx context.Context, messages []Message) (Message, error)

type SummaryPolicy struct {
	Threshold  int
	Summarizer SummarizerFunc
}

type TokenEstimatorFunc func(Message) int

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

type UserInteraction struct {
	URL         string `json:"url" yaml:"url"`
	Description string `json:"description,omitempty" yaml:"description,omitempty"`
}

type VisibilityPolicy struct {
	Mode       string
	Summarizer SummarizerFunc
}