Skip to content

UBOS-tech/node-red-contrib-chromadb

Repository files navigation

node-red-contrib-chromadb

Chroma is the open-source embedding database. Chroma makes it easy to build LLM apps by making knowledge, facts, and skills pluggable for LLMs.

Ubos - End-to-End Software Development Platform

UBOSCommunityYoutubeDiscordGitHub

flow npm


Install

cd ~/.node-red
npm install node-red-contrib-chromadb

Restart your Node-RED instance.


Methods Client

createCollection (more details)

createCollection(params): Promise<Collection>

Creates a new collection with the specified properties.

Throws

If there is an issue creating the collection.

Example

msg.collection = "my_collection";
msg.payload = {
    metadata: {
        description: "My first collection"
    }
};

return msg;

Parameters

Name Type Description
params Object The parameters for creating a new collection.
params.embeddingFunction? IEmbeddingFunction Optional custom embedding function for the collection.
params.metadata? CollectionMetadata Optional metadata associated with the collection.
params.name string The name of the collection.

Returns

Promise<Collection>

A promise that resolves to the created collection.


deleteCollection (more details)

deleteCollection(params): Promise<void>

Deletes a collection with the specified name.

Throws

If there is an issue deleting the collection.

Example

msg.collection = "my_collection";
msg.payload = {};

return msg;

Parameters

Name Type Description
params Object The parameters for deleting a collection.
params.name string The name of the collection.

Returns

Promise<void>

A promise that resolves when the collection is deleted.


getCollection (more details)

getCollection(params): Promise<Collection>

Gets a collection with the specified name.

Throws

If there is an issue getting the collection.

Example

msg.collection = "my_collection";
msg.payload = {};

return msg;

Parameters

Name Type Description
params Object The parameters for getting a collection.
params.embeddingFunction? IEmbeddingFunction Optional custom embedding function for the collection.
params.name string The name of the collection.

Returns

Promise<Collection>

A promise that resolves to the collection.


getOrCreateCollection (more details)

getOrCreateCollection(params): Promise<Collection>

Gets or creates a collection with the specified properties.

Throws

If there is an issue getting or creating the collection.

Example

msg.collection = "my_collection";
msg.payload = {
    metadata: {
        description: "My first collection"
    }
};

return msg;

Parameters

Name Type Description
params Object The parameters for creating a new collection.
params.embeddingFunction? IEmbeddingFunction Optional custom embedding function for the collection.
params.metadata? CollectionMetadata Optional metadata associated with the collection.
params.name string The name of the collection.

Returns

Promise<Collection>

A promise that resolves to the got or created collection.


heartbeat (more details)

heartbeat(): Promise<number>

Returns a heartbeat from the Chroma API.

Returns

Promise<number>

A promise that resolves to the heartbeat from the Chroma API.


listCollections (more details)

listCollections(): Promise<CollectionType[]>

Lists all collections.

Throws

If there is an issue listing the collections.

Returns

Promise<CollectionType[]>

A promise that resolves to a list of collection names.


reset (more details)

reset(): Promise<Reset200Response>

Resets the state of the object by making an API call to the reset endpoint.

Throws

If there is an issue resetting the state.

Returns

Promise<Reset200Response>

A promise that resolves when the reset operation is complete.


version (more details)

version(): Promise<string>

Returns the version of the Chroma API.

Returns

Promise<string>

A promise that resolves to the version of the Chroma API.

Methods Collection

add(params): Promise<AddResponse>

Add items to the collection

Example

