Semantic search with Cloudflare Worker AI embeddings
Introduction
This guide will walk you through the process of setting up Meilisearch with Cloudflare Worker AI embeddings to enable semantic search capabilities. By leveraging Meilisearch's AI features and Cloudflare Worker AI's embedding API, you can enhance your search experience and retrieve more relevant results.
Requirements
To follow this guide, you'll need:
- A Meilisearch Cloud project running version 1.10 or above with the Vector store activated.
- A Cloudflare account with access to Worker AI and an API key. You can sign up for a Cloudflare account at Cloudflare.
- Your Cloudflare account ID.
- No backend required.
Setting up Meilisearch
To set up an embedder in Meilisearch, you need to configure it to your settings. You can refer to the Meilisearch documentation for more details on updating the embedder settings.
Cloudflare Worker AI offers the following embedding models:
baai/bge-base-en-v1.5
: 768 dimensionsbaai/bge-large-en-v1.5
: 1024 dimensionsbaai/bge-small-en-v1.5
: 384 dimensions
Here's an example of embedder settings for Cloudflare Worker AI:
{
"cloudflare": {
"source": "rest",
"apiKey": "<API Key>",
"dimensions": 384,
"documentTemplate": "<Custom template (Optional, but recommended)>",
"url": "https://api.cloudflare.com/client/v4/accounts/<ACCOUNT_ID>/ai/run/@cf/<Model>",
"request": {
"text": ["{{text}}", "{{..}}"]
},
"response": {
"result": {
"data": ["{{embedding}}", "{{..}}"]
}
}
}
}
In this configuration:
source
: Specifies the source of the embedder, which is set to "rest" for using a REST API.apiKey
: Replace<API Key>
with your actual Cloudflare API key.dimensions
: Specifies the dimensions of the embeddings. Set to 384 forbaai/bge-small-en-v1.5
, 768 forbaai/bge-base-en-v1.5
, or 1024 forbaai/bge-large-en-v1.5
.documentTemplate
: Optionally, you can provide a custom template for generating embeddings from your documents.url
: Specifies the URL of the Cloudflare Worker AI API endpoint.request
: Defines the request structure for the Cloudflare Worker AI API, including the input parameters.response
: Defines the expected response structure from the Cloudflare Worker AI API, including the embedding data.
Be careful when setting up the url
field in your configuration. The URL contains your Cloudflare account ID (<ACCOUNT_ID>
) and the specific model you want to use (<Model>
). Make sure to replace these placeholders with your actual account ID and the desired model name (e.g., baai/bge-small-en-v1.5
).
Once you've configured the embedder settings, Meilisearch will automatically generate embeddings for your documents and store them in the vector store.
Please note that Cloudflare may have rate limiting, which is managed by Meilisearch. If you have a free account, the indexation process may take some time, but Meilisearch will handle it with a retry strategy.
It's recommended to monitor the tasks queue to ensure everything is running smoothly. You can access the tasks queue using the Cloud UI or the Meilisearch API.
Testing semantic search
With the embedder set up, you can now perform semantic searches using Meilisearch. When you send a search query, Meilisearch will generate an embedding for the query using the configured embedder and then use it to find the most semantically similar documents in the vector store. To perform a semantic search, you simply need to make a normal search request but include the hybrid parameter:
{
"q": "<Query made by the user>",
"hybrid": {
"semanticRatio": 1,
"embedder": "cloudflare"
}
}
In this request:
q
: Represents the user's search query.hybrid
: Specifies the configuration for the hybrid search.semanticRatio
: Allows you to control the balance between semantic search and traditional search. A value of 1 indicates pure semantic search, while a value of 0 represents full-text search. You can adjust this parameter to achieve a hybrid search experience.embedder
: The name of the embedder used for generating embeddings. Make sure to use the same name as specified in the embedder configuration, which in this case is "cf-bge-small-en-v1.5".
You can use the Meilisearch API or client libraries to perform searches and retrieve the relevant documents based on semantic similarity.
Conclusion
By following this guide, you should now have Meilisearch set up with Cloudflare Worker AI embedding, enabling you to leverage semantic search capabilities in your application. Meilisearch's auto-batching and efficient handling of embeddings make it a powerful choice for integrating semantic search into your project.
To explore further configuration options for embedders, consult the detailed documentation about the embedder setting possibilities.