diff --git a/docs/docs/integrations/retrievers/self_query/pinecone.ipynb b/docs/docs/integrations/retrievers/self_query/pinecone.ipynb index 5555715585b..aabc5c284bd 100644 --- a/docs/docs/integrations/retrievers/self_query/pinecone.ipynb +++ b/docs/docs/integrations/retrievers/self_query/pinecone.ipynb @@ -42,7 +42,7 @@ "metadata": {}, "outputs": [], "source": [ - "%pip install --upgrade --quiet pinecone-client" + "%pip install --upgrade --quiet pinecone-notebooks pinecone-client==3.2.2" ] }, { @@ -61,13 +61,50 @@ } ], "source": [ + "# Connect to Pinecone and get an API key.\n", + "from pinecone_notebooks.colab import Authenticate\n", + "\n", + "Authenticate()\n", + "\n", "import os\n", "\n", - "import pinecone\n", + "api_key = os.environ[\"PINECONE_API_KEY\"]" + ] + }, + { + "cell_type": "markdown", + "id": "bdaebe0d", + "metadata": {}, + "source": [ + "We want to use `OpenAIEmbeddings` so we have to get the OpenAI API Key." + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "04b33384", + "metadata": {}, + "outputs": [], + "source": [ + "import getpass\n", "\n", - "pinecone.init(\n", - " api_key=os.environ[\"PINECONE_API_KEY\"], environment=os.environ[\"PINECONE_ENV\"]\n", - ")" + "os.environ[\"OPENAI_API_KEY\"] = getpass.getpass(\"OpenAI API Key:\")" + ] + }, + { + "cell_type": "code", + "execution_count": null, + "id": "c9ab3a96", + "metadata": {}, + "outputs": [], + "source": [ + "from pinecone import Pinecone, ServerlessSpec\n", + "\n", + "api_key = os.getenv(\"PINECONE_API_KEY\") or \"PINECONE_API_KEY\"\n", + "\n", + "index_name = \"langchain-self-retriever-demo\"\n", + "\n", + "pc = Pinecone(api_key=api_key)" ] }, { @@ -82,8 +119,15 @@ "from langchain_pinecone import PineconeVectorStore\n", "\n", "embeddings = OpenAIEmbeddings()\n", + "\n", "# create new index\n", - "pinecone.create_index(\"langchain-self-retriever-demo\", dimension=1536)" + "if index_name not in pc.list_indexes().names():\n", + " pc.create_index(\n", + " name=index_name,\n", + " dimension=1536,\n", + " metric=\"cosine\",\n", + " spec=ServerlessSpec(cloud=\"aws\", region=\"us-east-1\"),\n", + " )" ] }, {