diff --git a/docs/extras/integrations/vectorstores/pinecone.ipynb b/docs/extras/integrations/vectorstores/pinecone.ipynb index 837dcf9dbf1..8d5cb9d4234 100644 --- a/docs/extras/integrations/vectorstores/pinecone.ipynb +++ b/docs/extras/integrations/vectorstores/pinecone.ipynb @@ -25,7 +25,7 @@ }, "outputs": [], "source": [ - "!pip install pinecone-client openai tiktoken" + "!pip install pinecone-client openai tiktoken langchain" ] }, { @@ -38,7 +38,7 @@ "import os\n", "import getpass\n", "\n", - "PINECONE_API_KEY = getpass.getpass(\"Pinecone API Key:\")" + "os.environ[\"PINECONE_API_KEY\"] = getpass.getpass(\"Pinecone API Key:\")" ] }, { @@ -48,7 +48,7 @@ "metadata": {}, "outputs": [], "source": [ - "PINECONE_ENV = getpass.getpass(\"Pinecone Environment:\")" + "os.environ[\"PINECONE_ENV\"] = getpass.getpass(\"Pinecone Environment:\")" ] }, { @@ -113,12 +113,22 @@ "\n", "# initialize pinecone\n", "pinecone.init(\n", - " api_key=PINECONE_API_KEY, # find at app.pinecone.io\n", - " environment=PINECONE_ENV, # next to api key in console\n", + " api_key=os.getenv(\"PINECONE_API_KEY\"), # find at app.pinecone.io\n", + " environment=os.getenv(\"PINECONE_ENV\"), # next to api key in console\n", ")\n", "\n", "index_name = \"langchain-demo\"\n", "\n", + "# First, check if our index already exists. If it doesn't, we create it\n", + "if index_name not in pinecone.list_indexes():\n", + " # we create a new index\n", + " pinecone.create_index(\n", + " name=index_name,\n", + " metric='cosine',\n", + " dimension=1536 \n", + ")\n", + "# The OpenAI embedding model `text-embedding-ada-002 uses 1536 dimensions`\n", + "docsearch = Pinecone.from_documents(docs, embeddings, index_name=index_name)\n", "\n", "# if you already have an index, you can load it like this\n",