update

app.py

@@ -1,8 +1,6 @@
-from pinecone import Pinecone
-
-from openai import OpenAI
 import streamlit as st
 import uuid
+import json
 
 from components import PersonalIndexClient, LLMClient
 
@@ -16,10 +14,12 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
     st.session_state['sid'] = uuid.uuid1().hex
 
+# Display History
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# New Click
 if prompt := st.chat_input("What's up?"):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
@@ -30,4 +30,12 @@ if prompt := st.chat_input("What's up?"):
         response = st.write_stream(stream)
     st.session_state.messages.append({"role": "assistant", "content": response})
     
+    current_history = json.dumps(st.session_state.messages)
+    
+    pinecone_client.create(data=[{
+        'id': '123',
+        'content': current_history,
+        'metadata': {'hello': 'world', 'content': current_history},
+    }])
+    
     

components.py

@@ -16,7 +16,7 @@ class LLMClient():
         current_time = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
         
         # System Prompt
-        messages += [{"role": "system", "content": f"1) You're Du Mingzhe, a computer science researcher. 2) Don't claim you are created by OpenAI. 3) Current time is {current_time}."}]
+        messages += [{"role": "system", "content": f"1) You're Du Mingzhe. 2) Don't claim you are created by OpenAI. 3) Current time is {current_time}."}]
         
         # Session History
         messages += [{"role": h["role"], "content": h["content"]} for h in history]
@@ -27,6 +27,7 @@ class LLMClient():
             messages = messages,
             stream=True,
         )
+        
         return stream
     
 class EmbeddingModel(object):