format code

Author: Nyakult

evaluation/scripts/personamem/pm_search.py

@@ -315,9 +315,7 @@ def main(frame, version, top_k=20, num_workers=2):
 
     with open(f"results/pm/{frame}-{version}/{frame}_pm_search_results.json", "w") as f:
         json.dump(dict(all_search_results), f, indent=4)
-    print(
-        f"📁 Results saved to: mresults/pm/{frame}-{version}/{frame}_pm_search_results.json"
-    )
+    print(f"📁 Results saved to: mresults/pm/{frame}-{version}/{frame}_pm_search_results.json")
     print("=" * 80 + "\n")
 
 

evaluation/scripts/utils/client.py

@@ -54,7 +54,7 @@ def __init__(self, enable_graph=False):
     def add(self, messages, user_id, timestamp, batch_size=2):
         max_retries = 5
         for i in range(0, len(messages), batch_size):
-            batch_messages = messages[i: i + batch_size]
+            batch_messages = messages[i : i + batch_size]
             for attempt in range(max_retries):
                 try:
                     if self.enable_graph:
@@ -67,12 +67,15 @@ def add(self, messages, user_id, timestamp, batch_size=2):
                         )
                     else:
                         self.client.add(
-                            messages=batch_messages, timestamp=timestamp, user_id=user_id, async_mode=False
+                            messages=batch_messages,
+                            timestamp=timestamp,
+                            user_id=user_id,
+                            async_mode=False,
                         )
                     break
                 except Exception as e:
                     if attempt < max_retries - 1:
-                        time.sleep(2 ** attempt)
+                        time.sleep(2**attempt)
                     else:
                         raise e
 
@@ -108,17 +111,18 @@ def add(self, messages, user_id, batch_size=2):
         messages = [{"role": "assistant", "content": data, "created_at": iso_date}]
         """
         from memobase import ChatBlob
+
         real_uid = self.string_to_uuid(user_id)
         user = self.client.get_or_create_user(real_uid)
         for i in range(0, len(messages), batch_size):
-            batch_messages = messages[i: i + batch_size]
+            batch_messages = messages[i : i + batch_size]
             max_retries = 5
             for attempt in range(max_retries):
                 try:
                     _ = user.insert(ChatBlob(messages=batch_messages), sync=True)
                 except Exception as e:
                     if attempt < max_retries - 1:
-                        time.sleep(2 ** attempt)
+                        time.sleep(2**attempt)
                     else:
                         raise e
 
@@ -135,6 +139,7 @@ def search(self, query, user_id, top_k):
 
     def delete_user(self, user_id):
         from memobase.error import ServerError
+
         real_uid = self.string_to_uuid(user_id)
         with suppress(ServerError):
             self.client.delete_user(real_uid)
@@ -211,7 +216,7 @@ def add(self, messages, user_id, conv_id=None):
                 return response.text
             except Exception as e:
                 if attempt < max_retries - 1:
-                    time.sleep(2 ** attempt)
+                    time.sleep(2**attempt)
                 else:
                     raise e
 
@@ -238,7 +243,7 @@ def search(self, query, user_id, top_k):
                 return {"text_mem": [{"memories": res}]}
             except Exception as e:
                 if attempt < max_retries - 1:
-                    time.sleep(2 ** attempt)
+                    time.sleep(2**attempt)
                 else:
                     raise e
 
@@ -260,7 +265,7 @@ def add(self, messages, user_id):
                 break
             except Exception as e:
                 if attempt < max_retries - 1:
-                    time.sleep(2 ** attempt)
+                    time.sleep(2**attempt)
                 else:
                     raise e
 
@@ -280,7 +285,7 @@ def search(self, query, user_id, top_k):
                 return context
             except Exception as e:
                 if attempt < max_retries - 1:
-                    time.sleep(2 ** attempt)
+                    time.sleep(2**attempt)
                 else:
                     raise e
 
@@ -340,5 +345,3 @@ def wait_for_completion(self, task_id):
         m["created_at"] = iso_date
     client.add(messages, user_id)
     memories = client.search(query, user_id, top_k)
-
-