chukbert commited on
Commit
e0cb3cd
·
verified ·
1 Parent(s): 031ce83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- from chromadb.config import Settings
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
- client_settings=Settings(
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