sqlc

type CountMemoryNodeEmbeddingsParams struct {
	TeamID  pgtype.UUID `json:"team_id"`
	BotID   pgtype.UUID `json:"bot_id"`
	ModelID pgtype.UUID `json:"model_id"`
}

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 DeleteBotMemoryNodeEmbeddingsParams struct {
	TeamID pgtype.UUID `json:"team_id"`
	BotID  pgtype.UUID `json:"bot_id"`
}

type DeleteMemoryNodeEmbeddingsParams struct {
	TeamID  pgtype.UUID `json:"team_id"`
	BotID   pgtype.UUID `json:"bot_id"`
	NodeIds []string    `json:"node_ids"`
}

type MemoryNodeEmbedding struct {
	BotID      pgtype.UUID        `json:"bot_id"`
	NodeID     string             `json:"node_id"`
	ModelID    pgtype.UUID        `json:"model_id"`
	Dimensions int32              `json:"dimensions"`
	BodyHash   string             `json:"body_hash"`
	Embedding  pgvector_go.Vector `json:"embedding"`
	UpdatedAt  pgtype.Timestamptz `json:"updated_at"`
	CreatedAt  pgtype.Timestamptz `json:"created_at"`
	TeamID     pgtype.UUID        `json:"team_id"`
}

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

type SearchMemoryNodeEmbeddingsParams struct {
	Embedding pgvector_go.Vector `json:"embedding"`
	TeamID    pgtype.UUID        `json:"team_id"`
	BotID     pgtype.UUID        `json:"bot_id"`
	ModelID   pgtype.UUID        `json:"model_id"`
	RowLimit  int32              `json:"row_limit"`
}

type SearchMemoryNodeEmbeddingsRow struct {
	NodeID string  `json:"node_id"`
	Score  float64 `json:"score"`
}

type UpsertMemoryNodeEmbeddingParams struct {
	TeamID     pgtype.UUID        `json:"team_id"`
	BotID      pgtype.UUID        `json:"bot_id"`
	NodeID     string             `json:"node_id"`
	ModelID    pgtype.UUID        `json:"model_id"`
	Dimensions int32              `json:"dimensions"`
	BodyHash   string             `json:"body_hash"`
	Embedding  pgvector_go.Vector `json:"embedding"`
}