Trình bổ trợ Pinecone

Trình bổ trợ Pinecone cung cấp các phương thức triển khai trình lập chỉ mục và trình truy xuất sử dụng cơ sở dữ liệu vectơ trên đám mây Pinecone.

Lắp đặt

npm i --save genkitx-pinecone

Cấu hình

Để sử dụng trình bổ trợ này, hãy chỉ định trình bổ trợ đó khi bạn khởi chạy Genkit:

import { genkit } from 'genkit';
import { pinecone } from 'genkitx-pinecone';

const ai = genkit({
  plugins: [
    pinecone([
      {
        indexId: 'bob-facts',
        embedder: textEmbedding004,
      },
    ]),
  ],
});

Bạn phải chỉ định mã chỉ mục Pinecone và mô hình nhúng mà bạn muốn sử dụng.

Ngoài ra, bạn phải định cấu hình Genkit bằng khoá API Pinecone. Có hai cách để thực hiện việc này:

  • Đặt biến môi trường PINECONE_API_KEY.
  • Chỉ định giá trị này trong tham số tuỳ chọn clientParams:

    clientParams: {
      apiKey: ...,
    }
    

    Giá trị của tham số này là một đối tượng PineconeConfiguration, được truyền đến ứng dụng Pinecone; bạn có thể sử dụng đối tượng này để truyền bất kỳ tham số nào mà ứng dụng hỗ trợ.

Cách sử dụng

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

import { pineconeRetrieverRef } from 'genkitx-pinecone';
import { pineconeIndexerRef } from 'genkitx-pinecone';

Sau đó, hãy sử dụng các tệp tham chiếu này với ai.retrieve()ai.index():

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

// To specify an index:
export const bobFactsRetriever = pineconeRetrieverRef({
  indexId: 'bob-facts',
});
docs = await ai.retrieve({ retriever: bobFactsRetriever, query });
// To use the index you configured when you loaded the plugin:
await ai.index({ indexer: pineconeIndexerRef, documents });

// To specify an index:
export const bobFactsIndexer = pineconeIndexerRef({
  indexId: 'bob-facts',
});
await ai.index({ indexer: bobFactsIndexer, documents });

Hãy xem trang Tạo bằng phương thức truy xuất tăng cường để thảo luận chung về trình lập chỉ mục và trình truy xuất.