Skip to content

Commit eb73b75

Browse files
authored
fix: reorganize and schedular bug (#223)
* feat: illumination fix * fix: reorganize bug * fix: bug for schedular config
1 parent ae3fc53 commit eb73b75

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

src/memos/api/config.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,6 @@ def get_scheduler_config() -> dict[str, Any]:
251251
"backend": "general_scheduler",
252252
"config": {
253253
"top_k": int(os.getenv("MOS_SCHEDULER_TOP_K", "10")),
254-
"top_n": int(os.getenv("MOS_SCHEDULER_TOP_N", "5")),
255254
"act_mem_update_interval": int(
256255
os.getenv("MOS_SCHEDULER_ACT_MEM_UPDATE_INTERVAL", "300")
257256
),

src/memos/memories/textual/tree_text_memory/organize/reorganizer.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def __init__(
6565
self.resolver = NodeHandler(graph_store=graph_store, llm=llm, embedder=embedder)
6666

6767
self.is_reorganize = is_reorganize
68-
self._reorganize_needed = False
68+
self._reorganize_needed = True
6969
if self.is_reorganize:
7070
# ____ 1. For queue message driven thread ___________
7171
self.thread = threading.Thread(target=self._run_message_consumer_loop)
@@ -124,8 +124,8 @@ def _run_structure_organizer_loop(self):
124124
"""
125125
import schedule
126126

127-
schedule.every(600).seconds.do(self.optimize_structure, scope="LongTermMemory")
128-
schedule.every(600).seconds.do(self.optimize_structure, scope="UserMemory")
127+
schedule.every(100).seconds.do(self.optimize_structure, scope="LongTermMemory")
128+
schedule.every(100).seconds.do(self.optimize_structure, scope="UserMemory")
129129

130130
logger.info("Structure optimizer schedule started.")
131131
while not getattr(self, "_stop_scheduler", False):
@@ -165,7 +165,7 @@ def handle_add(self, message: QueueMessage):
165165
for added_node, existing_node, relation in detected_relationships:
166166
self.resolver.resolve(added_node, existing_node, relation)
167167

168-
self._reorganize_needed = False
168+
self._reorganize_needed = True
169169

170170
def handle_remove(self, message: QueueMessage):
171171
logger.debug(f"Handling remove operation: {str(message)[:50]}")

0 commit comments

Comments
 (0)