Skip to content

Commit cfda529

Browse files
SanjoDeundiaketorreborre
SanjoDeundiak
authored and
etorreborre
committed
replacing logging -> logger 
1 parent bc7162e commit cfda529

File tree

1 file changed

+22
-22
lines changed

1 file changed

+22
-22
lines changed

‎mem0/memory/main.py‎

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,7 @@ def _add_to_vector_store(self, messages, metadata, filters, infer):
336336
response = remove_code_blocks(response)
337337
new_retrieved_facts = json.loads(response)["facts"]
338338
except Exception as e:
339-
logging.error(f"Error in new_retrieved_facts: {e}")
339+
logger.error(f"Error in new_retrieved_facts: {e}")
340340
new_retrieved_facts = []
341341

342342
if not new_retrieved_facts:
@@ -360,7 +360,7 @@ def _add_to_vector_store(self, messages, metadata, filters, infer):
360360
for item in retrieved_old_memory:
361361
unique_data[item["id"]] = item
362362
retrieved_old_memory = list(unique_data.values())
363-
logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
363+
logger.info(f"Total existing memories: {len(retrieved_old_memory)}")
364364

365365
# mapping UUIDs with integers for handling UUID hallucinations
366366
temp_uuid_mapping = {}
@@ -379,26 +379,26 @@ def _add_to_vector_store(self, messages, metadata, filters, infer):
379379
response_format={"type": "json_object"},
380380
)
381381
except Exception as e:
382-
logging.error(f"Error in new memory actions response: {e}")
382+
logger.error(f"Error in new memory actions response: {e}")
383383
response = ""
384384

385385
try:
386386
response = remove_code_blocks(response)
387387
new_memories_with_actions = json.loads(response)
388388
except Exception as e:
389-
logging.error(f"Invalid JSON response: {e}")
389+
logger.error(f"Invalid JSON response: {e}")
390390
new_memories_with_actions = {}
391391
else:
392392
new_memories_with_actions = {}
393393

394394
returned_memories = []
395395
try:
396396
for resp in new_memories_with_actions.get("memory", []):
397-
logging.info(resp)
397+
logger.info(resp)
398398
try:
399399
action_text = resp.get("text")
400400
if not action_text:
401-
logging.info("Skipping memory entry because of empty `text` field.")
401+
logger.info("Skipping memory entry because of empty `text` field.")
402402
continue
403403

404404
event_type = resp.get("event")
@@ -434,11 +434,11 @@ def _add_to_vector_store(self, messages, metadata, filters, infer):
434434
}
435435
)
436436
elif event_type == "NONE":
437-
logging.info("NOOP for Memory.")
437+
logger.info("NOOP for Memory.")
438438
except Exception as e:
439-
logging.error(f"Error processing memory action: {resp}, Error: {e}")
439+
logger.error(f"Error processing memory action: {resp}, Error: {e}")
440440
except Exception as e:
441-
logging.error(f"Error iterating new_memories_with_actions: {e}")
441+
logger.error(f"Error iterating new_memories_with_actions: {e}")
442442

