ykl7 commited on
Commit
e4cab9a
·
1 Parent(s): eb9293b

fix argument passing to function

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -147,7 +147,7 @@ def retriever(query: str, selected_retriever: str):
147
  # You could return retrieved info here.
148
  return corpus
149
 
150
- def reasoner(query: str, documents: list[str], llm_client: Any):
151
  """Simulate a 'reasoner' step, thinking about how to answer."""
152
  with st.chat_message("assistant"):
153
  placeholder = st.empty()
@@ -263,7 +263,7 @@ def main():
263
  st.markdown(display_message)
264
 
265
  retrieved_documents = retriever(prompt, selected_retriever)
266
- reasoning, decision = reasoner(prompt, retrieved_documents, llm_client)
267
 
268
  # Display assistant response in chat message container
269
  with st.chat_message("assistant"):
 
147
  # You could return retrieved info here.
148
  return corpus
149
 
150
+ def reasoner(query: str, documents: list[str], selected_reasoner: str, llm_client: Any):
151
  """Simulate a 'reasoner' step, thinking about how to answer."""
152
  with st.chat_message("assistant"):
153
  placeholder = st.empty()
 
263
  st.markdown(display_message)
264
 
265
  retrieved_documents = retriever(prompt, selected_retriever)
266
+ reasoning, decision = reasoner(prompt, retrieved_documents, selected_reasoner, llm_client)
267
 
268
  # Display assistant response in chat message container
269
  with st.chat_message("assistant"):