ykl7 commited on
Commit
921698f
·
1 Parent(s): 0b633df

fix prompt parsing

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -217,11 +217,11 @@ def reasoner(query: str, documents: list[str], llm_client: Any):
217
  # Accept user input
218
  if prompt := st.chat_input("Type here"):
219
  # Add user message to chat history
220
- prompt = prompt + " \n"+ " \n"+ f"Retriever: {selected_retriever}, Reasoner: {selected_reasoner}"
221
- st.session_state.messages.append({"role": "user", "content": prompt})
222
  # Display user message in chat message container
223
  with st.chat_message("user"):
224
- st.markdown(prompt)
225
 
226
  options = {}
227
  options["max_tokens"] = 500
 
217
  # Accept user input
218
  if prompt := st.chat_input("Type here"):
219
  # Add user message to chat history
220
+ display_message = prompt + " \n"+ " \n"+ f"Retriever: {selected_retriever}, Reasoner: {selected_reasoner}"
221
+ st.session_state.messages.append({"role": "user", "content": display_message})
222
  # Display user message in chat message container
223
  with st.chat_message("user"):
224
+ st.markdown(display_message)
225
 
226
  options = {}
227
  options["max_tokens"] = 500