Plugin Chroma

Plugin Chroma menyediakan implementasi pengindeks dan retriever yang menggunakan Database vektor kroma dalam mode klien/server.

Penginstalan

npm i --save genkitx-chromadb

Konfigurasi

Untuk menggunakan plugin ini, tentukan 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 penyematan yang ingin digunakan. Di beberapa Selain itu, ada dua parameter opsional:

• clientParams: Jika Anda tidak menjalankan server Chroma di komputer yang sama sebagai alur Genkit, Anda harus menentukan opsi auth, atau Anda tidak harus menjalankan konfigurasi server Chroma {i> default<i}, Anda dapat menentukan objek ChromaClientParams untuk diteruskan ke klien Chroma:

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

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

  embedderOptions: { taskType: 'RETRIEVAL_DOCUMENT' },
  

Penggunaan

Impor referensi retriever dan pengindeks seperti berikut:

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 ditingkatkan untuk diskusi tentang pengindeks dan pengambil.