dbgen

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

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 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 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 ListCheckpointWritesParams struct {
	UserID       string
	ThreadID     string
	CheckpointNs string
	CheckpointID 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 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
}

type Querier interface {
	DeleteAgentMemory(ctx context.Context, arg DeleteAgentMemoryParams) error
	DeleteExpiredMemories(ctx context.Context) 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)
	GetTask(ctx context.Context, id string) (Task, error)
	GetTool(ctx context.Context, id string) (Tool, error)
	GetToolServer(ctx context.Context, name string) (Toolserver, error)
	HardDeleteCrewAIMemory(ctx context.Context, arg HardDeleteCrewAIMemoryParams) error
	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)
	ListCheckpointWrites(ctx context.Context, arg ListCheckpointWritesParams) ([]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)
	ListSessions(ctx context.Context, userID string) ([]Session, error)
	ListSessionsForAgent(ctx context.Context, arg ListSessionsForAgentParams) ([]Session, error)
	ListSessionsForAgentAllUsers(ctx context.Context, agentID *string) ([]Session, error)
	ListTasksForSession(ctx context.Context, sessionID *string) ([]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, id string) 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
	UpsertTask(ctx context.Context, arg UpsertTaskParams) 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
	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 SoftDeleteCheckpointWritesParams struct {
	UserID   string
	ThreadID string
}

type SoftDeleteCheckpointsParams struct {
	UserID   string
	ThreadID string
}

type SoftDeleteSessionParams 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
}

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
}

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

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

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

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