Update models/langOpen.py

models/langOpen.py

@@ -1,18 +1,13 @@
 import os
 
 import openai
-from dotenv import load_dotenv
+
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
-from langchain.document_loaders import PyPDFLoader
-from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.prompts import PromptTemplate
-from langchain.vectorstores import FAISS
-
-loader = PyPDFLoader("./assets/pdf/CADWReg.pdf")
-pages = loader.load_and_split()
 
-load_dotenv()
+from langchain.embeddings import OpenAIEmbeddings
+from langchain.prompts import PromptTemplate
+from langchain_pinecone import PineconeVectorStore
 
 prompt_template = """Answer the question using the given context to the best of your ability. 
 If you don't know, answer I don't know.
@@ -29,15 +24,11 @@ class LangOpen:
         self.chain = LLMChain(llm=self.llm, prompt=PROMPT)
 
     def initialize_index(self, index_name):
-        path = f"./vectorStores/{index_name}"
-        embeddings = OpenAIEmbeddings()
-        
-        if os.path.exists(path=path):
-            return FAISS.load_local(folder_path=path, embeddings=embeddings)
-        else:
-            faiss = FAISS.from_documents(pages, embeddings)
-            faiss.save_local(path)
-            return faiss
+        embeddings = OpenAIEmbeddings(model="text-embedding-3-large")
+        index_name = "openai-embeddings"
+        vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
+        return vectorstore
+
 
     def get_response(self, query_str):
         print("query_str: ", query_str)