Pinecone

Pineconeopen in new window 是一个功能广泛的向量数据库。

本文档展示了如何使用与 Pinecone 向量数据库相关的功能。

要使用 Pinecone，您必须拥有一个 API 密钥。以下是安装指南open in new window。

!pip install pinecone-client

import os
import getpass

PINECONE_API_KEY = getpass.getpass('Pinecone API Key:')

PINECONE_ENV = getpass.getpass('Pinecone Environment:')

We want to use OpenAIEmbeddings so we have to get the OpenAI API Key.

os.environ['OPENAI_API_KEY'] = getpass.getpass('OpenAI API Key:')

from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from langchain.vectorstores import Pinecone
from langchain.document_loaders import TextLoader

from langchain.document_loaders import TextLoader
loader = TextLoader('../../../state_of_the_union.txt')
documents = loader.load()
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
docs = text_splitter.split_documents(documents)

embeddings = OpenAIEmbeddings()

import pinecone 

# 初始化 Pinecone
pinecone.init(
    api_key=PINECONE_API_KEY,  # 在 app.pinecone.io 上找到
    environment=PINECONE_ENV  # 在控制台中与 API 密钥旁边找到
)

index_name = "langchain-demo"

docsearch = Pinecone.from_documents(docs, embeddings, index_name=index_name)

# 如果您已经有一个索引，可以像这样加载它
# docsearch = Pinecone.from_existing_index(index_name, embeddings)

query = "What did the president say about Ketanji Brown Jackson"
docs = docsearch.similarity_search(query)

print(docs[0].page_content)

# Pinecone

Pinecone