ykl7 commited on
Commit
f6f7e66
·
1 Parent(s): 96ee798
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -137,7 +137,6 @@ def retriever(query: str, selected_retriever: str):
137
  message = "Using the trained retriever to fetch detailed documents relevant to your query..."
138
  else:
139
  message = "No retriever selected. Skipping document retrieval."
140
-
141
 
142
  for chunk in message.split():
143
  text += chunk + " "
@@ -210,7 +209,7 @@ if prompt := st.chat_input("Type here"):
210
  llm_client = LLMReasoner(options)
211
 
212
 
213
- retrieved_documents = retriever(prompt)
214
  reasoning = reasoner(prompt, retrieved_documents, llm_client)
215
 
216
  # Display assistant response in chat message container
 
137
  message = "Using the trained retriever to fetch detailed documents relevant to your query..."
138
  else:
139
  message = "No retriever selected. Skipping document retrieval."
 
140
 
141
  for chunk in message.split():
142
  text += chunk + " "
 
209
  llm_client = LLMReasoner(options)
210
 
211
 
212
+ retrieved_documents = retriever(prompt, selected_retriever)
213
  reasoning = reasoner(prompt, retrieved_documents, llm_client)
214
 
215
  # Display assistant response in chat message container