Update other_function.py

other_function.py

@@ -30,6 +30,16 @@ def generate_response(query):
         if chunk.choices[0].delta.content:
             response += chunk.choices[0].delta.content
     return response.replace("###", '').replace('\nUser:','')
+class ConversationBufferMemory:
+    def __init__(self, memory_key="chat_history"):
+        self.memory_key = memory_key
+        self.buffer = []
+
+    def add_to_memory(self, interaction):
+        self.buffer.append(interaction)
+
+    def get_memory(self):
+        return "\n".join([f"Human: {entry['user']}\nAssistant: {entry['assistant']}" for entry in self.buffer])
 
 
 def predict_pest(filepath):