@@ -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-
125110def 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