hbui commited on
Commit
f7898d4
·
1 Parent(s): ecb943c

Update models/langOpen.py

Browse files
Files changed (1) hide show
  1. models/langOpen.py +5 -3
models/langOpen.py CHANGED
@@ -7,6 +7,10 @@ from langchain.chat_models import ChatOpenAI
7
  from langchain.embeddings.openai import OpenAIEmbeddings
8
  from langchain.prompts import PromptTemplate
9
  from langchain.vectorstores import FAISS
 
 
 
 
10
 
11
  load_dotenv()
12
 
@@ -31,9 +35,7 @@ class LangOpen:
31
  if os.path.exists(path=path):
32
  return FAISS.load_local(folder_path=path, embeddings=embeddings)
33
  else:
34
- faiss = FAISS.from_texts(
35
- "./assets/updated_calregs.txt", embedding=embeddings
36
- )
37
  faiss.save_local(path)
38
  return faiss
39
 
 
7
  from langchain.embeddings.openai import OpenAIEmbeddings
8
  from langchain.prompts import PromptTemplate
9
  from langchain.vectorstores import FAISS
10
+ from langchain.document_loaders import PyPDFLoader
11
+
12
+ loader = PyPDFLoader("./assets/CADWReg.pdf")
13
+ pages = loader.load_and_split()
14
 
15
  load_dotenv()
16
 
 
35
  if os.path.exists(path=path):
36
  return FAISS.load_local(folder_path=path, embeddings=embeddings)
37
  else:
38
+ faiss = FAISS.from_documents(pages, embedding)
 
 
39
  faiss.save_local(path)
40
  return faiss
41