bstraehle commited on
Commit
a52667e
·
1 Parent(s): 1701897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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