Update rag.py
Browse files
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(
|
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():
|