Pinecone vector store integration.

Setup: Install @langchain/pinecone and @pinecone-database/pinecone to pass a client in.

npm install @langchain/pinecone @pinecone-database/pinecone
Instantiate
import { PineconeStore } from '@langchain/pinecone';
// Or other embeddings
import { OpenAIEmbeddings } from '@langchain/openai';

import { Pinecone as PineconeClient } from "@pinecone-database/pinecone";

const pinecone = new PineconeClient();

// Will automatically read the PINECONE_API_KEY and PINECONE_ENVIRONMENT env vars
const pineconeIndex = pinecone.Index(process.env.PINECONE_INDEX!);

const embeddings = new OpenAIEmbeddings({
model: "text-embedding-3-small",
});

const vectorStore = await PineconeStore.fromExistingIndex(embeddings, {
pineconeIndex,
// Maximum number of batch requests to allow at once. Each batch is 1000 vectors.
maxConcurrency: 5,
// You can pass a namespace here too
// namespace: "foo",
});

Add documents
import type { Document } from '@langchain/core/documents';

const document1 = { pageContent: "foo", metadata: { baz: "bar" } };
const document2 = { pageContent: "thud", metadata: { bar: "baz" } };
const document3 = { pageContent: "i will be deleted :(", metadata: {} };

const documents: Document[] = [document1, document2, document3];
const ids = ["1", "2", "3"];
await vectorStore.addDocuments(documents, { ids });

Delete documents
await vectorStore.delete({ ids: ["3"] });

Similarity search
const results = await vectorStore.similaritySearch("thud", 1);
for (const doc of results) {
console.log(`* ${doc.pageContent} [${JSON.stringify(doc.metadata, null)}]`);
}
// Output: * thud [{"baz":"bar"}]

Similarity search with filter
const resultsWithFilter = await vectorStore.similaritySearch("thud", 1, { baz: "bar" });

for (const doc of resultsWithFilter) {
console.log(`* ${doc.pageContent} [${JSON.stringify(doc.metadata, null)}]`);
}
// Output: * foo [{"baz":"bar"}]

Similarity search with score
const resultsWithScore = await vectorStore.similaritySearchWithScore("qux", 1);
for (const [doc, score] of resultsWithScore) {
console.log(`* [SIM=${score.toFixed(6)}] ${doc.pageContent} [${JSON.stringify(doc.metadata, null)}]`);
}
// Output: * [SIM=0.000000] qux [{"bar":"baz","baz":"bar"}]

As a retriever
const retriever = vectorStore.asRetriever({
searchType: "mmr", // Leave blank for standard similarity search
k: 1,
});
const resultAsRetriever = await retriever.invoke("thud");
console.log(resultAsRetriever);

// Output: [Document({ metadata: { "baz":"bar" }, pageContent: "thud" })]

Hierarchy

  • VectorStore
    • PineconeStore

Constructors

Properties

FilterType: PineconeMetadata
caller: AsyncCaller
embeddings: EmbeddingsInterface
filter?: PineconeMetadata
namespace?: string
pineconeIndex: Index<RecordMetadata>
textKey: string

Methods

  • Method that adds documents to the Pinecone database.

    Parameters

    • documents: Document<Record<string, any>>[]

      Array of documents to add to the Pinecone database.

    • Optionaloptions: string[] | {
          ids?: string[];
          namespace?: string;
      }

      Optional ids for the documents.

    Returns Promise<string[]>

    Promise that resolves with the ids of the added documents.

  • Method that adds vectors to the Pinecone database.

    Parameters

    • vectors: number[][]

      Array of vectors to add to the Pinecone database.

    • documents: Document<Record<string, any>>[]

      Array of documents associated with the vectors.

    • Optionaloptions: string[] | {
          ids?: string[];
          namespace?: string;
      }

      Optional ids for the vectors.

    Returns Promise<string[]>

    Promise that resolves with the ids of the added vectors.

  • Parameters

    • OptionalkOrFields: number | Partial<VectorStoreRetrieverInput<PineconeStore>>
    • Optionalfilter: PineconeMetadata
    • Optionalcallbacks: Callbacks
    • Optionaltags: string[]
    • Optionalmetadata: Record<string, unknown>
    • Optionalverbose: boolean

    Returns VectorStoreRetriever<PineconeStore>

  • Return documents selected using the maximal marginal relevance. Maximal marginal relevance optimizes for similarity to the query AND diversity among selected documents.

    Parameters

    • query: string

      Text to look up documents similar to.

    • options: MaxMarginalRelevanceSearchOptions<PineconeMetadata>

    Returns Promise<DocumentInterface<Record<string, any>>[]>

    • List of documents selected by maximal marginal relevance.
  • Parameters

    • query: string
    • Optionalk: number
    • Optionalfilter: PineconeMetadata
    • Optional_callbacks: Callbacks

    Returns Promise<DocumentInterface<Record<string, any>>[]>

  • Method that performs a similarity search in the Pinecone database and returns the results along with their scores.

    Parameters

    • query: number[]

      Query vector for the similarity search.

    • k: number

      Number of top results to return.

    • Optionalfilter: PineconeMetadata

      Optional filter to apply to the search.

    Returns Promise<[Document<Record<string, any>>, number][]>

    Promise that resolves with an array of documents and their scores.

  • Parameters

    • query: string
    • Optionalk: number
    • Optionalfilter: PineconeMetadata
    • Optional_callbacks: Callbacks

    Returns Promise<[DocumentInterface<Record<string, any>>, number][]>

  • Returns Serialized

  • Static method that creates a new instance of the PineconeStore class from documents.

    Parameters

    • docs: Document<Record<string, any>>[]

      Array of documents to add to the Pinecone database.

    • embeddings: EmbeddingsInterface

      Embeddings to use for the documents.

    • dbConfig: PineconeStoreParams

      Configuration for the Pinecone database.

    Returns Promise<PineconeStore>

    Promise that resolves with a new instance of the PineconeStore class.

  • Static method that creates a new instance of the PineconeStore class from an existing index.

    Parameters

    • embeddings: EmbeddingsInterface

      Embeddings to use for the documents.

    • dbConfig: PineconeStoreParams

      Configuration for the Pinecone database.

    Returns Promise<PineconeStore>

    Promise that resolves with a new instance of the PineconeStore class.

  • Static method that creates a new instance of the PineconeStore class from texts.

    Parameters

    • texts: string[]

      Array of texts to add to the Pinecone database.

    • metadatas: object | object[]

      Metadata associated with the texts.

    • embeddings: EmbeddingsInterface

      Embeddings to use for the texts.

    • dbConfig: PineconeStoreParams | {
          namespace?: string;
          pineconeIndex: Index<RecordMetadata>;
          textKey?: string;
      }

      Configuration for the Pinecone database.

    Returns Promise<PineconeStore>

    Promise that resolves with a new instance of the PineconeStore class.