Du Mingzhe
commited on
Commit
·
e05958b
1
Parent(s):
e59ee58
Update
Browse files
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)
|