Trình bổ trợ sắc độ

Trình bổ trợ Chroma cung cấp các phương thức triển khai cho trình lập chỉ mục và chức năng truy xuất dữ liệu sử dụng Cơ sở dữ liệu vectơ Chroma ở chế độ máy khách/máy chủ.

Lắp đặt

npm i --save genkitx-chromadb

Cấu hình

Để sử dụng trình bổ trợ này, hãy chỉ định trình bổ trợ này khi bạn gọi configureGenkit():

import { chroma } from 'genkitx-chromadb';

export default configureGenkit({
  plugins: [
    chroma([
      {
        collectionName: 'bob_collection',
        embedder: textEmbeddingGecko,
      },
    ]),
  ],
  // ...
});

Bạn phải chỉ định một bộ sưu tập Chroma và mô hình nhúng mà bạn muốn sử dụng. Ngang bằng Ngoài ra, còn có hai tham số không bắt buộc:

• clientParams: Nếu bạn hiện không chạy máy chủ Chroma trên cùng một máy như quy trình Genkit của bạn, bạn cần chỉ định các tuỳ chọn xác thực, hoặc bạn không cần chạy một cấu hình máy chủ Chroma mặc định, bạn có thể chỉ định một Chroma Đối tượng ChromaClientParams để truyền đến ứng dụng Chroma:

  clientParams: {
    path: "http://192.168.10.42:8000",
  }
  

• embedderOptions: Sử dụng tham số này để truyền các tuỳ chọn đến trình nhúng:

  embedderOptions: { taskType: 'RETRIEVAL_DOCUMENT' },
  

Cách sử dụng

Nhập thông tin tham chiếu đến trình truy xuất và chỉ mục như sau:

import { chromaRetrieverRef } from 'genkitx-chromadb';
import { chromaIndexerRef } from 'genkitx-chromadb';

Sau đó, truyền tham chiếu đến retrieve() và index():

// To use the index you configured when you loaded the plugin:
let docs = await retrieve({ retriever: chromaRetrieverRef, query });

// To specify an index:
export const bobFactsRetriever = chromaRetrieverRef({
  collectionName: 'bob-facts',
});
docs = await retrieve({ retriever: bobFactsRetriever, query });

// To use the index you configured when you loaded the plugin:
await index({ indexer: chromaIndexerRef, documents });

// To specify an index:
export const bobFactsIndexer = chromaIndexerRef({
  collectionName: 'bob-facts',
});
await index({ indexer: bobFactsIndexer, documents });

Hãy xem trang Thế hệ tăng cường truy xuất để biết thông tin chung thảo luận về trình lập chỉ mục và chó tha mồi.