Chroma 插件

Chroma 插件提供使用 客户端/服务器模式下的 Chroma 矢量数据库。

安装

npm i --save genkitx-chromadb

配置

如需使用此插件，请在调用 configureGenkit() 时指定它：

import { chroma } from 'genkitx-chromadb';

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

您必须指定色度集合和要使用的嵌入模型。在 此外，还有两个可选参数：

• clientParams：如果您并非在同一台计算机上运行 Chroma 服务器 Genkit 流程需要指定身份验证选项，否则 运行默认的 Chroma 服务器配置时，您可以指定 ChromaClientParams 对象 传递给 Chroma 客户端：

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

• embedderOptions：使用此参数将选项传递给嵌入器：

  embedderOptions: { taskType: 'RETRIEVAL_DOCUMENT' },
  

用法

导入检索器和索引器引用，如下所示：

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

然后，将引用传递给 retrieve() 和 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 });

请参阅检索增强生成页面，了解常规 索引器和检索器。