Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -137,12 +137,14 @@ def invoke(openai_api_key, rag_option, prompt):
|
|
137 |
db = document_retrieval_chroma(llm, prompt)
|
138 |
completion = rag_chain(llm, prompt, db)
|
139 |
result = completion["result"]
|
|
|
140 |
elif (rag_option == "MongoDB"):
|
141 |
#splits = document_loading_splitting()
|
142 |
#document_storage_mongodb(splits)
|
143 |
db = document_retrieval_mongodb(llm, prompt)
|
144 |
completion = rag_chain(llm, prompt, db)
|
145 |
result = completion["result"]
|
|
|
146 |
else:
|
147 |
result = llm_chain(llm, prompt)
|
148 |
completion = result
|
|
|
137 |
db = document_retrieval_chroma(llm, prompt)
|
138 |
completion = rag_chain(llm, prompt, db)
|
139 |
result = completion["result"]
|
140 |
+
completion = completion.toJSON()
|
141 |
elif (rag_option == "MongoDB"):
|
142 |
#splits = document_loading_splitting()
|
143 |
#document_storage_mongodb(splits)
|
144 |
db = document_retrieval_mongodb(llm, prompt)
|
145 |
completion = rag_chain(llm, prompt, db)
|
146 |
result = completion["result"]
|
147 |
+
completion = completion.toJSON()
|
148 |
else:
|
149 |
result = llm_chain(llm, prompt)
|
150 |
completion = result
|