update

app.py

@@ -30,12 +30,4 @@ 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},
-    }])
-    
-    
+    pinecone_client.update_conversation(sid=st.session_state['sid'], messages=st.session_state.messages, namespace='default')

components.py

@@ -1,6 +1,8 @@
 # Author: Du Mingzhe ([email protected])
 # Date: 2024/03/09
 
+import json
+
 from openai import OpenAI
 from pinecone import Pinecone
 from datetime import datetime
@@ -77,4 +79,21 @@ class PersonalIndexClient(object):
             include_metadata = True,
             filter = filter,
         )
-        return results    
+        return results  
+    
+    def update_conversation(self, sid, messages, namespace='default'):
+        index_id = f'conv_{sid}'
+        
+        messages_dump = json.dumps(messages)
+        
+        metadata = {
+            'time': datetime.now().strftime("%d/%m/%Y %H:%M:%S"),
+            'type': 'conversation',
+            'content': messages_dump,
+        }
+        
+        self.create(data=[{
+            'id': index_id,
+            'content': messages_dump,
+            'metadata': metadata,
+        }])