storage

Documentation

Index

• type FetchModelInput
• type FetchNearestNeighborInput
• type Postgres
  • func NewPostgres(dbConn *sqlx.DB) *Postgres
  • func (p *Postgres) FetchModel(ctx context.Context, in FetchModelInput) (string, error)
  • func (p *Postgres) FetchNearestNeighbor(ctx context.Context, in FetchNearestNeighborInput) (string, float64, error)
  • func (p *Postgres) StoreEmbeddings(ctx context.Context, in StoreEmbeddingInput) error
• type StoreEmbeddingInput

Types

type FetchModelInput

type FetchModelInput struct {
	UserID       string
	CollectionID string
}

type FetchNearestNeighborInput

type FetchNearestNeighborInput struct {
	UserID       string
	CollectionID string
	Vector       []float32
}

type Postgres

type Postgres struct{ *sqlx.DB }

func NewPostgres

func NewPostgres(dbConn *sqlx.DB) *Postgres

func (*Postgres) FetchModel

func (p *Postgres) FetchModel(ctx context.Context, in FetchModelInput) (string, error)

func (*Postgres) FetchNearestNeighbor

func (p *Postgres) FetchNearestNeighbor(ctx context.Context, in FetchNearestNeighborInput) (string, float64, error)

func (*Postgres) StoreEmbeddings

func (p *Postgres) StoreEmbeddings(ctx context.Context, in StoreEmbeddingInput) error

type StoreEmbeddingInput

type StoreEmbeddingInput struct {
	ID           string
	UserID       string
	CollectionID string
	Model        string
	Text         string
	Tokens       int64
	Vector       []float32
	CreatedAt    time.Time
}