Formatting and Client changes (#2247)

mem0/memory/main.py

@@ -178,7 +178,7 @@ class Memory(MemoryBase):
                 retrieved_old_memory.append({"id": mem.id, "text": mem.payload["data"]})
         unique_data = {}
         for item in retrieved_old_memory:
-            unique_data[item['id']] = item
+            unique_data[item["id"]] = item
         retrieved_old_memory = list(unique_data.values())
         logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
 

mem0/memory/telemetry.py

@@ -29,7 +29,7 @@ class AnonymousTelemetry:
         if not MEM0_TELEMETRY:
             self.posthog.disabled = True
 
-    def capture_event(self, event_name, properties=None):
+    def capture_event(self, event_name, properties=None, user_email=None):
         if properties is None:
             properties = {}
         properties = {
@@ -43,7 +43,8 @@ class AnonymousTelemetry:
             "machine": platform.machine(),
             **properties,
         }
-        self.posthog.capture(distinct_id=self.user_id, event=event_name, properties=properties)
+        distinct_id = self.user_id if user_email is None else user_email
+        self.posthog.capture(distinct_id=distinct_id, event=event_name, properties=properties)
 
     def close(self):
         self.posthog.shutdown()
@@ -82,4 +83,4 @@ def capture_client_event(event_name, instance, additional_data=None):
     if additional_data:
         event_data.update(additional_data)
 
-    telemetry.capture_event(event_name, event_data)
+    telemetry.capture_event(event_name, event_data, instance.user_email)

mem0/memory/utils.py

@@ -55,8 +55,11 @@ def get_image_description(image_url):
             {
                 "role": "user",
                 "content": [
-                    {"type": "text", "text": "Provide a description of the image and do not include any additional text."},
-                    {"type": "image_url", "image_url": {"url": image_url}}
+                    {
+                        "type": "text",
+                        "text": "Provide a description of the image and do not include any additional text.",
+                    },
+                    {"type": "image_url", "image_url": {"url": image_url}},
                 ],
             },
         ],