store

type AIConversation struct {
	UID       string
	Title     string
	ParrotID  string
	RowStatus RowStatus
	CreatedTs int64
	UpdatedTs int64
	ID        int32
	CreatorID int32
	Pinned    bool
}

type AIMessage struct {
	UID            string
	Type           AIMessageType
	Role           AIMessageRole
	Content        string
	Metadata       string
	CreatedTs      int64
	ID             int32
	ConversationID int32
}

type AIMessageRole string

type AIMessageType string

type Activity struct {
	Payload   *storepb.ActivityPayload
	Type      ActivityType
	Level     ActivityLevel
	CreatedTs int64
	ID        int32
	CreatorID int32
}

type ActivityLevel string

type ActivityType string

type AgentMetrics struct {
	HourBucket   time.Time
	AgentType    string
	Errors       string
	ID           int64
	RequestCount int64
	SuccessCount int64
	LatencySumMs int64
	LatencyP50Ms int32
	LatencyP95Ms int32
}

type Attachment struct {
	Payload       *storepb.AttachmentPayload
	MemoUID       *string
	MemoID        *int32
	Type          string
	FilePath      string
	RowStatus     string
	Filename      string
	UID           string
	OCRText       string
	ExtractedText string
	ThumbnailPath string
	Reference     string
	Blob          []byte
	CreatedTs     int64
	Size          int64
	UpdatedTs     int64
	StorageType   storepb.AttachmentStorageType
	ID            int32
	CreatorID     int32
}

type BM25Result struct {
	Memo  *Memo
	Score float32 // BM25 relevance score
}

type BM25SearchOptions struct {
	Query    string
	Limit    int
	UserID   int32
	MinScore float32
}

type DeleteAIConversation struct {
	ID int32
}

type DeleteAIMessage struct {
	ID             *int32
	ConversationID *int32
}

type DeleteAgentMetrics struct {
	BeforeTime *time.Time // Delete records older than this time
}

type DeleteAttachment struct {
	MemoID *int32
	ID     int32
}

type DeleteEpisodicMemory struct {
	ID     *int64
	UserID *int32
}

type DeleteIdentityProvider struct {
	ID int32
}

type DeleteInbox struct {
	ID int32
}

type DeleteInstanceSetting struct {
	Name string
}

type DeleteMemo struct {
	ID int32
}

type DeleteMemoRelation struct {
	MemoID        *int32
	RelatedMemoID *int32
	Type          *MemoRelationType
}

type DeleteReaction struct {
	ID int32
}

type DeleteSchedule struct {
	ID int32
}

type DeleteToolMetrics struct {
	BeforeTime *time.Time // Delete records older than this time
}

type DeleteUser struct {
	ID int32
}

