Du Mingzhe commited on
Commit
e05958b
·
1 Parent(s): e59ee58
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ if prompt := st.chat_input("What's up?"):
26
  with st.chat_message("user"):
27
  st.markdown(prompt)
28
  with st.chat_message("assistant"):
29
- memory = pinecone_client.query_conversation(messages=st.session_state.messages, user=st.session_state['user'], top_k=3)
30
  web_result = web_searcher.query_web_llm(query=prompt, num_web_results=5)
31
  stream = llm_client.response_generate(prompt, st.session_state.messages, memory, web_result)
32
  response = st.write_stream(stream)
 
26
  with st.chat_message("user"):
27
  st.markdown(prompt)
28
  with st.chat_message("assistant"):
29
+ memory = pinecone_client.query_conversation(messages=st.session_state.messages[:-5], user=st.session_state['user'], top_k=3)
30
  web_result = web_searcher.query_web_llm(query=prompt, num_web_results=5)
31
  stream = llm_client.response_generate(prompt, st.session_state.messages, memory, web_result)
32
  response = st.write_stream(stream)