sqlite

type Allocation struct {
	ProjectID   string `json:"project_id"`
	ProjectSlug string `json:"project_slug"`
	BasisPoints int64  `json:"basis_points"`
	Method      string `json:"method"`
	Confidence  string `json:"confidence"`
}

type AllocationInput struct {
	ProjectID   string
	BasisPoints int64
}

type ExportRecord struct {
	ID                     string       `json:"id"`
	OccurredAt             time.Time    `json:"occurred_at"`
	ProjectSlug            string       `json:"project_slug"`
	ProjectLocationPath    string       `json:"project_location_path,omitempty"`
	Provider               string       `json:"provider"`
	Model                  string       `json:"model"`
	Agent                  string       `json:"agent"`
	InputTokens            int64        `json:"input_tokens"`
	OutputTokens           int64        `json:"output_tokens"`
	ReasoningTokens        int64        `json:"reasoning_tokens"`
	CachedInputTokens      int64        `json:"cached_input_tokens"`
	CacheWriteTokens       int64        `json:"cache_write_tokens"`
	TotalTokens            int64        `json:"total_tokens"`
	EstimatedCostUSDMicros int64        `json:"estimated_cost_usd_micros"`
	CaptureQuality         string       `json:"capture_quality"`
	Allocations            []Allocation `json:"allocations"`
}

type ModelCallInput struct {
	ProjectID              string
	ProjectLocationID      string
	WorkContextID          string
	TaskID                 string
	SessionID              string
	TurnID                 string
	Provider               string
	ModelID                string
	AgentName              string
	InputTokens            int64
	OutputTokens           int64
	ReasoningTokens        int64
	CachedInputTokens      int64
	CacheWriteTokens       int64
	EstimatedCostUSDMicros int64
	EstimatedCostEURMicros int64
	OccurredAt             time.Time
	CaptureQuality         string
}

type PricingRecalculateQuery struct {
	From time.Time
	To   time.Time
}

type PricingRuleRecord struct {
	ID        string       `json:"id"`
	Rule      pricing.Rule `json:"rule"`
	CreatedAt time.Time    `json:"created_at"`
}

type ProjectSummary struct {
	ID            string    `json:"id"`
	Slug          string    `json:"slug"`
	Name          string    `json:"name"`
	LocationCount int64     `json:"location_count"`
	TagCount      int64     `json:"tag_count"`
	CreatedAt     time.Time `json:"created_at"`
}

type ProjectTag struct {
	Key   string `json:"key"`
	Value string `json:"value"`
}

type RawEventInput struct {
	Source               string
	SessionID            string
	EventType            string
	Payload              []byte
	OccurredAt           time.Time
	ProjectID            string
	ProjectLocationID    string
	WorkContextID        string
	ResolutionMethod     string
	ResolutionConfidence string
	EvidenceJSON         string
}

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

type TaskInput struct {
	ProjectID string
	Title     string
	TaskType  string
}

type TaskRecord struct {
	ID          string     `json:"id"`
	ProjectSlug string     `json:"project_slug"`
	Title       string     `json:"title"`
	TaskType    string     `json:"task_type"`
	Status      string     `json:"status"`
	Result      string     `json:"result"`
	StartedAt   time.Time  `json:"started_at"`
	FinishedAt  *time.Time `json:"finished_at,omitempty"`
}

type UsageQuery struct {
	From    time.Time
	To      time.Time
	GroupBy []string
}

type UsageReport struct {
	GroupBy                []string   `json:"group_by"`
	Rows                   []UsageRow `json:"rows"`
	TotalTokens            int64      `json:"total_tokens"`
	AllocatedCostUSDMicros int64      `json:"allocated_cost_usd_micros"`
}

type UsageRow struct {
	ProjectSlug            string `json:"project_slug"`
	Provider               string `json:"provider"`
	Model                  string `json:"model"`
	InputTokens            int64  `json:"input_tokens"`
	OutputTokens           int64  `json:"output_tokens"`
	TotalTokens            int64  `json:"total_tokens"`
	AllocatedCostUSDMicros int64  `json:"allocated_cost_usd_micros"`
}

type WorkContextInput struct {
	ProjectID            string
	LocationID           string
	SessionID            string
	CWD                  string
	GitRoot              string
	GitBranch            string
	GitCommit            string
	StartedAt            time.Time
	ResolutionMethod     string
	ResolutionConfidence string
	EvidenceJSON         string
}