MukulRay commited on
Commit
f5f98aa
·
1 Parent(s): f494f57

fix: downgrade pinecone-client for langchain_community compatibility

Browse files
Files changed (2) hide show
  1. rag.py +3 -3
  2. requirements.txt +1 -1
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
- from pinecone import Pinecone as PineconeClient
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 = 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(
 
10
  from langchain_core.prompts import PromptTemplate
11
  from langchain_community.embeddings import HuggingFaceEmbeddings
12
  from langchain_community.vectorstores import Pinecone as LangchainPinecone
13
+ from pinecone import Pinecone
14
 
15
  logger = logging.getLogger(__name__)
16
 
 
109
 
110
  logger.info("Connecting to Pinecone...")
111
  embeddings = HuggingFaceEmbeddings(model_name=EMBED_MODEL)
112
+ pc = Pinecone(api_key=PINECONE_API_KEY)
113
  index = pc.Index(PINECONE_INDEX)
114
+ self.vectorstore = LangchainPinecone(index, embeddings, "text")
115
  logger.info("Pinecone connected.")
116
 
117
  prompt = PromptTemplate(
requirements.txt CHANGED
@@ -16,7 +16,7 @@ langchain-community==0.4.1
16
  langchain-classic==1.0.3
17
  langchain-groq==1.1.2
18
  groq==0.37.1
19
- pinecone==7.3.0
20
  sentence-transformers==4.1.0
21
 
22
  # Utilities
 
16
  langchain-classic==1.0.3
17
  langchain-groq==1.1.2
18
  groq==0.37.1
19
+ pinecone-client==3.2.2
20
  sentence-transformers==4.1.0
21
 
22
  # Utilities