dbgen

type Agent struct {
	ID           string
	CreatedAt    *time.Time
	UpdatedAt    *time.Time
	DeletedAt    *time.Time
	Type         string
	Config       *adk.AgentConfig
	WorkloadType string
}

type CreateSessionShareParams struct {
	Token     string
	SessionID string
	UserID    string
	ReadOnly  bool
}

type CrewaiAgentMemory struct {
	UserID     string
	ThreadID   string
	CreatedAt  *time.Time
	UpdatedAt  *time.Time
	DeletedAt  *time.Time
	MemoryData string
}

type CrewaiFlowState struct {
	UserID     string
	ThreadID   string
	MethodName string
	CreatedAt  *time.Time
	UpdatedAt  *time.Time
	DeletedAt  *time.Time
	StateData  string
}

type DBTX interface {
	Exec(context.Context, string, ...interface{}) (pgconn.CommandTag, error)
	Query(context.Context, string, ...interface{}) (pgx.Rows, error)
	QueryRow(context.Context, string, ...interface{}) pgx.Row
}

type DeleteAgentMemoryParams struct {
	AgentName *string
	UserID    *string
}

type DeleteSessionShareParams struct {
	Token     string
	SessionID string
	UserID    string
}

type Event struct {
	ID        string
	UserID    string
	SessionID *string
	CreatedAt *time.Time
	UpdatedAt *time.Time
	DeletedAt *time.Time
	Data      string
}

type Feedback struct {
	ID           int64
	CreatedAt    *time.Time
	UpdatedAt    *time.Time
	DeletedAt    *time.Time
	UserID       string
	MessageID    *int64
	IsPositive   bool
	FeedbackText string
	IssueType    *database.FeedbackIssueType
}

type GetCheckpointParams struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
	CheckpointID string
}

type GetEventParams struct {
	ID     string
	UserID string
}

type GetLatestCrewAIFlowStateParams struct {
	UserID   string
	ThreadID string
}

type GetPushNotificationParams struct {
	TaskID string
	ID     string
}

type GetSessionParams struct {
	ID     string
	UserID string
}

type GetTaskParams struct {
	ID     string
	UserID *string
}

type HardDeleteCrewAIMemoryParams struct {
	UserID   string
	ThreadID string
}

type InsertEventParams struct {
	ID        string
	UserID    string
	SessionID *string
	Data      string
}

type InsertFeedbackParams struct {
	UserID       string
	MessageID    *int64
	IsPositive   bool
	FeedbackText string
	IssueType    *database.FeedbackIssueType
}

type InsertMemoryParams struct {
	AgentName   *string
	UserID      *string
	Content     *string
	Embedding   pgvector_go.Vector
	Metadata    *string
	ExpiresAt   *time.Time
	AccessCount *int64
}

type LgCheckpoint struct {
	UserID             string
	ThreadID           string
	CheckpointNs       string
	CheckpointID       string
	ParentCheckpointID *string
	CreatedAt          *time.Time
	UpdatedAt          *time.Time
	DeletedAt          *time.Time
	Metadata           string
	Checkpoint         string
	CheckpointType     string
	Version            int64
}

type LgCheckpointWrite struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
	CheckpointID string
	WriteIdx     int64
	Value        string
	ValueType    string
	Channel      string
	TaskID       string
	CreatedAt    *time.Time
	UpdatedAt    *time.Time
	DeletedAt    *time.Time
}

type ListAgentMemoriesParams struct {
	AgentName   *string
	AgentName_2 *string
	UserID      *string
}

type ListCheckpointWritesForCheckpointsParams struct {
	UserID        string
	ThreadID      string
	CheckpointNs  string
	CheckpointIds []string
}

type ListCheckpointsLimitParams struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
	Limit        int32
}

type ListCheckpointsParams struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
}

type ListEventsByContextIDLimitParams struct {
	SessionID *string
	Limit     int32
}

type ListEventsForSessionAscLimitParams struct {
	SessionID *string
	UserID    string
	Column3   time.Time
	Limit     int32
}

type ListEventsForSessionAscParams struct {
	SessionID *string
	UserID    string
	Column3   time.Time
}

type ListEventsForSessionDescLimitParams struct {
	SessionID *string
	UserID    string
	Column3   time.Time
	Limit     int32
}

