Create indexes
Indexes are the “atom” of Vectorize. Vectors are inserted into an index and enable you to query the index for similar vectors for a given input vector.
Creating an index requires three inputs:
- A name, for example
prod-search-index
orrecommendations-idx-dev
. - The (fixed) dimension size of each vector, for example 384 or 1536.
- The (fixed) distance metric to use for calculating vector similarity.
The configuration of an index cannot be changed after creation.
To create an index with wrangler
:
To create an index that can accept vector embeddings from Worker’s AI’s @cf/baai/bge-base-en-v1.5
embedding model, which outputs vectors with 768 dimensions, use the following command:
Vectorize also supports creating indexes via REST API.
For example, to create an index directly from a Python script:
This script should print the response with a status code 201
, along with a JSON response body indicating the creation of an index with the provided configuration.
Dimensions are determined from the output size of the machine learning (ML) model used to generate them, and are a function of how the model encodes and describes features into a vector embedding.
The number of output dimensions can determine vector search accuracy, search performance (latency), and the overall size of the index. Smaller output dimensions can be faster to search across, which can be useful for user-facing applications. Larger output dimensions can provide more accurate search, especially over larger datasets and/or datasets with substantially similar inputs.
The number of dimensions an index is created for cannot change. Indexes expect to receive dense vectors with the same number of dimensions.
The following table highlights some example embeddings models and their output dimensions:
Model / Embeddings API | Output dimensions | Use-case |
---|---|---|
Workers AI - @cf/baai/bge-base-en-v1.5 | 768 | Text |
OpenAI - ada-002 | 1536 | Text |
Cohere - embed-multilingual-v2.0 | 768 | Text |
Google Cloud - multimodalembedding | 1408 | Multi-modal (text, images) |
Distance metrics are functions that determine how close vectors are from each other. Vectorize indexes support the following distance metrics:
Metric | Details |
---|---|
cosine | Distance is measured between -1 (most dissimilar) to 1 (identical). 0 denotes an orthogonal vector. |
euclidean | Euclidean (L2) distance. 0 denotes identical vectors. The larger the positive number, the further the vectors are apart. |
dot-product | Negative dot product. Larger negative values or smaller positive values denote more similar vectors. A score of -1000 is more similar than -500 , and a score of 15 more similar than 50 . |
Determining the similarity between vectors can be subjective based on how the machine-learning model that represents features in the resulting vector embeddings. For example, a score of 0.8511
when using a cosine
metric means that two vectors are close in distance, but whether data they represent is similar is a function of how well the model is able to represent the original content.
Distance metrics cannot be changed after index creation, and that each metric has a different scoring function.