fix: update Pinecone SDK v7 compatibility
Browse files
rag.py
CHANGED
|
@@ -10,7 +10,7 @@ from langchain_classic.chains import RetrievalQA
|
|
| 10 |
from langchain_core.prompts import PromptTemplate
|
| 11 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 12 |
from langchain_community.vectorstores import Pinecone as LangchainPinecone
|
| 13 |
-
|
| 14 |
|
| 15 |
logger = logging.getLogger(__name__)
|
| 16 |
|
|
@@ -109,9 +109,9 @@ class RAGChain:
|
|
| 109 |
|
| 110 |
logger.info("Connecting to Pinecone...")
|
| 111 |
embeddings = HuggingFaceEmbeddings(model_name=EMBED_MODEL)
|
| 112 |
-
pc =
|
| 113 |
index = pc.Index(PINECONE_INDEX)
|
| 114 |
-
self.vectorstore = LangchainPinecone(index, embeddings, "text")
|
| 115 |
logger.info("Pinecone connected.")
|
| 116 |
|
| 117 |
prompt = PromptTemplate(
|
|
|
|
| 10 |
from langchain_core.prompts import PromptTemplate
|
| 11 |
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 12 |
from langchain_community.vectorstores import Pinecone as LangchainPinecone
|
| 13 |
+
import Pinecone as PineconeClient
|
| 14 |
|
| 15 |
logger = logging.getLogger(__name__)
|
| 16 |
|
|
|
|
| 109 |
|
| 110 |
logger.info("Connecting to Pinecone...")
|
| 111 |
embeddings = HuggingFaceEmbeddings(model_name=EMBED_MODEL)
|
| 112 |
+
pc = PineconeClient(api_key=PINECONE_API_KEY)
|
| 113 |
index = pc.Index(PINECONE_INDEX)
|
| 114 |
+
self.vectorstore = LangchainPinecone(index=index, embedding=embeddings, text_key="text")
|
| 115 |
logger.info("Pinecone connected.")
|
| 116 |
|
| 117 |
prompt = PromptTemplate(
|