database

type ConnectionConfig struct {
	Path            string        // Database file path
	MaxOpenConns    int           // Maximum number of open connections
	MaxIdleConns    int           // Maximum number of idle connections
	ConnMaxLifetime time.Duration // Maximum lifetime of a connection
	ConnMaxIdleTime time.Duration // Maximum idle time of a connection
	BusyTimeout     int           // SQLite busy timeout in milliseconds
	JournalMode     string        // SQLite journal mode (WAL, DELETE, etc.)
	Synchronous     string        // SQLite synchronous mode (NORMAL, FULL, OFF)
}

type CreateMessageParams struct {
	ID           string  `json:"id"`
	SessionID    string  `json:"session_id"`
	Role         string  `json:"role"`
	Content      string  `json:"content"`
	ParentID     *string `json:"parent_id"`
	TokensUsed   *int64  `json:"tokens_used"`
	Model        *string `json:"model"`
	FinishReason *string `json:"finish_reason"`
	Metadata     *string `json:"metadata"`
}

type CreateSessionParams struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Status      string   `json:"status"`
	Model       *string  `json:"model"`
	Temperature *float64 `json:"temperature"`
	MaxTokens   *int64   `json:"max_tokens"`
	Settings    *string  `json:"settings"`
}

type CreateToolExecutionParams struct {
	ID          string  `json:"id"`
	MessageID   string  `json:"message_id"`
	ToolName    string  `json:"tool_name"`
	Input       string  `json:"input"`
	Output      *string `json:"output"`
	Status      string  `json:"status"`
	Error       *string `json:"error"`
	StartedAt   string  `json:"started_at"`
	CompletedAt *string `json:"completed_at"`
	DurationMs  *int64  `json:"duration_ms"`
	RetryCount  int64   `json:"retry_count"`
	Metadata    *string `json:"metadata"`
}

type DB struct {
	*Queries
	// contains filtered or unexported fields
}

type DBTX interface {
	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
	PrepareContext(context.Context, string) (*sql.Stmt, error)
	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}

type GetConversationThreadRow struct {
	ID           string  `json:"id"`
	SessionID    string  `json:"session_id"`
	Role         string  `json:"role"`
	Content      string  `json:"content"`
	Timestamp    string  `json:"timestamp"`
	ParentID     *string `json:"parent_id"`
	TokensUsed   *int64  `json:"tokens_used"`
	Model        *string `json:"model"`
	FinishReason *string `json:"finish_reason"`
	Metadata     *string `json:"metadata"`
}

type GetFailedToolExecutionsParams struct {
	Limit  int64 `json:"limit"`
	Offset int64 `json:"offset"`
}

type GetMessageCountByRoleParams struct {
	SessionID string `json:"session_id"`
	Role      string `json:"role"`
}

type GetMessagesByTimeRangeParams struct {
	SessionID   string `json:"session_id"`
	Timestamp   string `json:"timestamp"`
	Timestamp_2 string `json:"timestamp_2"`
}

type GetSessionWithMessageCountRow struct {
	ID           string   `json:"id"`
	Name         string   `json:"name"`
	CreatedAt    string   `json:"created_at"`
	UpdatedAt    string   `json:"updated_at"`
	Status       string   `json:"status"`
	Model        *string  `json:"model"`
	Temperature  *float64 `json:"temperature"`
	MaxTokens    *int64   `json:"max_tokens"`
	Settings     *string  `json:"settings"`
	MessageCount int64    `json:"message_count"`
}

type GetToolExecutionStatsRow struct {
	TotalExecutions      int64       `json:"total_executions"`
	SuccessfulExecutions int64       `json:"successful_executions"`
	FailedExecutions     int64       `json:"failed_executions"`
	TimeoutExecutions    int64       `json:"timeout_executions"`
	AvgDurationMs        *float64    `json:"avg_duration_ms"`
	MaxDurationMs        interface{} `json:"max_duration_ms"`
	MinDurationMs        interface{} `json:"min_duration_ms"`
}

type GetToolExecutionsByTimeRangeParams struct {
	StartedAt   string `json:"started_at"`
	StartedAt_2 string `json:"started_at_2"`
}

type GetToolUsageStatsRow struct {
	ToolName       string   `json:"tool_name"`
	TotalUses      int64    `json:"total_uses"`
	SuccessfulUses int64    `json:"successful_uses"`
	FailedUses     int64    `json:"failed_uses"`
	AvgDurationMs  *float64 `json:"avg_duration_ms"`
}

type ListMessagesByRoleParams struct {
	SessionID string `json:"session_id"`
	Role      string `json:"role"`
}

