Skip to content

Commit 731f00d

Browse files
committed
revise naive memcube creation in server router
1 parent f0e8aab commit 731f00d

File tree

1 file changed

+11
-18
lines changed

1 file changed

+11
-18
lines changed

src/memos/api/routers/server_router.py

Lines changed: 11 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -107,21 +107,6 @@ def _get_default_memory_size(cube_config) -> dict[str, int]:
107107
}
108108

109109

110-
def _create_naive_mem_cube() -> NaiveMemCube:
111-
"""Create a NaiveMemCube instance with initialized components."""
112-
naive_mem_cube = NaiveMemCube(
113-
llm=llm,
114-
embedder=embedder,
115-
mem_reader=mem_reader,
116-
graph_db=graph_db,
117-
reranker=reranker,
118-
internet_retriever=internet_retriever,
119-
memory_manager=memory_manager,
120-
default_cube_config=default_cube_config,
121-
)
122-
return naive_mem_cube
123-
124-
125110
def init_server():
126111
"""Initialize server components and configurations."""
127112
# Get default cube configuration
@@ -176,7 +161,17 @@ def init_server():
176161
# Initialize SchedulerAPIModule
177162
api_module = mem_scheduler.api_module
178163

179-
naive_mem_cube = _create_naive_mem_cube()
164+
naive_mem_cube = NaiveMemCube(
165+
llm=llm,
166+
embedder=embedder,
167+
mem_reader=mem_reader,
168+
graph_db=graph_db,
169+
reranker=reranker,
170+
internet_retriever=internet_retriever,
171+
memory_manager=memory_manager,
172+
default_cube_config=default_cube_config,
173+
)
174+
180175
return (
181176
graph_db,
182177
mem_reader,
@@ -433,7 +428,6 @@ def add_memories(add_req: APIADDRequest):
433428
mem_cube_id=add_req.mem_cube_id,
434429
session_id=add_req.session_id or "default_session",
435430
)
436-
naive_mem_cube = _create_naive_mem_cube()
437431
target_session_id = add_req.session_id
438432
if not target_session_id:
439433
target_session_id = "default_session"
@@ -477,7 +471,6 @@ def chat_complete(chat_req: APIChatCompleteRequest):
477471
"""Chat with MemOS for a specific user. Returns complete response (non-streaming)."""
478472
try:
479473
# Collect all responses from the generator
480-
naive_mem_cube = _create_naive_mem_cube()
481474
content, references = mos_server.chat(
482475
query=chat_req.query,
483476
user_id=chat_req.user_id,

0 commit comments

Comments
 (0)