Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -98,7 +98,7 @@ def invoke(openai_api_key, rag, prompt):
|
|
98 |
# persist_directory = CHROMA_DIR)
|
99 |
# Document retrieval
|
100 |
result = document_retrieval_chroma(llm, prompt)
|
101 |
-
|
102 |
# Document retrieval
|
103 |
result = document_retrieval_mongodb(llm, prompt)
|
104 |
else:
|
|
|
98 |
# persist_directory = CHROMA_DIR)
|
99 |
# Document retrieval
|
100 |
result = document_retrieval_chroma(llm, prompt)
|
101 |
+
elif (rag == "MongoDB"):
|
102 |
# Document retrieval
|
103 |
result = document_retrieval_mongodb(llm, prompt)
|
104 |
else:
|