bstraehle commited on
Commit
3754c7e
·
1 Parent(s): 290fab7

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -80,7 +80,7 @@ def rag_batch(config):
80
  document_storage_mongodb(chunks)
81
 
82
  def document_retrieval_chroma():
83
- return Chroma(embedding_function = OpenAIEmbeddings(disallowed_special = ()),
84
  persist_directory = CHROMA_DIR)
85
 
86
  def document_retrieval_mongodb():
 
80
  document_storage_mongodb(chunks)
81
 
82
  def document_retrieval_chroma():
83
+ return Chroma(embedding_function = OpenAIEmbeddings(),
84
  persist_directory = CHROMA_DIR)
85
 
86
  def document_retrieval_mongodb():