type ListEventsForSessionDescParams struct {
	SessionID *string
	UserID    string
	Column3   time.Time
}

type ListSessionsForAgentParams struct {
	AgentID *string
	UserID  string
}

type ListSessionsForAgentRow struct {
	ID            string
	UserID        string
	Name          *string
	CreatedAt     *time.Time
	UpdatedAt     *time.Time
	DeletedAt     *time.Time
	AgentID       *string
	Source        *string
	ShareToken    interface{}
	ShareReadOnly interface{}
}

type ListTasksForSessionParams struct {
	SessionID *string
	UserID    *string
}

type ListToolsForServerParams struct {
	ServerName string
	GroupKind  string
}

type Memory struct {
	ID          string
	AgentName   *string
	UserID      *string
	Content     *string
	Embedding   pgvector_go.Vector
	Metadata    *string
	CreatedAt   *time.Time
	ExpiresAt   *time.Time
	AccessCount *int64
}

type PushNotification struct {
	ID              string
	TaskID          string
	CreatedAt       *time.Time
	UpdatedAt       *time.Time
	DeletedAt       *time.Time
	Data            string
	ProtocolVersion *string
}

type Querier interface {
	CreateSessionShare(ctx context.Context, arg CreateSessionShareParams) (SessionShare, error)
	DeleteAgentMemory(ctx context.Context, arg DeleteAgentMemoryParams) error
	DeleteExpiredMemories(ctx context.Context) error
	DeleteSessionShare(ctx context.Context, arg DeleteSessionShareParams) error
	ExtendMemoryTTL(ctx context.Context) error
	GetAgent(ctx context.Context, id string) (Agent, error)
	GetCheckpoint(ctx context.Context, arg GetCheckpointParams) (LgCheckpoint, error)
	GetEvent(ctx context.Context, arg GetEventParams) (Event, error)
	GetLatestCrewAIFlowState(ctx context.Context, arg GetLatestCrewAIFlowStateParams) (CrewaiFlowState, error)
	GetPushNotification(ctx context.Context, arg GetPushNotificationParams) (PushNotification, error)
	GetSession(ctx context.Context, arg GetSessionParams) (Session, error)
	GetSessionShareByToken(ctx context.Context, token string) (SessionShare, error)
	// Task ownership: a task belongs to task.user_id. A NULL user_id (row written
	// before the owner column existed, or by a pre-upgrade pod during a rolling
	// upgrade) is only visible to, and claimable by, a caller whose session id
	// maps to exactly one user across its whole history (deleted sessions
	// included) and that user is the caller. Anything ambiguous stays hidden
	// rather than guessed. This mirrors the backfill rule in migration
	// 000007_task_owner.
	//
	// The resolving session must also have existed at or before the task was
	// written (s.created_at <= task.created_at). Without that bound, a task
	// whose original session is gone entirely (hard deleted, or never migrated)
	// would become claimable by whoever is first to create a brand new session
	// reusing that same id after the fact, handing them a stranger's task.
	GetTask(ctx context.Context, arg GetTaskParams) (Task, error)
	GetTaskOwner(ctx context.Context, id string) (*string, error)
	GetTool(ctx context.Context, id string) (Tool, error)
	GetToolServer(ctx context.Context, name string) (Toolserver, error)
	HardDeleteCrewAIMemory(ctx context.Context, arg HardDeleteCrewAIMemoryParams) error
	// Lock rows in id order to avoid deadlocks between concurrent overlapping increments.
	IncrementMemoryAccessCount(ctx context.Context, dollar_1 []string) error
	InsertEvent(ctx context.Context, arg InsertEventParams) error
	InsertFeedback(ctx context.Context, arg InsertFeedbackParams) error
	InsertMemory(ctx context.Context, arg InsertMemoryParams) (string, error)
	ListAgentMemories(ctx context.Context, arg ListAgentMemoriesParams) ([]Memory, error)
	ListAgents(ctx context.Context) ([]Agent, error)
	ListCheckpointWritesForCheckpoints(ctx context.Context, arg ListCheckpointWritesForCheckpointsParams) ([]LgCheckpointWrite, error)
	ListCheckpoints(ctx context.Context, arg ListCheckpointsParams) ([]LgCheckpoint, error)
	ListCheckpointsLimit(ctx context.Context, arg ListCheckpointsLimitParams) ([]LgCheckpoint, error)
	ListEventsByContextID(ctx context.Context, sessionID *string) ([]Event, error)
	ListEventsByContextIDLimit(ctx context.Context, arg ListEventsByContextIDLimitParams) ([]Event, error)
	ListEventsForSessionAsc(ctx context.Context, arg ListEventsForSessionAscParams) ([]Event, error)
	ListEventsForSessionAscLimit(ctx context.Context, arg ListEventsForSessionAscLimitParams) ([]Event, error)
	ListEventsForSessionDesc(ctx context.Context, arg ListEventsForSessionDescParams) ([]Event, error)
	ListEventsForSessionDescLimit(ctx context.Context, arg ListEventsForSessionDescLimitParams) ([]Event, error)
	ListFeedback(ctx context.Context, userID string) ([]Feedback, error)
	ListPushNotifications(ctx context.Context, taskID string) ([]PushNotification, error)
	ListSessionSharesBySession(ctx context.Context, sessionID string) ([]SessionShare, error)
	ListSessions(ctx context.Context, userID string) ([]Session, error)
	ListSessionsForAgent(ctx context.Context, arg ListSessionsForAgentParams) ([]ListSessionsForAgentRow, error)
	ListSessionsForAgentAllUsers(ctx context.Context, agentID *string) ([]Session, error)
	ListTasksForSession(ctx context.Context, arg ListTasksForSessionParams) ([]Task, error)
	ListToolServers(ctx context.Context) ([]Toolserver, error)
	ListTools(ctx context.Context) ([]Tool, error)
	ListToolsForServer(ctx context.Context, arg ListToolsForServerParams) ([]Tool, error)
	// Memory uses hard DELETE (not soft deletes), so no deleted_at filter is needed.
	// COALESCE guards against NULL embeddings (score=0 rather than NULL); rows are still ordered last by the ORDER BY clause.
	SearchAgentMemory(ctx context.Context, arg SearchAgentMemoryParams) ([]SearchAgentMemoryRow, error)
	SearchCrewAIMemoryByTask(ctx context.Context, arg SearchCrewAIMemoryByTaskParams) ([]CrewaiAgentMemory, error)
	SearchCrewAIMemoryByTaskLimit(ctx context.Context, arg SearchCrewAIMemoryByTaskLimitParams) ([]CrewaiAgentMemory, error)
	SoftDeleteAgent(ctx context.Context, id string) error
	SoftDeleteCheckpointWrites(ctx context.Context, arg SoftDeleteCheckpointWritesParams) error
	SoftDeleteCheckpoints(ctx context.Context, arg SoftDeleteCheckpointsParams) error
	SoftDeleteEvent(ctx context.Context, id string) error
	SoftDeletePushNotification(ctx context.Context, taskID string) error
	SoftDeleteSession(ctx context.Context, arg SoftDeleteSessionParams) error
	SoftDeleteTask(ctx context.Context, arg SoftDeleteTaskParams) error
	SoftDeleteToolServer(ctx context.Context, arg SoftDeleteToolServerParams) error
	SoftDeleteToolsForServer(ctx context.Context, arg SoftDeleteToolsForServerParams) error
	TaskExists(ctx context.Context, id string) (bool, error)
	UpsertAgent(ctx context.Context, arg UpsertAgentParams) error
	UpsertCheckpoint(ctx context.Context, arg UpsertCheckpointParams) error
	UpsertCheckpointWrite(ctx context.Context, arg UpsertCheckpointWriteParams) error
	UpsertCrewAIFlowState(ctx context.Context, arg UpsertCrewAIFlowStateParams) error
	UpsertCrewAIMemory(ctx context.Context, arg UpsertCrewAIMemoryParams) error
	UpsertPushNotification(ctx context.Context, arg UpsertPushNotificationParams) error
	UpsertSession(ctx context.Context, arg UpsertSessionParams) error
	UpsertShareAccess(ctx context.Context, arg UpsertShareAccessParams) error
	// UpsertTask returns the upserted id, or no rows when the write was rejected:
	// the id belongs to another user, or it belongs to a soft-deleted task (a
	// deleted id is never updated or resurrected, it stays burned). Callers map
	// "no rows" to a conflict error.
	UpsertTask(ctx context.Context, arg UpsertTaskParams) (string, error)
	UpsertTool(ctx context.Context, arg UpsertToolParams) error
	UpsertToolServer(ctx context.Context, arg UpsertToolServerParams) (Toolserver, error)
}

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