msg.collection = "my_collection";
msg.payload = {
    ids: ["id1", "id2"],
    embeddings: [
        [1, 2, 3],
        [4, 5, 6],
    ],
    metadatas: [{ key: "value" }, { key: "value" }],
    documents: ["document1", "document2"],
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.documents? string | Documents Optional documents of the items to add.
params.embeddings? Embedding | Embeddings Optional embeddings of the items to add.
params.ids string | IDs IDs of the items to add.
params.metadatas? Metadata | Metadatas Optional metadata of the items to add.

Returns

Promise<AddResponse>

  • The response from the API. True if successful.

count (more details)

count(): Promise<number>

Count the number of items in the collection

Example

msg.collection = "my_collection";
msg.payload = {};

return msg;

Returns

Promise<number>

  • The response from the API.

delete (more details)

delete(params?): Promise<string[]>

Deletes items from the collection.

Throws

If there is an issue deleting items from the collection.

Example

msg.collection = "my_collection";
msg.payload = {
    ids: "some_id",
    where: { key: "value" },
    whereDocument: { "$contains": "search_string" }
};

return msg;

Parameters

Name Type Description
params Object The parameters for deleting items from the collection.
params.ids? string | IDs Optional ID or array of IDs of items to delete.
params.where? Where Optional query condition to filter items to delete based on metadata values.
params.whereDocument? WhereDocument Optional query condition to filter items to delete based on document content.

Returns

Promise<string[]>

A promise that resolves to the IDs of the deleted items.


get(params?): Promise<GetResponse>

Get items from the collection

Example

msg.collection = "my_collection";
msg.payload = {
    ids: ["id1", "id2"],
    where: { key: "value" },
    limit: 10,
    offset: 0,
    include: ["embeddings", "metadatas", "documents"],
    whereDocument: { $contains: "value" },
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.ids? string | IDs Optional IDs of the items to get.
params.include? IncludeEnum[] Optional list of items to include in the response.
params.limit? number Optional limit on the number of items to get.
params.offset? number Optional offset on the items to get.
params.where? Where Optional where clause to filter items by.
params.whereDocument? WhereDocument Optional where clause to filter items by.

Returns

Promise<GetResponse>

  • The response from the server.

modify (more details)

modify(params?): Promise<void>

Modify the collection name or metadata

Example

msg.collection = "my_collection";
msg.payload = {
    name: "new collection name",
    metadata: { key: "value" },
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.metadata? CollectionMetadata Optional new metadata for the collection.
params.name? string Optional new name for the collection.

Returns

Promise<void>

  • The response from the API.

peek(params?): Promise<GetResponse>

Peek inside the collection

Throws

If there is an issue executing the query.

Example

msg.collection = "my_collection";
msg.payload = {
    limit: 10,
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.limit? number Optional number of results to return (default is 10).

Returns

Promise<GetResponse>

A promise that resolves to the query results.


query (more details)

query(params): Promise<QueryResponse>

Performs a query on the collection using the specified parameters.

Throws

If there is an issue executing the query.

Example

// Query the collection using embeddings
msg.collection = "my_collection";
msg.payload = {
    queryEmbeddings: [0.1, 0.2],
    nResults: 10,
    where: { key: "value" },
    include: ["metadata", "document"]
};

return msg;

Example

// Query the collection using query text
msg.collection = "my_collection";
msg.payload = {
    queryTexts: "some text",
    nResults: 10,
    where: { key: "value" },
    include: ["metadata", "document"]
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.include? IncludeEnum[] Optional array of fields to include in the result, such as "metadata" and "document".
params.nResults? number Optional number of results to return (default is 10).
params.queryEmbeddings? Embedding | Embeddings Optional query embeddings to use for the search.
params.queryTexts? string | string[] Optional query text(s) to search for in the collection.
params.where? Where Optional query condition to filter results based on metadata values.
params.whereDocument? WhereDocument Optional query condition to filter results based on document content.

Returns

Promise<QueryResponse>

A promise that resolves to the query results.


update (more details)

update(params): Promise<boolean>

Update the embeddings, documents, and/or metadatas of existing items

Example

msg.collection = "my_collection";
msg.payload = {
    ids: ["id1", "id2"],
    embeddings: [
        [1, 2, 3],
        [4, 5, 6],
    ],
    metadatas: [{ key: "value" }, { key: "value" }],
    documents: ["new document 1", "new document 2"],
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.documents? string | Documents Optional documents to update.
params.embeddings? Embedding | Embeddings Optional embeddings to update.
params.ids string | IDs The IDs of the items to update.
params.metadatas? Metadata | Metadatas Optional metadatas to update.

Returns

Promise<boolean>

  • The API Response. True if successful. Else, error.

upsert (more details)

upsert(params): Promise<boolean>

Upsert items to the collection

Example

msg.collection = "my_collection";
msg.payload = {
    ids: ["id1", "id2"],
    embeddings: [
        [1, 2, 3],
        [4, 5, 6],
    ],
    metadatas: [{ key: "value" }, { key: "value" }],
    documents: ["document1", "document2"],
};

return msg;

Parameters

Name Type Description
params Object The parameters for the query.
params.documents? string | Documents Optional documents of the items to add.
params.embeddings? Embedding | Embeddings Optional embeddings of the items to add.
params.ids string | IDs IDs of the items to add.
params.metadatas? Metadata | Metadatas Optional metadata of the items to add.

Returns

Promise<boolean>

  • The response from the API. True if successful.

Bugs reports and feature requests

Please report any issues or feature requests at GitHub.

License

MIT License