type Driver interface {
	GetDB() *sql.DB
	Close() error

	IsInitialized(ctx context.Context) (bool, error)

	// Activity model related methods.
	CreateActivity(ctx context.Context, create *Activity) (*Activity, error)
	ListActivities(ctx context.Context, find *FindActivity) ([]*Activity, error)

	// Attachment model related methods.
	CreateAttachment(ctx context.Context, create *Attachment) (*Attachment, error)
	ListAttachments(ctx context.Context, find *FindAttachment) ([]*Attachment, error)
	UpdateAttachment(ctx context.Context, update *UpdateAttachment) error
	DeleteAttachment(ctx context.Context, delete *DeleteAttachment) error

	// Memo model related methods.
	CreateMemo(ctx context.Context, create *Memo) (*Memo, error)
	ListMemos(ctx context.Context, find *FindMemo) ([]*Memo, error)
	UpdateMemo(ctx context.Context, update *UpdateMemo) error
	DeleteMemo(ctx context.Context, delete *DeleteMemo) error

	// UpdateMemoEmbedding updates the embedding vector for a memo.
	UpdateMemoEmbedding(ctx context.Context, id int32, embedding []float32) error

	// SearchMemosByVector performs semantic search using vector similarity.
	// Returns memos and their similarity scores.
	SearchMemosByVector(ctx context.Context, embedding []float32, limit int) ([]*Memo, []float32, error)

	// MemoRelation model related methods.
	UpsertMemoRelation(ctx context.Context, create *MemoRelation) (*MemoRelation, error)
	ListMemoRelations(ctx context.Context, find *FindMemoRelation) ([]*MemoRelation, error)
	DeleteMemoRelation(ctx context.Context, delete *DeleteMemoRelation) error

	// InstanceSetting model related methods.
	UpsertInstanceSetting(ctx context.Context, upsert *InstanceSetting) (*InstanceSetting, error)
	ListInstanceSettings(ctx context.Context, find *FindInstanceSetting) ([]*InstanceSetting, error)
	DeleteInstanceSetting(ctx context.Context, delete *DeleteInstanceSetting) error

	// User model related methods.
	CreateUser(ctx context.Context, create *User) (*User, error)
	UpdateUser(ctx context.Context, update *UpdateUser) (*User, error)
	ListUsers(ctx context.Context, find *FindUser) ([]*User, error)
	DeleteUser(ctx context.Context, delete *DeleteUser) error

	// UserSetting model related methods.
	UpsertUserSetting(ctx context.Context, upsert *UserSetting) (*UserSetting, error)
	ListUserSettings(ctx context.Context, find *FindUserSetting) ([]*UserSetting, error)
	GetUserByPATHash(ctx context.Context, tokenHash string) (*PATQueryResult, error)

	// IdentityProvider model related methods.
	CreateIdentityProvider(ctx context.Context, create *IdentityProvider) (*IdentityProvider, error)
	ListIdentityProviders(ctx context.Context, find *FindIdentityProvider) ([]*IdentityProvider, error)
	UpdateIdentityProvider(ctx context.Context, update *UpdateIdentityProvider) (*IdentityProvider, error)
	DeleteIdentityProvider(ctx context.Context, delete *DeleteIdentityProvider) error

	// Inbox model related methods.
	CreateInbox(ctx context.Context, create *Inbox) (*Inbox, error)
	ListInboxes(ctx context.Context, find *FindInbox) ([]*Inbox, error)
	UpdateInbox(ctx context.Context, update *UpdateInbox) (*Inbox, error)
	DeleteInbox(ctx context.Context, delete *DeleteInbox) error

	// Reaction model related methods.
	UpsertReaction(ctx context.Context, create *Reaction) (*Reaction, error)
	ListReactions(ctx context.Context, find *FindReaction) ([]*Reaction, error)
	GetReaction(ctx context.Context, find *FindReaction) (*Reaction, error)
	DeleteReaction(ctx context.Context, delete *DeleteReaction) error

	// MemoEmbedding model related methods.
	UpsertMemoEmbedding(ctx context.Context, embedding *MemoEmbedding) (*MemoEmbedding, error)
	ListMemoEmbeddings(ctx context.Context, find *FindMemoEmbedding) ([]*MemoEmbedding, error)
	DeleteMemoEmbedding(ctx context.Context, memoID int32) error
	FindMemosWithoutEmbedding(ctx context.Context, find *FindMemosWithoutEmbedding) ([]*Memo, error)
	VectorSearch(ctx context.Context, opts *VectorSearchOptions) ([]*MemoWithScore, error)
	BM25Search(ctx context.Context, opts *BM25SearchOptions) ([]*BM25Result, error)

	// Schedule model related methods.
	CreateSchedule(ctx context.Context, create *Schedule) (*Schedule, error)
	ListSchedules(ctx context.Context, find *FindSchedule) ([]*Schedule, error)
	UpdateSchedule(ctx context.Context, update *UpdateSchedule) error
	DeleteSchedule(ctx context.Context, delete *DeleteSchedule) error

	// AIConversation model related methods.
	CreateAIConversation(ctx context.Context, create *AIConversation) (*AIConversation, error)
	ListAIConversations(ctx context.Context, find *FindAIConversation) ([]*AIConversation, error)
	UpdateAIConversation(ctx context.Context, update *UpdateAIConversation) (*AIConversation, error)
	DeleteAIConversation(ctx context.Context, delete *DeleteAIConversation) error

	// AIMessage model related methods.
	CreateAIMessage(ctx context.Context, create *AIMessage) (*AIMessage, error)
	ListAIMessages(ctx context.Context, find *FindAIMessage) ([]*AIMessage, error)
	DeleteAIMessage(ctx context.Context, delete *DeleteAIMessage) error

	// EpisodicMemory model related methods.
	CreateEpisodicMemory(ctx context.Context, create *EpisodicMemory) (*EpisodicMemory, error)
	ListEpisodicMemories(ctx context.Context, find *FindEpisodicMemory) ([]*EpisodicMemory, error)
	ListActiveUserIDs(ctx context.Context, cutoff time.Time) ([]int32, error)
	DeleteEpisodicMemory(ctx context.Context, delete *DeleteEpisodicMemory) error

	// UserPreferences model related methods.
	UpsertUserPreferences(ctx context.Context, upsert *UpsertUserPreferences) (*UserPreferences, error)
	GetUserPreferences(ctx context.Context, find *FindUserPreferences) (*UserPreferences, error)

	// AgentMetrics model related methods.
	UpsertAgentMetrics(ctx context.Context, upsert *UpsertAgentMetrics) (*AgentMetrics, error)
	ListAgentMetrics(ctx context.Context, find *FindAgentMetrics) ([]*AgentMetrics, error)
	DeleteAgentMetrics(ctx context.Context, delete *DeleteAgentMetrics) error

	// ToolMetrics model related methods.
	UpsertToolMetrics(ctx context.Context, upsert *UpsertToolMetrics) (*ToolMetrics, error)
	ListToolMetrics(ctx context.Context, find *FindToolMetrics) ([]*ToolMetrics, error)
	DeleteToolMetrics(ctx context.Context, delete *DeleteToolMetrics) error
}

