diff --git a/src/memos/api/handlers/component_init.py b/src/memos/api/handlers/component_init.py index aa2525878..fc8ce311f 100644 --- a/src/memos/api/handlers/component_init.py +++ b/src/memos/api/handlers/component_init.py @@ -233,7 +233,7 @@ def init_server() -> dict[str, Any]: searcher: Searcher = tree_mem.get_searcher( manual_close_internet=os.getenv("ENABLE_INTERNET", "true").lower() == "false", moscube=False, - process_llm=mem_reader.llm, + process_llm=mem_reader.general_llm, ) logger.debug("Searcher created") @@ -260,7 +260,7 @@ def init_server() -> dict[str, Any]: mem_scheduler: OptimizedScheduler = SchedulerFactory.from_config(scheduler_config) mem_scheduler.initialize_modules( chat_llm=llm, - process_llm=mem_reader.llm, + process_llm=mem_reader.general_llm, db_engine=BaseDBManager.create_default_sqlite_engine(), mem_reader=mem_reader, redis_client=redis_client, diff --git a/src/memos/mem_os/core.py b/src/memos/mem_os/core.py index 22cd0e9cb..54f8f01e0 100644 --- a/src/memos/mem_os/core.py +++ b/src/memos/mem_os/core.py @@ -132,7 +132,7 @@ def _initialize_mem_scheduler(self) -> GeneralScheduler: # Configure scheduler general_modules self._mem_scheduler.initialize_modules( chat_llm=self.chat_llm, - process_llm=self.mem_reader.llm, + process_llm=self.mem_reader.general_llm, db_engine=self.user_manager.engine, ) self._mem_scheduler.start() diff --git a/src/memos/mem_scheduler/general_modules/init_components_for_scheduler.py b/src/memos/mem_scheduler/general_modules/init_components_for_scheduler.py index 8777b9f2e..ec431c253 100644 --- a/src/memos/mem_scheduler/general_modules/init_components_for_scheduler.py +++ b/src/memos/mem_scheduler/general_modules/init_components_for_scheduler.py @@ -305,7 +305,7 @@ def init_components() -> dict[str, Any]: searcher: Searcher = tree_mem.get_searcher( manual_close_internet=os.getenv("ENABLE_INTERNET", "true").lower() == "false", moscube=False, - process_llm=mem_reader.llm, + process_llm=mem_reader.general_llm, ) # Initialize feedback server feedback_server = SimpleMemFeedback(