Dylan-Kaneshiro commited on
Commit
b0809f3
·
1 Parent(s): bd21bb4

Create matt_function.py

Browse files
Files changed (1) hide show
  1. matt_function.py +27 -0
matt_function.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.llms import OpenAI
2
+ from langchain.document_loaders import PyPDFLoader
3
+ from langchain.embeddings import OpenAIEmbeddings
4
+ from langchain.vectorstores import Chroma
5
+
6
+ # Import OpenAI as main LLM service
7
+ from langchain.llms import OpenAI
8
+ from langchain.embeddings import OpenAIEmbeddings
9
+
10
+ # Import PDF document loaders...there's other ones as well!
11
+ from langchain.document_loaders import PyPDFLoader
12
+ from langchain.indexes import VectorstoreIndexCreator
13
+
14
+
15
+
16
+ def create_VSI(files, index): #add vsi and filepaths
17
+
18
+ pathnames = [file.name for file in files]
19
+
20
+ loaders = []
21
+ for pdf in pathnames:
22
+ loader = PyPDFLoader(pdf)
23
+ loaders.append(loader)
24
+
25
+ index = VectorstoreIndexCreator().from_loaders(loaders)
26
+
27
+ return index, "Upload complete"