zzejiao commited on
Commit
78717f7
·
1 Parent(s): fa9acd2

change last 4 history(2 round) to last 5 round

Browse files
Files changed (1) hide show
  1. src/app.py +2 -3
src/app.py CHANGED
@@ -288,8 +288,8 @@ if user_input := st.chat_input("Ask me questions about the CANMAT depression gui
288
  st.session_state.last_top_p_used = top_p
289
  st.session_state.last_max_length_used = max_length
290
 
291
- # ===== latest 4 histories =====
292
- history = st.session_state.messages[:-1][-4:]
293
 
294
  placeholder = st.chat_message("assistant").empty()
295
  collected = ""
@@ -316,7 +316,6 @@ if user_input := st.chat_input("Ask me questions about the CANMAT depression gui
316
  collected += chunk
317
  placeholder.markdown(collected)
318
  finally:
319
- # 恢复原始的 client
320
  Rag.llm_client = original_client
321
 
322
  t1 = time.perf_counter()
 
288
  st.session_state.last_top_p_used = top_p
289
  st.session_state.last_max_length_used = max_length
290
 
291
+ # ===== latest 10 histories(5 round) =====
292
+ history = st.session_state.messages[:-1][-10:]
293
 
294
  placeholder = st.chat_message("assistant").empty()
295
  collected = ""
 
316
  collected += chunk
317
  placeholder.markdown(collected)
318
  finally:
 
319
  Rag.llm_client = original_client
320
 
321
  t1 = time.perf_counter()