type ListMessagesBySessionPaginatedParams struct {
	SessionID string `json:"session_id"`
	Limit     int64  `json:"limit"`
	Offset    int64  `json:"offset"`
}

type ListSessionsByStatusParams struct {
	Status string `json:"status"`
	Limit  int64  `json:"limit"`
	Offset int64  `json:"offset"`
}

type ListSessionsParams struct {
	Limit  int64 `json:"limit"`
	Offset int64 `json:"offset"`
}

type ListToolExecutionsByNameParams struct {
	ToolName string `json:"tool_name"`
	Limit    int64  `json:"limit"`
	Offset   int64  `json:"offset"`
}

type ListToolExecutionsBySessionParams struct {
	SessionID string `json:"session_id"`
	Limit     int64  `json:"limit"`
	Offset    int64  `json:"offset"`
}

type ListToolExecutionsByStatusParams struct {
	Status string `json:"status"`
	Limit  int64  `json:"limit"`
	Offset int64  `json:"offset"`
}

type Message struct {
	ID           string  `json:"id"`
	SessionID    string  `json:"session_id"`
	Role         string  `json:"role"`
	Content      string  `json:"content"`
	Timestamp    string  `json:"timestamp"`
	ParentID     *string `json:"parent_id"`
	TokensUsed   *int64  `json:"tokens_used"`
	Model        *string `json:"model"`
	FinishReason *string `json:"finish_reason"`
	Metadata     *string `json:"metadata"`
}

type Metadata struct {
	Key       string `json:"key"`
	Value     string `json:"value"`
	CreatedAt string `json:"created_at"`
	UpdatedAt string `json:"updated_at"`
}

type Migration struct {
	Version     int
	Name        string
	UpSQL       string
	DownSQL     string
	AppliedAt   time.Time
	Description string
}

type MigrationStatus struct {
	CurrentVersion int
	Applied        []Migration
	Pending        []Migration
}

type Querier interface {
	ArchiveSession(ctx context.Context, id string) error
	CountSessions(ctx context.Context) (int64, error)
	CountSessionsByStatus(ctx context.Context, status string) (int64, error)
	CreateMessage(ctx context.Context, arg CreateMessageParams) error
	CreateSession(ctx context.Context, arg CreateSessionParams) error
	CreateToolExecution(ctx context.Context, arg CreateToolExecutionParams) error
	DeleteMessage(ctx context.Context, id string) error
	DeleteMessagesBySession(ctx context.Context, sessionID string) error
	DeleteMetadata(ctx context.Context, key string) error
	DeleteSession(ctx context.Context, id string) error
	DeleteToolExecution(ctx context.Context, id string) error
	DeleteToolExecutionsByMessage(ctx context.Context, messageID string) error
	GetConversationThread(ctx context.Context, id string) ([]GetConversationThreadRow, error)
	GetFailedToolExecutions(ctx context.Context, arg GetFailedToolExecutionsParams) ([]ToolExecution, error)
	GetLatestMessage(ctx context.Context, sessionID string) (Message, error)
	GetMessage(ctx context.Context, id string) (Message, error)
	GetMessageCount(ctx context.Context, sessionID string) (int64, error)
	GetMessageCountByRole(ctx context.Context, arg GetMessageCountByRoleParams) (int64, error)
	GetMessagesByTimeRange(ctx context.Context, arg GetMessagesByTimeRangeParams) ([]Message, error)
	GetMetadata(ctx context.Context, key string) (Metadata, error)
	GetPendingToolExecutions(ctx context.Context) ([]ToolExecution, error)
	GetRecentToolExecutions(ctx context.Context, limit int64) ([]ToolExecution, error)
	GetSession(ctx context.Context, id string) (Session, error)
	GetSessionByID(ctx context.Context, id string) (Session, error)
	GetSessionWithMessageCount(ctx context.Context, id string) (GetSessionWithMessageCountRow, error)
	GetToolExecution(ctx context.Context, id string) (ToolExecution, error)
	GetToolExecutionCount(ctx context.Context, messageID string) (int64, error)
	GetToolExecutionCountByStatus(ctx context.Context, status string) (int64, error)
	GetToolExecutionStats(ctx context.Context, toolName string) (GetToolExecutionStatsRow, error)
	GetToolExecutionsByTimeRange(ctx context.Context, arg GetToolExecutionsByTimeRangeParams) ([]ToolExecution, error)
	GetToolUsageStats(ctx context.Context) ([]GetToolUsageStatsRow, error)
	GetTotalTokensUsed(ctx context.Context, sessionID string) (interface{}, error)
	HardDeleteSession(ctx context.Context, id string) error
	IncrementRetryCount(ctx context.Context, id string) error
	ListAllSessions(ctx context.Context) ([]Session, error)
	ListMessagesByParent(ctx context.Context, parentID *string) ([]Message, error)
	ListMessagesByRole(ctx context.Context, arg ListMessagesByRoleParams) ([]Message, error)
	ListMessagesBySession(ctx context.Context, sessionID string) ([]Message, error)
	ListMessagesBySessionPaginated(ctx context.Context, arg ListMessagesBySessionPaginatedParams) ([]Message, error)
	ListMetadata(ctx context.Context) ([]Metadata, error)
	ListSessions(ctx context.Context, arg ListSessionsParams) ([]Session, error)
	ListSessionsByStatus(ctx context.Context, arg ListSessionsByStatusParams) ([]Session, error)
	ListToolExecutionsByMessage(ctx context.Context, messageID string) ([]ToolExecution, error)
	ListToolExecutionsByName(ctx context.Context, arg ListToolExecutionsByNameParams) ([]ToolExecution, error)
	ListToolExecutionsBySession(ctx context.Context, arg ListToolExecutionsBySessionParams) ([]ToolExecution, error)
	ListToolExecutionsByStatus(ctx context.Context, arg ListToolExecutionsByStatusParams) ([]ToolExecution, error)
	MetadataExists(ctx context.Context, key string) (bool, error)
	SearchMessages(ctx context.Context, arg SearchMessagesParams) ([]Message, error)
	SearchSessions(ctx context.Context, arg SearchSessionsParams) ([]Session, error)
	SetMetadata(ctx context.Context, arg SetMetadataParams) error
	TouchSession(ctx context.Context, id string) error
	UpdateMessage(ctx context.Context, arg UpdateMessageParams) error
	UpdateSession(ctx context.Context, arg UpdateSessionParams) error
	UpdateSessionStatus(ctx context.Context, arg UpdateSessionStatusParams) error
	UpdateToolExecutionError(ctx context.Context, arg UpdateToolExecutionErrorParams) error
	UpdateToolExecutionOutput(ctx context.Context, arg UpdateToolExecutionOutputParams) error
	UpdateToolExecutionStatus(ctx context.Context, arg UpdateToolExecutionStatusParams) error
}

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

