Update application/utils/convs_handler.py

application/utils/convs_handler.py

@@ -1,27 +1,31 @@
+# application/utils/convs_handler.py
 import uuid
+
 baseSysPrompt = "You are a helpful and harmless AI assistant. You are Xylaria, made by sk md saad amin. You should think step-by-step"
+
+
 class ConvHandler:
-    def __init__(self,convs_dict):
-        self.convs_dict = convs_dict;
+    def __init__(self, convs_dict):
+        self.convs_dict = convs_dict
 
-    def get_conv(self,ip):
-        if(ip not in self.convs_dict):
-            self.convs_dict[ip] = {"metadata": []}
+    def get_conv(self, ip):
+        if ip not in self.convs_dict:
+            self.convs_dict[ip] = {"metadata": [], "memory": ""}  # Initialize
         return self.convs_dict[ip]['metadata']
 
-    def create_conv(self,ip,sysPrompt):
-        user = self.convs_dict.get(ip,False)
-        if(user==False):
+    def create_conv(self, ip, sysPrompt):
+        user = self.convs_dict.get(ip, False)
+        if user == False:
             return f"user not found. {self.convs_dict}", 404
         convId = str(uuid.uuid4())
         user[convId] = {
-            "messages": [{"role":"system", "content": baseSysPrompt + sysPrompt}],
+            "messages": [{"role": "system", "content": baseSysPrompt + sysPrompt}],
             "title": "New Chat"
         }
         return {"convId": convId}
-        
-    def fetch_conv(self,ip,convId):
-        user = self.convs_dict.get(ip,False)
-        if(user==False):
+
+    def fetch_conv(self, ip, convId):
+        user = self.convs_dict.get(ip, False)
+        if user == False:
             return f"user not found. {self.convs_dict}", 404
-        return user[convId]   
+        return user[convId]