type EpisodicMemory struct {
	Timestamp  time.Time
	AgentType  string
	UserInput  string
	Outcome    string
	Summary    string
	ID         int64
	CreatedTs  int64
	UserID     int32
	Importance float32
}

type FindAIConversation struct {
	ID        *int32
	UID       *string
	CreatorID *int32
	Pinned    *bool
	RowStatus *RowStatus
}

type FindAIMessage struct {
	ID             *int32
	UID            *string
	ConversationID *int32
}

type FindActivity struct {
	ID   *int32
	Type *ActivityType
}

type FindAgentMetrics struct {
	AgentType *string
	StartTime *time.Time
	EndTime   *time.Time
	Limit     int
}

type FindAttachment struct {
	MemoID         *int32
	ID             *int32
	UID            *string
	CreatorID      *int32
	Filename       *string
	FilenameSearch *string
	StorageType    *storepb.AttachmentStorageType
	Limit          *int
	Offset         *int
	MemoIDList     []int32
	Filters        []string
	GetBlob        bool
	HasRelatedMemo bool
}

type FindEpisodicMemory struct {
	ID        *int64
	UserID    *int32
	AgentType *string
	Query     *string // For text search in user_input and summary
	Limit     int
	Offset    int
}

type FindIdentityProvider struct {
	ID *int32
}

type FindInbox struct {
	ID          *int32
	SenderID    *int32
	ReceiverID  *int32
	Status      *InboxStatus
	MessageType *storepb.InboxMessage_Type

	// Pagination
	Limit  *int
	Offset *int
}

type FindInstanceSetting struct {
	Name string
}

type FindMemo struct {
	ID               *int32
	UID              *string
	Offset           *int
	Limit            *int
	RowStatus        *RowStatus
	CreatorID        *int32
	VisibilityList   []Visibility
	Filters          []string
	UIDList          []string
	IDList           []int32
	ExcludeContent   bool
	ExcludeComments  bool
	OrderByPinned    bool
	OrderByUpdatedTs bool
	OrderByTimeAsc   bool
}

type FindMemoEmbedding struct {
	MemoID *int32
	Model  *string
}

type FindMemoPayload struct {
	Raw                *string
	TagSearch          []string
	HasLink            bool
	HasTaskList        bool
	HasCode            bool
	HasIncompleteTasks bool
}

