Trình bổ trợ Chroma

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

Cài đặ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ợ 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. Ngoài ra, có 2 tham số không bắt buộc:

• clientParams: Nếu không chạy máy chủ Chroma trên cùng một máy với quy trình Genkit, bạn cần chỉ định các tuỳ chọn xác thực. Nếu không, nếu không chạy cấu hình máy chủ Chroma mặc định, bạn có thể chỉ định đối tượng ChromaClientParams Chroma để chuyể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' },
  

Hoạt động sử dụng

Nhập các tham chiếu trình truy xuất và lập chỉ mục như sau:

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

Sau đó, hãy 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 });

Vui lòng xem trang Tạo tăng cường truy xuất để biết nội dung thảo luận chung về trình lập chỉ mục và trình truy xuất.