Plugin Chroma

Plugin Chroma menyediakan implementasi pengindeks dan pengambil yang menggunakan database vektor Chroma dalam mode klien/server.

Penginstalan

npm i --save genkitx-chromadb

Konfigurasi

Untuk menggunakan plugin ini, tentukan plugin ini saat Anda memanggil configureGenkit():

import { chroma } from 'genkitx-chromadb';

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

Anda harus menentukan koleksi Chroma dan model embedding yang ingin digunakan. Selain itu, ada dua parameter opsional:

• clientParams: Jika Anda tidak menjalankan server Chroma di komputer yang sama dengan alur Genkit, Anda harus menentukan opsi autentikasi, atau tidak menjalankan konfigurasi server Chroma default, Anda dapat menentukan objek ChromaClientParams Chroma untuk diteruskan ke klien Chroma:

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

• embedderOptions: Gunakan parameter ini untuk meneruskan opsi ke penyemat:

  embedderOptions: { taskType: 'RETRIEVAL_DOCUMENT' },
  

Penggunaan

Impor referensi retriever dan pengindeks seperti:

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

Lalu, teruskan referensi ke retrieve() dan 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 });

Lihat halaman Pembuatan yang diotomatiskan dengan pengambilan untuk diskusi umum tentang pengindeks dan pengambil.