Skip to content

Commit c407987

Browse files
authored
Feat/add timerlog (#317)
* feat: add logs for cube * feat: add loggers for mem * add dict timer log * fix: change ci code * update size * fix ci
1 parent fcdb21c commit c407987

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/memos/mem_os/product.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import json
33
import os
44
import random
5+
import sys
56
import time
67

78
from collections.abc import Generator
@@ -1315,6 +1316,9 @@ def search(
13151316
# Load user cubes if not already loaded
13161317
time_start = time.time()
13171318
self._load_user_cubes(user_id, self.default_cube_config)
1319+
dict_size = sys.getsizeof(self.mem_cubes._dict)
1320+
size_mb = dict_size / (1024 * 1024)
1321+
logger.info(f"now search memcubes_size is : {len(self.mem_cubes)} {size_mb}MB")
13181322
load_user_cubes_time_end = time.time()
13191323
logger.info(
13201324
f"time search: load_user_cubes time user_id: {user_id} time is: {load_user_cubes_time_end - time_start}"
@@ -1363,7 +1367,9 @@ def add(
13631367

13641368
# Load user cubes if not already loaded
13651369
self._load_user_cubes(user_id, self.default_cube_config)
1366-
1370+
dict_size = sys.getsizeof(self.mem_cubes._dict)
1371+
size_mb = dict_size / (1024 * 1024)
1372+
logger.info(f"now add memcubes_size is : {len(self.mem_cubes)} {size_mb}MB")
13671373
result = super().add(
13681374
messages, memory_content, doc_path, mem_cube_id, user_id, session_id=session_id
13691375
)

0 commit comments

Comments
 (0)