The Chroma component is a data component that allows users to build and search vector datasets.
It can carry out the following tasks:
Alpha
The component definition and tasks are defined in the definition.yaml and tasks.yaml files respectively.
In order to communicate with Chroma, the following connection details need to be
provided. You may specify them directly in a pipeline recipe as key-value pairs
within the component's setup
block, or you can create a Connection from
the Integration Settings
page and reference the whole setup
as setup: ${connection.<my-connection-id>}
.
Field Field ID Type Note API Key (required) api-key
string Fill in your Chroma API key. Chroma URL Endpoint (required) url
string Fill in your Chroma hosted public URL endpoint with port, e.g http://1.2.3:8000 .
Upsert multiple vector items into a collection, existing items will be updated
Input Field ID Type Description Task ID (required) task
string TASK_BATCH_UPSERT
Collection Name (required) collection-name
string The name of the collection to upsert the item into. Array ID (required) array-id
array[string] The array of id. Array Metadata (required)array-metadata
array[object] The array of vector metadata. Array Vector (required) array-vector
array[array] The array of vector values. Array Document array-document
array[string] The array of document string values. Array URI array-uri
array[string] The array of uri.
Output Field ID Type Description Status status
string Batch add status.
Upsert a vector item into a collection, existing item will be updated
Input Field ID Type Description Task ID (required) task
string TASK_UPSERT
Collection Name (required) collection-name
string The name of the collection to upsert the item into. ID (required) id
string The ID of the item. Metadata (required) metadata
object The vector metadata. Vector (required) vector
array[number] An array of dimensions for the vector value. Document document
string The document string value. URI uri
string The uri of the item.
Output Field ID Type Description Status status
string Add status.
Perform a vector search on a collection
Input Field ID Type Description Task ID (required) task
string TASK_QUERY
Collection Name (required) collection-name
string The name of the collection to perform vector similarity search on. Vector (required) vector
array[number] An array of dimensions for the vector query. N Results (required) n-results
integer The N amount of items to return from the vector search operation. Filter filter
object The metadata filter to be applied to the data with Chroma where filter, please refer to using-where-filters . Filter Document filter-document
string The document content filter to be applied to the data, please refer to filtering-by-document-contents . Fields fields
array[string] The fields to be returned from the vector search operation.
Output Field ID Type Description Result result
object Result of the vector search operation. Status status
string Vector search status.
Output Objects in Query Result Field Field ID Type Note IDs ids
array The ids returned from the vector search operation. Items items
array The items returned from the vector search operation. Metadata metadata
array The metadata returned from the vector search operation. Vectors vectors
array The vectors returned from the vector search operation.
Delete vector items from a collection
Input Field ID Type Description Task ID (required) task
string TASK_DELETE
Collection Name (required) collection-name
string The name of the collection to delete the object from. ID id
string The ID of the item. Filter filter
object The metadata filter to be applied to the data with Chroma where filter, please refer to using-where-filters . Filter Document filter-document
string The document content filter to be applied to the data, please refer to filtering-by-document-contents .
Output Field ID Type Description Status status
string Delete status.
Create a collection
Input Field ID Type Description Task ID (required) task
string TASK_CREATE_COLLECTION
Collection Name (required) collection-name
string The name of the collection to create. Config metadata
object The metadata of the collection. Please refer to creating-inspecting-and-deleting-collection . Configuration configuration
object The configuration of the collection. Please refer to creating-inspecting-and-deleting-collection . Get or Create get-or-create
boolean If true, the collection will be created if it does not exist.
Output Field ID Type Description Status status
string Create collection status.
Delete a collection
Input Field ID Type Description Task ID (required) task
string TASK_DELETE_COLLECTION
Collection Name (required) collection-name
string The name of the collection to delete.
Output Field ID Type Description Status status
string Delete collection status.