type SearchAgentMemoryParams struct {
	Embedding   pgvector_go.Vector
	AgentName   *string
	AgentName_2 *string
	UserID      *string
	Limit       int32
}

type SearchAgentMemoryRow struct {
	ID          string
	AgentName   *string
	UserID      *string
	Content     *string
	Embedding   pgvector_go.Vector
	Metadata    *string
	CreatedAt   *time.Time
	ExpiresAt   *time.Time
	AccessCount *int64
	Score       interface{}
}

type SearchCrewAIMemoryByTaskLimitParams struct {
	UserID     string
	ThreadID   string
	MemoryData string
	Limit      int32
}

type SearchCrewAIMemoryByTaskParams struct {
	UserID     string
	ThreadID   string
	MemoryData string
}

type Session struct {
	ID        string
	UserID    string
	Name      *string
	CreatedAt *time.Time
	UpdatedAt *time.Time
	DeletedAt *time.Time
	AgentID   *string
	Source    *string
}

type SessionShare struct {
	ID        int64
	Token     string
	SessionID string
	UserID    string
	ReadOnly  bool
	CreatedAt pgtype.Timestamp
}

type SessionShareAccess struct {
	UserID     string
	ShareID    int64
	AccessedAt pgtype.Timestamp
}

type SoftDeleteCheckpointWritesParams struct {
	UserID   string
	ThreadID string
}