type SearchMessagesParams struct {
	SessionID string `json:"session_id"`
	Content   string `json:"content"`
	Limit     int64  `json:"limit"`
	Offset    int64  `json:"offset"`
}

type SearchSessionsParams struct {
	Name   string `json:"name"`
	ID     string `json:"id"`
	Limit  int64  `json:"limit"`
	Offset int64  `json:"offset"`
}

type Session struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	CreatedAt   string   `json:"created_at"`
	UpdatedAt   string   `json:"updated_at"`
	Status      string   `json:"status"`
	Model       *string  `json:"model"`
	Temperature *float64 `json:"temperature"`
	MaxTokens   *int64   `json:"max_tokens"`
	Settings    *string  `json:"settings"`
}

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

type ToolExecution struct {
	ID          string  `json:"id"`
	MessageID   string  `json:"message_id"`
	ToolName    string  `json:"tool_name"`
	Input       string  `json:"input"`
	Output      *string `json:"output"`
	Status      string  `json:"status"`
	Error       *string `json:"error"`
	StartedAt   string  `json:"started_at"`
	CompletedAt *string `json:"completed_at"`
	DurationMs  *int64  `json:"duration_ms"`
	RetryCount  int64   `json:"retry_count"`
	Metadata    *string `json:"metadata"`
}

type UpdateMessageParams struct {
	Content      string  `json:"content"`
	TokensUsed   *int64  `json:"tokens_used"`
	FinishReason *string `json:"finish_reason"`
	Metadata     *string `json:"metadata"`
	ID           string  `json:"id"`
}

type UpdateSessionParams struct {
	Name        string   `json:"name"`
	Model       *string  `json:"model"`
	Temperature *float64 `json:"temperature"`
	MaxTokens   *int64   `json:"max_tokens"`
	Settings    *string  `json:"settings"`
	ID          string   `json:"id"`
}

type UpdateSessionStatusParams struct {
	Status string `json:"status"`
	ID     string `json:"id"`
}

type UpdateToolExecutionErrorParams struct {
	Error      *string `json:"error"`
	DurationMs *int64  `json:"duration_ms"`
	ID         string  `json:"id"`
}

type UpdateToolExecutionOutputParams struct {
	Output     *string `json:"output"`
	Status     string  `json:"status"`
	DurationMs *int64  `json:"duration_ms"`
	ID         string  `json:"id"`
}

type UpdateToolExecutionStatusParams struct {
	Status string `json:"status"`
	ID     string `json:"id"`
}