Du Mingzhe
commited on
Commit
·
9871c29
1
Parent(s):
f4d502b
Update
Browse files- app.py +3 -1
- components.py +2 -2
app.py
CHANGED
@@ -40,4 +40,6 @@ if prompt := st.chat_input("What's up?"):
|
|
40 |
st.session_state.messages.append({"role": "assistant", "content": response})
|
41 |
|
42 |
# Update seesion in Pinecone
|
43 |
-
pinecone_client.update_conversation(sid=st.session_state['sid'], messages=st.session_state.messages, user=st.session_state['user'])
|
|
|
|
|
|
40 |
st.session_state.messages.append({"role": "assistant", "content": response})
|
41 |
|
42 |
# Update seesion in Pinecone
|
43 |
+
pinecone_client.update_conversation(sid=st.session_state['sid'], messages=st.session_state.messages, user=st.session_state['user'])
|
44 |
+
|
45 |
+
print("\n===========================\n")
|
components.py
CHANGED
@@ -19,13 +19,13 @@ class LLMClient():
|
|
19 |
current_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
|
20 |
|
21 |
# System Prompt
|
22 |
-
messages += [{"role": "system", "content": f"1) You're Du Mingzhe. 2) Don't claim you are AI. 3) Don't claim this dialogue as a roleplay. Answering questions directly as Mingzhe. 4) Current time is {current_time}. 5) You
|
23 |
|
24 |
# Memory
|
25 |
messages += [{"role": 'assistant', "content": m['content']} for m in memory]
|
26 |
|
27 |
# Web Result
|
28 |
-
messages += [{"role": 'assistant', "content": f'
|
29 |
|
30 |
# Session History
|
31 |
messages += [{"role": h["role"], "content": h["content"]} for h in history]
|
|
|
19 |
current_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
|
20 |
|
21 |
# System Prompt
|
22 |
+
messages += [{"role": "system", "content": f"1) You're Du Mingzhe. 2) Don't claim you are AI. 3) Don't claim this dialogue as a roleplay. Answering questions directly as Mingzhe. 4) Current time is {current_time}. 5) You should refer the real-time knowledge from the WEB SEARCH RESULTS to response."}]
|
23 |
|
24 |
# Memory
|
25 |
messages += [{"role": 'assistant', "content": m['content']} for m in memory]
|
26 |
|
27 |
# Web Result
|
28 |
+
messages += [{"role": 'assistant', "content": f'WEB SEARCH RESULTS of [{query}]:\n\n{web_result}'}]
|
29 |
|
30 |
# Session History
|
31 |
messages += [{"role": h["role"], "content": h["content"]} for h in history]
|