Du Mingzhe commited on
Commit
9871c29
·
1 Parent(s): f4d502b
Files changed (2) hide show
  1. app.py +3 -1
  2. 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 are able to refer the real-time knowledge from the web search results."}]
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: {query}\n\n{web_result}'}]
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]