checkpoint

type Checkpoint struct {
	// ID is the unique identifier for this checkpoint.
	ID string `json:"id"`

	// SessionID is the session this checkpoint belongs to.
	SessionID string `json:"session_id"`

	// TenantID is the organization this checkpoint belongs to.
	TenantID string `json:"tenant_id"`

	// TeamID is the team this checkpoint belongs to.
	TeamID string `json:"team_id"`

	// ProjectID is the project identifier for this checkpoint.
	ProjectID string `json:"project_id"`

	// ProjectPath is the project context for this checkpoint.
	ProjectPath string `json:"project_path"`

	// Name is a human-readable name for the checkpoint.
	Name string `json:"name"`

	// Description describes what was happening at checkpoint time.
	Description string `json:"description"`

	// Summary is a condensed summary of the session state.
	Summary string `json:"summary"`

	// Context contains relevant context fragments.
	Context string `json:"context"`

	// FullState contains the complete session state.
	FullState string `json:"full_state"`

	// TokenCount is the approximate token count of full state.
	TokenCount int32 `json:"token_count"`

	// Threshold is the context percentage at which this was created.
	Threshold float64 `json:"threshold"`

	// AutoCreated indicates if this was auto-generated.
	AutoCreated bool `json:"auto_created"`

	// Metadata contains additional checkpoint metadata.
	Metadata map[string]string `json:"metadata,omitempty"`

	// CreatedAt is when this checkpoint was created.
	CreatedAt time.Time `json:"created_at"`
}

type Config struct {
	// VectorSize is the dimension of embedding vectors (default: 1536)
	VectorSize uint64

	// MaxCheckpointsPerSession limits checkpoints per session (default: 10)
	MaxCheckpointsPerSession int

	// AutoCheckpointThresholds are context % levels for auto-checkpoint.
	AutoCheckpointThresholds []float64
}

type ListRequest struct {
	SessionID   string
	TenantID    string
	TeamID      string
	ProjectID   string
	ProjectPath string
	Limit       int
	AutoOnly    bool // Only return auto-created checkpoints
}

type ResumeLevel string

type ResumeRequest struct {
	CheckpointID string
	TenantID     string
	TeamID       string
	ProjectID    string
	Level        ResumeLevel
}

type ResumeResponse struct {
	Checkpoint *Checkpoint
	Content    string // Content based on resume level
	TokenCount int32
}

type SaveRequest struct {
	SessionID   string
	TenantID    string
	TeamID      string
	ProjectID   string
	ProjectPath string
	Name        string
	Description string
	Summary     string
	Context     string
	FullState   string
	TokenCount  int32
	Threshold   float64
	AutoCreated bool
	Metadata    map[string]string
}

type Service interface {
	// Save creates a new checkpoint.
	Save(ctx context.Context, req *SaveRequest) (*Checkpoint, error)

	// List retrieves checkpoints for a session or project.
	List(ctx context.Context, req *ListRequest) ([]*Checkpoint, error)

	// Resume restores a checkpoint at the specified level.
	Resume(ctx context.Context, req *ResumeRequest) (*ResumeResponse, error)

	// Get retrieves a checkpoint by ID.
	Get(ctx context.Context, tenantID, teamID, projectID, checkpointID string) (*Checkpoint, error)

	// Delete removes a checkpoint.
	Delete(ctx context.Context, tenantID, teamID, projectID, checkpointID string) error

	// Close closes the service.
	Close() error
}