klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixes: Mem0 Setup, Logging, Docs (#3080)

Browse Source
This commit is contained in:
Jainish
2025-07-04 03:10:39 +05:30
committed by GitHub
parent eb7c712aa6
commit a20b68fcec
5 changed files with 34 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
mem0/memory/main.py
View File

@@ -336,7 +336,7 @@ class Memory(MemoryBase):
response = remove_code_blocks(response)
new_retrieved_facts = json.loads(response)["facts"]
except Exception as e:
logging.error(f"Error in new_retrieved_facts: {e}")
logger.error(f"Error in new_retrieved_facts: {e}")
new_retrieved_facts = []
if not new_retrieved_facts:
@@ -360,7 +360,7 @@ class Memory(MemoryBase):
for item in retrieved_old_memory:
unique_data[item["id"]] = item
retrieved_old_memory = list(unique_data.values())
logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
logger.info(f"Total existing memories: {len(retrieved_old_memory)}")
# mapping UUIDs with integers for handling UUID hallucinations
temp_uuid_mapping = {}
@@ -379,14 +379,14 @@ class Memory(MemoryBase):
response_format={"type": "json_object"},
)
except Exception as e:
logging.error(f"Error in new memory actions response: {e}")
logger.error(f"Error in new memory actions response: {e}")
response = ""
try:
response = remove_code_blocks(response)
new_memories_with_actions = json.loads(response)
except Exception as e:
logging.error(f"Invalid JSON response: {e}")
logger.error(f"Invalid JSON response: {e}")
new_memories_with_actions = {}
else:
new_memories_with_actions = {}
@@ -394,11 +394,11 @@ class Memory(MemoryBase):
returned_memories = []
try:
for resp in new_memories_with_actions.get("memory", []):
logging.info(resp)
logger.info(resp)
try:
action_text = resp.get("text")
if not action_text:
logging.info("Skipping memory entry because of empty `text` field.")
logger.info("Skipping memory entry because of empty `text` field.")
continue
event_type = resp.get("event")
@@ -434,11 +434,11 @@ class Memory(MemoryBase):
}
)
elif event_type == "NONE":
logging.info("NOOP for Memory.")
logger.info("NOOP for Memory.")
except Exception as e:
logging.error(f"Error processing memory action: {resp}, Error: {e}")
logger.error(f"Error processing memory action: {resp}, Error: {e}")
except Exception as e:
logging.error(f"Error iterating new_memories_with_actions: {e}")
logger.error(f"Error iterating new_memories_with_actions: {e}")
keys, encoded_ids = process_telemetry_filters(filters)
capture_event(
@@ -801,7 +801,7 @@ class Memory(MemoryBase):
return self.db.get_history(memory_id)
def _create_memory(self, data, existing_embeddings, metadata=None):
logging.debug(f"Creating memory with {data=}")
logger.debug(f"Creating memory with {data=}")
if data in existing_embeddings:
embeddings = existing_embeddings[data]
else:
@@ -922,7 +922,7 @@ class Memory(MemoryBase):
return memory_id
def _delete_memory(self, memory_id):
logging.info(f"Deleting memory with {memory_id=}")
logger.info(f"Deleting memory with {memory_id=}")
existing_memory = self.vector_store.get(vector_id=memory_id)
prev_value = existing_memory.payload["data"]
self.vector_store.delete(vector_id=memory_id)
@@ -1164,7 +1164,7 @@ class AsyncMemory(MemoryBase):
response = remove_code_blocks(response)
new_retrieved_facts = json.loads(response)["facts"]
except Exception as e:
logging.error(f"Error in new_retrieved_facts: {e}")
logger.error(f"Error in new_retrieved_facts: {e}")
new_retrieved_facts = []
if not new_retrieved_facts:
@@ -1194,7 +1194,7 @@ class AsyncMemory(MemoryBase):
for item in retrieved_old_memory:
unique_data[item["id"]] = item
retrieved_old_memory = list(unique_data.values())
logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
logger.info(f"Total existing memories: {len(retrieved_old_memory)}")
temp_uuid_mapping = {}
for idx, item in enumerate(retrieved_old_memory):
temp_uuid_mapping[str(idx)] = item["id"]
@@ -1211,20 +1211,20 @@ class AsyncMemory(MemoryBase):
response_format={"type": "json_object"},
)
except Exception as e:
logging.error(f"Error in new memory actions response: {e}")
logger.error(f"Error in new memory actions response: {e}")
response = ""
try:
response = remove_code_blocks(response)
new_memories_with_actions = json.loads(response)
except Exception as e:
logging.error(f"Invalid JSON response: {e}")
logger.error(f"Invalid JSON response: {e}")
new_memories_with_actions = {}
returned_memories = []
try:
memory_tasks = []
for resp in new_memories_with_actions.get("memory", []):
logging.info(resp)
logger.info(resp)
try:
action_text = resp.get("text")
if not action_text:
@@ -1254,9 +1254,9 @@ class AsyncMemory(MemoryBase):
task = asyncio.create_task(self._delete_memory(memory_id=temp_uuid_mapping[resp.get("id")]))
memory_tasks.append((task, resp, "DELETE", temp_uuid_mapping[resp.get("id")]))
elif event_type == "NONE":
logging.info("NOOP for Memory (async).")
logger.info("NOOP for Memory (async).")
except Exception as e:
logging.error(f"Error processing memory action (async): {resp}, Error: {e}")
logger.error(f"Error processing memory action (async): {resp}, Error: {e}")
for task, resp, event_type, mem_id in memory_tasks:
try:
@@ -1275,9 +1275,9 @@ class AsyncMemory(MemoryBase):
elif event_type == "DELETE":
returned_memories.append({"id": mem_id, "memory": resp.get("text"), "event": event_type})
except Exception as e:
logging.error(f"Error awaiting memory task (async): {e}")
logger.error(f"Error awaiting memory task (async): {e}")
except Exception as e:
logging.error(f"Error in memory processing loop (async): {e}")
logger.error(f"Error in memory processing loop (async): {e}")
keys, encoded_ids = process_telemetry_filters(effective_filters)
capture_event(
@@ -1653,7 +1653,7 @@ class AsyncMemory(MemoryBase):
return await asyncio.to_thread(self.db.get_history, memory_id)
async def _create_memory(self, data, existing_embeddings, metadata=None):
logging.debug(f"Creating memory with {data=}")
logger.debug(f"Creating memory with {data=}")
if data in existing_embeddings:
embeddings = existing_embeddings[data]
else:
@@ -1795,7 +1795,7 @@ class AsyncMemory(MemoryBase):
return memory_id
async def _delete_memory(self, memory_id):
logging.info(f"Deleting memory with {memory_id=}")
logger.info(f"Deleting memory with {memory_id=}")
existing_memory = await asyncio.to_thread(self.vector_store.get, vector_id=memory_id)
prev_value = existing_memory.payload["data"]

2
mem0/vector_stores/pinecone.py
View File

@@ -98,7 +98,7 @@ class PineconeDB(VectorStoreBase):
existing_indexes = self.list_cols().names()
if self.collection_name in existing_indexes:
logging.debug(f"Index {self.collection_name} already exists. Skipping creation.")
logger.debug(f"Index {self.collection_name} already exists. Skipping creation.")
self.index = self.client.Index(self.collection_name)
return

2
mem0/vector_stores/qdrant.py
View File

@@ -83,7 +83,7 @@ class Qdrant(VectorStoreBase):
response = self.list_cols()
for collection in response.collections:
if collection.name == self.collection_name:
logging.debug(f"Collection {self.collection_name} already exists. Skipping creation.")
logger.debug(f"Collection {self.collection_name} already exists. Skipping creation.")
return
self.client.create_collection(

2
mem0/vector_stores/weaviate.py
View File

@@ -102,7 +102,7 @@ class Weaviate(VectorStoreBase):
distance (str, optional): Distance metric for vector similarity. Defaults to "cosine".
"""
if self.client.collections.exists(self.collection_name):
logging.debug(f"Collection {self.collection_name} already exists. Skipping creation.")
logger.debug(f"Collection {self.collection_name} already exists. Skipping creation.")
return
properties = [