type FindMemoRelation struct {
	MemoID        *int32
	RelatedMemoID *int32
	Type          *MemoRelationType
	MemoFilter    *string
}

type FindMemosWithoutEmbedding struct {
	Model string // Embedding model to check
	Limit int    // Maximum number of memos to return
}

type FindPromptVersionMetrics struct {
	AgentType     *string
	PromptVersion *string
	StartTime     *time.Time
	EndTime       *time.Time
	Limit         int
}

type FindReaction struct {
	ID            *int32
	CreatorID     *int32
	ContentID     *string
	ContentIDList []string
}

type FindSchedule struct {
	ID        *int32
	UID       *string
	CreatorID *int32

	// Time range filters
	StartTs *int64
	EndTs   *int64

	// Status filter
	RowStatus *RowStatus

	// P1: Schedule query mode
	// 0 = AUTO (auto-select based on query type)
	// 1 = STANDARD (return schedules with any part in range)
	// 2 = STRICT (return only schedules completely in range)
	QueryMode *int32

	// Pagination
	Limit  *int
	Offset *int
}

type FindToolMetrics struct {
	ToolName  *string
	StartTime *time.Time
	EndTime   *time.Time
	Limit     int
}

type FindUser struct {
	ID        *int32
	RowStatus *RowStatus
	Username  *string
	Role      *Role
	Email     *string
	Nickname  *string
	Limit     *int
	Filters   []string
}

type FindUserPreferences struct {
	UserID *int32
}

type FindUserSetting struct {
	UserID *int32
	Key    storepb.UserSetting_Key
}

type IdentityProvider struct {
	Name             string
	IdentifierFilter string
	Config           string
	ID               int32
	Type             storepb.IdentityProvider_Type
}

type Inbox struct {
	Message    *storepb.InboxMessage
	Status     InboxStatus
	CreatedTs  int64
	ID         int32
	SenderID   int32
	ReceiverID int32
}

type InboxStatus string

type InstanceSetting struct {
	Name        string
	Value       string
	Description string
}

type Memo struct {
	Payload    *storepb.MemoPayload
	ParentUID  *string
	UID        string
	RowStatus  RowStatus
	Content    string
	Visibility Visibility
	CreatedTs  int64
	UpdatedTs  int64
	ID         int32
	CreatorID  int32
	Pinned     bool
}

type MemoEmbedding struct {
	Model     string
	Embedding []float32
	CreatedTs int64
	UpdatedTs int64
	ID        int32
	MemoID    int32
}

type MemoRelation struct {
	Type          MemoRelationType
	MemoID        int32
	RelatedMemoID int32
}

type MemoRelationType string

type MemoWithScore struct {
	Memo  *Memo
	Score float32 // Similarity score (0-1, higher is more similar)
}

type PATQueryResult struct {
	User   *User
	PAT    *storepb.PersonalAccessTokensUserSetting_PersonalAccessToken
	UserID int32
}

type PromptExperimentSummary struct {
	AgentType                  string
	ControlVersion             string
	TreatmentVersion           string
	Recommendation             string
	ControlRequests            int64
	TreatmentRequests          int64
	ControlSuccessRate         float64
	TreatmentSuccessRate       float64
	ControlAvgLatencyMs        int64
	TreatmentAvgLatencyMs      int64
	ImprovementRate            float64
	IsStatisticallySignificant bool
}

type PromptVersionMetrics struct {
	HourBucket       time.Time
	AgentType        string
	PromptVersion    string
	ID               int64
	RequestCount     int64
	SuccessCount     int64
	AvgLatencyMs     int64
	UserSatisfaction float32
}

type Reaction struct {
	ContentID    string
	ReactionType string
	CreatedTs    int64
	ID           int32
	CreatorID    int32
}

type RefreshTokenQueryResult struct {
	RefreshToken *storepb.RefreshTokensUserSetting_RefreshToken
	UserID       int32
}

type Role string

type RowStatus string

