qdrant

Documentation

Index

• Constants
• type Match
• type NamespaceConfig
• type Point
• type Qdrant
  • func New(endpoint string) (*Qdrant, error)
  • func (q *Qdrant) CreateNamespace(uuid string) error
  • func (q *Qdrant) NamespaceExists(uuid string) (bool, error)
  • func (q *Qdrant) Retrieve(questionEmbedding []float32, topK int, uuid string) ([]vectordb.QueryMatch, error)
  • func (q *Qdrant) UpsertEmbeddings(embeddings [][]float32, chunks []chunk.Chunk, uuid string) error
• type SearchResult

Constants

const (
	VECTOR_SIZE     = 1536 // ada002
	VECTOR_DISTANCE = "Cosine"
	BATCH_SIZE      = 500
)

Types

type Match

type Match struct {
	ID      int               `json:"id"`
	Score   float32           `json:"score"`
	Payload map[string]string `json:"payload"`
	Version int               `json:"version"`
}

type NamespaceConfig

type NamespaceConfig struct {
	Vectors struct {
		Size     int    `json:"size"`
		Distance string `json:"distance"`
	} `json:"vectors"`
}

type Point

type Point struct {
	ID      int               `json:"id"`
	Vector  []float32         `json:"vector"`
	Payload map[string]string `json:"payload,omitempty"`
}

type Qdrant

type Qdrant struct {
	Endpoint string
	// contains filtered or unexported fields
}

func New

func New(endpoint string) (*Qdrant, error)

func (*Qdrant) CreateNamespace

func (q *Qdrant) CreateNamespace(uuid string) error

func (*Qdrant) NamespaceExists

func (q *Qdrant) NamespaceExists(uuid string) (bool, error)

func (*Qdrant) Retrieve

func (q *Qdrant) Retrieve(questionEmbedding []float32, topK int, uuid string) ([]vectordb.QueryMatch, error)

func (*Qdrant) UpsertEmbeddings

func (q *Qdrant) UpsertEmbeddings(embeddings [][]float32, chunks []chunk.Chunk, uuid string) error

type SearchResult

type SearchResult struct {
	Result []Match `json:"result"`
	Status string  `json:"status"`
	Time   float64 `json:"time"`
}