443443
keys, encoded_ids = process_telemetry_filters(filters)
444444
capture_event(
@@ -801,7 +801,7 @@ def history(self, memory_id):
801801
return self.db.get_history(memory_id)
802802

803803
def _create_memory(self, data, existing_embeddings, metadata=None):
804-
logging.debug(f"Creating memory with {data=}")
804+
logger.debug(f"Creating memory with {data=}")
805805
if data in existing_embeddings:
806806
embeddings = existing_embeddings[data]
807807
else:
@@ -922,7 +922,7 @@ def _update_memory(self, memory_id, data, existing_embeddings, metadata=None):
922922
return memory_id
923923

924924
def _delete_memory(self, memory_id):
925-
logging.info(f"Deleting memory with {memory_id=}")
925+
logger.info(f"Deleting memory with {memory_id=}")
926926
existing_memory = self.vector_store.get(vector_id=memory_id)
927927
prev_value = existing_memory.payload["data"]
928928
self.vector_store.delete(vector_id=memory_id)
@@ -1164,7 +1164,7 @@ async def _add_to_vector_store(
11641164
response = remove_code_blocks(response)
11651165
new_retrieved_facts = json.loads(response)["facts"]
11661166
except Exception as e:
1167-
logging.error(f"Error in new_retrieved_facts: {e}")
1167+
logger.error(f"Error in new_retrieved_facts: {e}")
11681168
new_retrieved_facts = []
11691169

11701170
if not new_retrieved_facts:
@@ -1195,7 +1195,7 @@ async def process_fact_for_search(new_mem):
11951195
for item in retrieved_old_memory:
11961196
unique_data[item["id"]] = item
11971197
retrieved_old_memory = list(unique_data.values())
1198-
logging.info(f"Total existing memories: {len(retrieved_old_memory)}")
1198+
logger.info(f"Total existing memories: {len(retrieved_old_memory)}")
11991199

12001200
# mapping UUIDs with integers for handling UUID hallucinations
12011201
temp_uuid_mapping = {}
@@ -1214,23 +1214,23 @@ async def process_fact_for_search(new_mem):
12141214
response_format={"type": "json_object"},
12151215
)
12161216
except Exception as e:
1217-
logging.error(f"Error in new memory actions response: {e}")
1217+
logger.error(f"Error in new memory actions response: {e}")
12181218
response = ""
12191219
try:
12201220
response = remove_code_blocks(response)
12211221
new_memories_with_actions = json.loads(response)
12221222
except Exception as e:
1223-
logging.error(f"Invalid JSON response: {e}")
1223+
logger.error(f"Invalid JSON response: {e}")
12241224
new_memories_with_actions = {}
12251225

12261226
returned_memories = []
12271227
try:
12281228
for resp in new_memories_with_actions.get("memory", []):
1229-
logging.info(resp)
1229+
logger.info(resp)
12301230
try:
12311231
action_text = resp.get("text")
12321232
if not action_text:
1233-
logging.info("Skipping memory entry because of empty `text` field.")
1233+
logger.info("Skipping memory entry because of empty `text` field.")
12341234
continue
12351235

12361236
event_type = resp.get("event")
@@ -1271,11 +1271,11 @@ async def process_fact_for_search(new_mem):
12711271
{"id": temp_uuid_mapping[id], "memory": action_text, "event": event_type}
12721272
)
12731273
elif event_type == "NONE":
1274-
logging.info("NOOP for Memory (async).")
1274+
logger.info("NOOP for Memory (async).")
12751275
except Exception as e:
1276-
logging.error(f"Error processing memory action (async): {resp}, Error: {e}")
1276+
logger.error(f"Error processing memory action (async): {resp}, Error: {e}")
12771277
except Exception as e:
1278-
logging.error(f"Error iterating new_memories_with_actions: {e}")
1278+
logger.error(f"Error iterating new_memories_with_actions: {e}")
12791279

12801280
keys, encoded_ids = process_telemetry_filters(filters)
12811281
capture_event(
@@ -1641,7 +1641,7 @@ async def history(self, memory_id):
16411641
return await asyncio.to_thread(self.db.get_history, memory_id)
16421642

16431643
async def _create_memory(self, data, existing_embeddings, metadata=None):
1644-
logging.debug(f"Creating memory with {data=}")
1644+
logger.debug(f"Creating memory with {data=}")
16451645
if data in existing_embeddings:
16461646
embeddings = existing_embeddings[data]
16471647
else:
@@ -1783,7 +1783,7 @@ async def _update_memory(self, memory_id, data, existing_embeddings, metadata=No
17831783
return memory_id
17841784

17851785
async def _delete_memory(self, memory_id):
1786-
logging.info(f"Deleting memory with {memory_id=}")
1786+
logger.info(f"Deleting memory with {memory_id=}")
17871787
existing_memory = await asyncio.to_thread(self.vector_store.get, vector_id=memory_id)
17881788
prev_value = existing_memory.payload["data"]
17891789

0 commit comments

Comments
 (0)