Weaviate
Overview
This page guides you through the process of setting up the Weaviate destination connector.
There are three parts to this:
- Processing - split up individual records in chunks so they will fit the context window and decide which fields to use as context and which are supplementary metadata.
- Embedding - convert the text into a vector representation using a pre-trained model (Currently, OpenAI's
text-embedding-ada-002and Cohere'sembed-english-light-v2.0are supported.) - Indexing - store the vectors in a vector database for similarity search
Prerequisites
To use the Weaviate destination, you'll need:
- Access to a running Weaviate instance (either self-hosted or via Weaviate Cloud Services), minimum version 1.21.2
- Either
- An account with API access for OpenAI or Cohere (depending on which embedding method you want to use)
- Pre-calculated embeddings stored in a field in your source database
You'll need the following information to configure the destination:
- Embedding service API Key - The API key for your OpenAI or Cohere account
- Weaviate cluster URL - The URL of the Weaviate cluster to load data into. Airbyte Cloud only supports connecting to your Weaviate Instance instance with TLS encryption.
- Weaviate credentials - The credentials for your Weaviate instance (either API token or username/password)
Features
| Feature | Supported?(Yes/No) | Notes |
|---|---|---|
| Full Refresh Sync | Yes | |
| Incremental - Append Sync | Yes | |
| Incremental - Append + Deduped | Yes | |
| Namespaces | No | |
| Provide vector | Yes | Either from field are calculated during the load process |
Data type mapping
All fields specified as metadata fields will be stored as properties in the object can be used for filtering. The following data types are allowed for metadata fields:
- String
- Number (integer or floating point, gets converted to a 64 bit floating point)
- Booleans (true, false)
- List of String
All other fields are serialized into their JSON representation.
Configuration
Processing
Each record will be split into text fields and metadata fields as configured in the "Processing" section. All text fields are concatenated into a single string and then split into chunks of configured length. If specified, the metadata fields are stored as-is along with the embedded text chunks. Options around configuring the chunking process use the Langchain Python library.
When specifying text fields, you can access nested fields in the record by using dot notation, e.g. user.name will access the name field in the user object. It's also possible to use wildcards to access all fields in an object, e.g. users.*.name will access all names fields in all entries of the users array.
The chunk length is measured in tokens produced by the tiktoken library. The maximum is 8191 tokens, which is the maximum length supported by the text-embedding-ada-002 model.
The stream name gets added as a metadata field _ab_stream to each document. If available, the primary key of the record is used to identify the document to avoid duplications when updated versions of records are indexed. It is added as the _ab_record_id metadata field.
Embedding
The connector can use one of the following embedding methods:
-
OpenAI - using OpenAI API , the connector will produce embeddings using the
text-embedding-ada-002model with 1536 dimensions. This integration will be constrained by the speed of the OpenAI embedding API. -
Cohere - using the Cohere API, the connector will produce embeddings using the
embed-english-light-v2.0model with 1024 dimensions. -
From field - if you have pre-calculated embeddings stored in a field in your source database, you can use the
From fieldintegration to load them into Weaviate. The field must be a JSON array of numbers, e.g.[0.1, 0.2, 0.3]. -
No embedding - if you don't want to use embeddings or have configured a vectorizer for your class, you can use the
No embeddingintegration.
For testing purposes, it's also possible to use the Fake embeddings integration. It will generate random embeddings and is suitable to test a data pipeline without incurring embedding costs.
Indexing
All streams will be indexed into separate classes derived from the stream name. If a class doesn't exist in the schema of the cluster, it will be created using the configure vectorizer configuration. In this case, dynamic schema has to be enabled on the server.
You can also create the class in Weaviate in advance if you need more control over the schema in Weaviate. In this case, the text properies _ab_stream and _ab_record_id need to be created for bookkeeping reasons. In case a sync is run in Overwrite mode, the class will be deleted and recreated.
As properties have to start will a lowercase letter in Weaviate and can't contain spaces or special characters. Field names might be updated during the loading process. The field names id, _id and _additional are reserved keywords in Weaviate, so they will be renamed to raw_id, raw__id and raw_additional respectively.
When using multi-tenancy, the tenant id can be configured in the connector configuration. If not specified, multi-tenancy will be disabled. In case you want to index into an already created class, you need to make sure the class is created with multi-tenancy enabled. In case the class doesn't exist, it will be created with multi-tenancy properly configured. If the class already exists but the tenant id is not associated with the class, the connector will automatically add the tenant id to the class. This allows you to configure multiple connections for different tenants on the same schema.
Changelog
Expand to review
| Version | Date | Pull Request | Subject |
|---|---|---|---|
| 0.2.23 | 2024-07-06 | 40943 | Update dependencies |
| 0.2.22 | 2024-06-29 | 40633 | Update dependencies |
| 0.2.21 | 2024-06-25 | 40274 | Update dependencies |
| 0.2.20 | 2024-06-22 | 40109 | Update dependencies |
| 0.2.19 | 2024-06-06 | 39212 | [autopull] Upgrade base image to v1.2.2 |
| 0.2.18 | 2024-05-15 | 38272 | Replace AirbyteLogger with logging.Logger |
| 0.2.17 | 2024-04-15 | #37333 | Update CDK & pytest version to fix security vulnerabilities. |
| 0.2.16 | 2024-03-22 | #35911 | Fix tests and move to Poetry |
| 0.2.15 | 2023-01-25 | #34529 | Fix tests |
| 0.2.14 | 2023-01-15 | #34229 | Allow configuring tenant id |
| 0.2.13 | 2023-12-11 | #33303 | Fix bug with embedding special tokens |
| 0.2.12 | 2023-12-07 | #33218 | Normalize metadata field names |
| 0.2.11 | 2023-12-01 | #32697 | Allow omitting raw text |
| 0.2.10 | 2023-11-16 | #32608 | Support deleting records for CDC sources |
| 0.2.9 | 2023-11-13 | #32357 | Improve spec schema |
| 0.2.8 | 2023-11-03 | #32134 | Improve test coverage |
| 0.2.7 | 2023-11-03 | #32134 | Upgrade weaviate client library |
| 0.2.6 | 2023-11-01 | #32038 | Retry failed object loads |
| 0.2.5 | 2023-10-24 | #31953 | Fix memory leak |
| 0.2.4 | 2023-10-23 | #31563 | Add field mapping option, improve append+dedupe sync performance and remove unnecessary retry logic |
| 0.2.3 | 2023-10-19 | #31599 | Base image migration: remove Dockerfile and use the python-connector-base image |
| 0.2.2 | 2023-10-15 | #31329 | Add OpenAI-compatible embedder option |
| 0.2.1 | 2023-10-04 | #31075 | Fix OpenAI embedder batch size and conflict field name handling |
| 0.2.0 | 2023-09-22 | #30151 | Add embedding capabilities, overwrite and dedup support and API key auth mode, make certified. 🚨 Breaking changes - check migrations guide. |
| 0.1.1 | 2022-02-08 | #22527 | Multiple bug fixes: Support String based IDs, arrays of uknown type and additionalProperties of type object and array of objects |
| 0.1.0 | 2022-12-06 | #20094 | Add Weaviate destination |