type SoftDeleteCheckpointsParams struct {
	UserID   string
	ThreadID string
}

type SoftDeleteSessionParams struct {
	ID     string
	UserID string
}

type SoftDeleteTaskParams struct {
	ID     string
	UserID *string
}

type SoftDeleteToolServerParams struct {
	Name      string
	GroupKind string
}

type SoftDeleteToolsForServerParams struct {
	ServerName string
	GroupKind  string
}

type Task struct {
	ID              string
	CreatedAt       *time.Time
	UpdatedAt       *time.Time
	DeletedAt       *time.Time
	Data            string
	SessionID       *string
	ProtocolVersion *string
	UserID          *string
}

type Tool struct {
	ID          string
	ServerName  string
	GroupKind   string
	CreatedAt   *time.Time
	UpdatedAt   *time.Time
	DeletedAt   *time.Time
	Description *string
}

type Toolserver struct {
	Name          string
	GroupKind     string
	CreatedAt     *time.Time
	UpdatedAt     *time.Time
	DeletedAt     *time.Time
	Description   *string
	LastConnected *time.Time
}

type UpsertAgentParams struct {
	ID           string
	Type         string
	WorkloadType string
	Config       *adk.AgentConfig
}

type UpsertCheckpointParams struct {
	UserID             string
	ThreadID           string
	CheckpointNs       string
	CheckpointID       string
	ParentCheckpointID *string
	Metadata           string
	Checkpoint         string
	CheckpointType     string
	Version            int64
}

type UpsertCheckpointWriteParams struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
	CheckpointID string
	WriteIdx     int64
	Value        string
	ValueType    string
	Channel      string
	TaskID       string
}

type UpsertCrewAIFlowStateParams struct {
	UserID     string
	ThreadID   string
	MethodName string
	StateData  string
}

type UpsertCrewAIMemoryParams struct {
	UserID     string
	ThreadID   string
	MemoryData string
}

type UpsertPushNotificationParams struct {
	ID              string
	TaskID          string
	Data            string
	ProtocolVersion *string
}

type UpsertSessionParams struct {
	ID      string
	UserID  string
	Name    *string
	AgentID *string
	Source  *string
}

type UpsertShareAccessParams struct {
	UserID  string
	ShareID int64
}

type UpsertTaskParams struct {
	ID              string
	Data            string
	SessionID       *string
	ProtocolVersion *string
	UserID          *string
}

type UpsertToolParams struct {
	ID          string
	ServerName  string
	GroupKind   string
	Description *string
}

type UpsertToolServerParams struct {
	Name          string
	GroupKind     string
	Description   *string
	LastConnected *time.Time
}