Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from langchain.callbacks import get_openai_callback
|
|
14 |
from io import BytesIO
|
15 |
from langchain.chains.combine_documents import create_stuff_documents_chain
|
16 |
from langchain_core.prompts import ChatPromptTemplate
|
17 |
-
|
18 |
|
19 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
20 |
|
@@ -32,12 +32,12 @@ def process_pdf(uploaded_file):
|
|
32 |
|
33 |
def create_vectorstore(splits):
|
34 |
embeddings = OpenAIEmbeddings()
|
|
|
|
|
35 |
vectorstore = Chroma.from_documents(
|
36 |
-
splits,
|
37 |
-
embeddings,
|
38 |
-
|
39 |
-
chroma_db_impl="duckdb+parquet"
|
40 |
-
)
|
41 |
)
|
42 |
retriever = vectorstore.as_retriever()
|
43 |
return retriever
|
|
|
14 |
from io import BytesIO
|
15 |
from langchain.chains.combine_documents import create_stuff_documents_chain
|
16 |
from langchain_core.prompts import ChatPromptTemplate
|
17 |
+
import chromadb
|
18 |
|
19 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
20 |
|
|
|
32 |
|
33 |
def create_vectorstore(splits):
|
34 |
embeddings = OpenAIEmbeddings()
|
35 |
+
client = chromadb.Client()
|
36 |
+
|
37 |
vectorstore = Chroma.from_documents(
|
38 |
+
splits,
|
39 |
+
embeddings,
|
40 |
+
client=client
|
|
|
|
|
41 |
)
|
42 |
retriever = vectorstore.as_retriever()
|
43 |
return retriever
|