sqlc

type AddMessageParams struct {
	SessionID      uuid.UUID `json:"session_id"`
	Role           string    `json:"role"`
	Content        []byte    `json:"content"`
	SequenceNumber int32     `json:"sequence_number"`
}

type AddMessageWithIDParams struct {
	ID             uuid.UUID `json:"id"`
	SessionID      uuid.UUID `json:"session_id"`
	Role           string    `json:"role"`
	Content        []byte    `json:"content"`
	Status         string    `json:"status"`
	SequenceNumber int32     `json:"sequence_number"`
}

type CreateSessionParams struct {
	Title        *string `json:"title"`
	ModelName    *string `json:"model_name"`
	SystemPrompt *string `json:"system_prompt"`
}

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 Document struct {
	ID         string           `json:"id"`
	Content    string           `json:"content"`
	Embedding  *pgvector.Vector `json:"embedding"`
	SourceType *string          `json:"source_type"`
	Metadata   []byte           `json:"metadata"`
}

type GetDocumentRow struct {
	ID       string `json:"id"`
	Content  string `json:"content"`
	Metadata []byte `json:"metadata"`
}

type GetMessagesParams struct {
	SessionID    uuid.UUID `json:"session_id"`
	ResultOffset int32     `json:"result_offset"`
	ResultLimit  int32     `json:"result_limit"`
}

type GetUserMessageBeforeParams struct {
	SessionID      uuid.UUID `json:"session_id"`
	BeforeSequence int32     `json:"before_sequence"`
}

type ListDocumentsBySourceTypeParams struct {
	SourceType  string `json:"source_type"`
	ResultLimit int32  `json:"result_limit"`
}

type ListDocumentsBySourceTypeRow struct {
	ID       string `json:"id"`
	Content  string `json:"content"`
	Metadata []byte `json:"metadata"`
}

type ListSessionsParams struct {
	ResultOffset int32 `json:"result_offset"`
	ResultLimit  int32 `json:"result_limit"`
}

type ListSessionsWithMessagesParams struct {
	ResultOffset int32 `json:"result_offset"`
	ResultLimit  int32 `json:"result_limit"`
}

type Message struct {
	ID             uuid.UUID          `json:"id"`
	SessionID      uuid.UUID          `json:"session_id"`
	Role           string             `json:"role"`
	Content        []byte             `json:"content"`
	SequenceNumber int32              `json:"sequence_number"`
	CreatedAt      pgtype.Timestamptz `json:"created_at"`
	Status         string             `json:"status"`
	UpdatedAt      pgtype.Timestamptz `json:"updated_at"`
}

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

type SearchBySourceTypeParams struct {
	QueryEmbedding *pgvector.Vector `json:"query_embedding"`
	SourceType     string           `json:"source_type"`
	ResultLimit    int32            `json:"result_limit"`
}

type SearchBySourceTypeRow struct {
	ID         string  `json:"id"`
	Content    string  `json:"content"`
	Metadata   []byte  `json:"metadata"`
	Similarity float64 `json:"similarity"`
}

type SearchDocumentsAllParams struct {
	QueryEmbedding *pgvector.Vector `json:"query_embedding"`
	ResultLimit    int32            `json:"result_limit"`
}

type SearchDocumentsAllRow struct {
	ID         string  `json:"id"`
	Content    string  `json:"content"`
	Metadata   []byte  `json:"metadata"`
	Similarity float64 `json:"similarity"`
}

type SearchDocumentsParams struct {
	QueryEmbedding *pgvector.Vector `json:"query_embedding"`
	FilterMetadata []byte           `json:"filter_metadata"`
	ResultLimit    int32            `json:"result_limit"`
}

type SearchDocumentsRow struct {
	ID         string  `json:"id"`
	Content    string  `json:"content"`
	Metadata   []byte  `json:"metadata"`
	Similarity float64 `json:"similarity"`
}

type Session struct {
	ID           uuid.UUID          `json:"id"`
	Title        *string            `json:"title"`
	CreatedAt    pgtype.Timestamptz `json:"created_at"`
	UpdatedAt    pgtype.Timestamptz `json:"updated_at"`
	ModelName    *string            `json:"model_name"`
	SystemPrompt *string            `json:"system_prompt"`
	MessageCount *int32             `json:"message_count"`
}

type UpdateMessageContentParams struct {
	ID      uuid.UUID `json:"id"`
	Content []byte    `json:"content"`
}

type UpdateMessageStatusParams struct {
	ID     uuid.UUID `json:"id"`
	Status string    `json:"status"`
}

type UpdateSessionTitleParams struct {
	Title     *string   `json:"title"`
	SessionID uuid.UUID `json:"session_id"`
}

type UpdateSessionUpdatedAtParams struct {
	MessageCount *int32    `json:"message_count"`
	SessionID    uuid.UUID `json:"session_id"`
}

type UpsertDocumentParams struct {
	ID         string           `json:"id"`
	Content    string           `json:"content"`
	Embedding  *pgvector.Vector `json:"embedding"`
	SourceType *string          `json:"source_type"`
	Metadata   []byte           `json:"metadata"`
}