storage

type Alias struct {
	Alias  string `json:"alias"`
	NodeID string `json:"node_id"`
}

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

type DAGNode struct {
	ID        string
	ParentID  string
	Role      string
	Content   string
	Model     string
	CreatedAt time.Time
	Metadata  map[string]string
}

type Node struct {
	ID                  string          `json:"id"`
	ParentID            string          `json:"parent_id,omitempty"`
	RootID              string          `json:"root_id,omitempty"`
	Sequence            int             `json:"sequence"`
	NodeType            NodeType        `json:"node_type"`
	Content             string          `json:"content"`
	Provider            string          `json:"provider,omitempty"`
	Model               string          `json:"model,omitempty"`
	TokensIn            int             `json:"tokens_in,omitempty"`
	TokensOut           int             `json:"tokens_out,omitempty"`
	TokensCacheRead     int             `json:"tokens_cache_read,omitempty"`
	TokensCacheCreation int             `json:"tokens_cache_creation,omitempty"`
	TokensReasoning     int             `json:"tokens_reasoning,omitempty"`
	LatencyMs           int             `json:"latency_ms,omitempty"`
	StopReason          string          `json:"stop_reason,omitempty"`
	OutputGroupID       string          `json:"output_group_id,omitempty"`
	Status              string          `json:"status,omitempty"`
	Title               string          `json:"title,omitempty"`
	SystemPrompt        string          `json:"system_prompt,omitempty"`
	CreatedAt           time.Time       `json:"created_at"`
	Metadata            json.RawMessage `json:"metadata,omitempty"`
}

type NodeType string

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

type Store interface {
	CreateNode(ctx context.Context, node *Node) error
	GetNode(ctx context.Context, id string) (*Node, error)
	GetNodeByPrefix(ctx context.Context, prefix string) (*Node, error)
	GetNodeChildren(ctx context.Context, id string) ([]*Node, error)
	GetSubtree(ctx context.Context, id string) ([]*Node, error)
	GetAncestors(ctx context.Context, id string) ([]*Node, error)
	ListRootNodes(ctx context.Context) ([]*Node, error)
	UpdateNode(ctx context.Context, node *Node) error
	DeleteNode(ctx context.Context, id string) error
	CreateAlias(ctx context.Context, alias, nodeID string) error
	DeleteAlias(ctx context.Context, alias string) error
	GetNodeByAlias(ctx context.Context, alias string) (*Node, error)
	ListAliases(ctx context.Context, nodeID string) ([]Alias, error)
	IndexToolIDs(ctx context.Context, nodeID string, toolIDs []string, role string) error
	GetOrphanedToolUses(ctx context.Context, rootID string) ([]string, error)
	Close() error
}