storage

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

type AnalyticsStore interface {
	// RecordCost persists a cost record for a completed request.
	RecordCost(ctx context.Context, rec *CostRecord) error
	// GetUsageStats returns aggregated usage stats grouped by provider/model,
	// filtered to records created after the given time.
	GetUsageStats(ctx context.Context, since time.Time) ([]UsageStats, error)
	// GetCostSummary returns total cost grouped by provider/model since the given time.
	GetCostSummary(ctx context.Context, since time.Time) ([]UsageStats, error)
	// GetProviderHealth returns health metrics derived from actual request data.
	GetProviderHealth(ctx context.Context, since time.Time) ([]ProviderHealthRecord, error)
}

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

type CostRecord struct {
	ID                  int64     `json:"id"`
	Provider            string    `json:"provider"`
	Model               string    `json:"model"`
	InputTokens         int       `json:"input_tokens"`
	OutputTokens        int       `json:"output_tokens"`
	CacheReadTokens     int       `json:"cache_read_tokens"`
	CacheCreationTokens int       `json:"cache_creation_tokens"`
	CostUSD             float64   `json:"cost_usd"`
	LatencyMs           int       `json:"latency_ms"`
	CreatedAt           time.Time `json:"created_at"`
}

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 Option func(*openConfig)

type ProviderHealthRecord struct {
	Provider      string    `json:"provider"`
	RequestCount  int       `json:"request_count"`
	ErrorCount    int       `json:"error_count"`
	ErrorRate     float64   `json:"error_rate"`
	AvgLatencyMs  float64   `json:"avg_latency_ms"`
	P50LatencyMs  float64   `json:"p50_latency_ms"`
	P95LatencyMs  float64   `json:"p95_latency_ms"`
	P99LatencyMs  float64   `json:"p99_latency_ms"`
	LastRequestAt time.Time `json:"last_request_at"`
}

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

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 returns tool_use IDs on ancestorIDs with no matching
	// tool_result on the same path. Map keys are node IDs.
	GetOrphanedToolUses(ctx context.Context, ancestorIDs []string) (map[string][]string, error)
	Close() error
}

type UsageStats struct {
	Provider           string  `json:"provider"`
	Model              string  `json:"model"`
	RequestCount       int     `json:"request_count"`
	TotalInputTokens   int     `json:"total_input_tokens"`
	TotalOutputTokens  int     `json:"total_output_tokens"`
	TotalCacheRead     int     `json:"total_cache_read_tokens"`
	TotalCacheCreation int     `json:"total_cache_creation_tokens"`
	ErrorCount         int     `json:"error_count"`
	ErrorRate          float64 `json:"error_rate"`
	AvgLatencyMs       float64 `json:"avg_latency_ms"`
	TotalCostUSD       float64 `json:"total_cost_usd"`
}

type VirtualKey struct {
	ID        string
	Name      string
	Provider  string
	LimitUSD  float64 // <= 0 means unlimited
	UsedUSD   float64
	TokensIn  int
	TokensOut int
	CreatedAt time.Time
}