type Schedule struct {
	EndTs           *int64
	Payload         *string
	Reminders       *string
	RecurrenceEndTs *int64
	RecurrenceRule  *string
	Timezone        string
	Title           string
	Description     string
	Location        string
	RowStatus       RowStatus
	UID             string
	StartTs         int64
	UpdatedTs       int64
	CreatedTs       int64
	ID              int32
	CreatorID       int32
	AllDay          bool
}

type ScheduleService interface {
	CreateSchedule(ctx context.Context, create *Schedule) (*Schedule, error)
	ListSchedules(ctx context.Context, find *FindSchedule) ([]*Schedule, error)
	UpdateSchedule(ctx context.Context, update *UpdateSchedule) error
	DeleteSchedule(ctx context.Context, delete *DeleteSchedule) error
}

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

type ToolMetrics struct {
	HourBucket   time.Time
	ToolName     string
	ID           int64
	CallCount    int64
	SuccessCount int64
	LatencySumMs int64
}

type UpdateAIConversation struct {
	Title     *string
	ParrotID  *string
	Pinned    *bool
	RowStatus *RowStatus
	UpdatedTs *int64
	ID        int32
}

type UpdateAttachment struct {
	UID           *string
	UpdatedTs     *int64
	Filename      *string
	MemoID        *int32
	Reference     *string
	Payload       *storepb.AttachmentPayload
	RowStatus     *string
	ExtractedText *string
	OCRText       *string
	ThumbnailPath *string
	ID            int32
}

type UpdateIdentityProvider struct {
	Name             *string
	IdentifierFilter *string
	Config           *string
	ID               int32
}

type UpdateIdentityProviderV1 struct {
	Name             *string
	IdentifierFilter *string
	Config           *storepb.IdentityProviderConfig
	ID               int32
	Type             storepb.IdentityProvider_Type
}

type UpdateInbox struct {
	Status InboxStatus
	ID     int32
}

type UpdateMemo struct {
	UID        *string
	CreatedTs  *int64
	UpdatedTs  *int64
	RowStatus  *RowStatus
	Content    *string
	Visibility *Visibility
	Pinned     *bool
	Payload    *storepb.MemoPayload
	ID         int32
}

type UpdateSchedule struct {
	StartTs         *int64
	EndTs           *int64
	CreatedTs       *int64
	UpdatedTs       *int64
	RowStatus       *RowStatus
	Title           *string
	UID             *string
	Description     *string
	Payload         *string
	Location        *string
	AllDay          *bool
	Timezone        *string
	RecurrenceRule  *string
	RecurrenceEndTs *int64
	Reminders       *string
	ID              int32
}

type UpdateUser struct {
	UpdatedTs    *int64
	RowStatus    *RowStatus
	Username     *string
	Role         *Role
	Email        *string
	Nickname     *string
	Password     *string
	AvatarURL    *string
	PasswordHash *string
	Description  *string
	ID           int32
}

type UpsertAgentMetrics struct {
	HourBucket   time.Time
	AgentType    string
	Errors       string
	RequestCount int64
	SuccessCount int64
	LatencySumMs int64
	LatencyP50Ms int32
	LatencyP95Ms int32
}

type UpsertPromptVersionMetrics struct {
	HourBucket    time.Time
	AgentType     string
	PromptVersion string
	RequestCount  int64
	SuccessCount  int64
	AvgLatencyMs  int64
}

type UpsertToolMetrics struct {
	HourBucket   time.Time
	ToolName     string
	CallCount    int64
	SuccessCount int64
	LatencySumMs int64
}

type UpsertUserPreferences struct {
	Preferences string
	UserID      int32
}

type User struct {
	RowStatus    RowStatus
	Username     string
	Role         Role
	Email        string
	Nickname     string
	PasswordHash string
	AvatarURL    string
	Description  string
	CreatedTs    int64
	UpdatedTs    int64
	ID           int32
}

type UserPreferences struct {
	Preferences string
	CreatedTs   int64
	UpdatedTs   int64
	UserID      int32
}

type UserSetting struct {
	Value  string
	UserID int32
	Key    storepb.UserSetting_Key
}

type VectorSearchOptions struct {
	Vector []float32
	Limit  int
	UserID int32
}

type Visibility string