diff --git a/src/memos/graph_dbs/polardb.py b/src/memos/graph_dbs/polardb.py index 603adbd7d..fd0c0d4dd 100644 --- a/src/memos/graph_dbs/polardb.py +++ b/src/memos/graph_dbs/polardb.py @@ -1,6 +1,7 @@ import json import random import textwrap +import time from contextlib import suppress from datetime import datetime @@ -152,7 +153,7 @@ def __init__(self, config: PolarDBGraphDBConfig): # Create connection pool self.connection_pool = psycopg2.pool.ThreadedConnectionPool( minconn=5, - maxconn=2000, + maxconn=100, host=host, port=port, user=user, @@ -277,6 +278,8 @@ def _get_connection(self): if attempt >= max_retries - 1: raise RuntimeError(f"Failed to get a valid connection from pool: {e}") from e + else: + time.sleep(0.1) continue def _return_connection(self, connection): diff --git a/src/memos/mem_scheduler/base_scheduler.py b/src/memos/mem_scheduler/base_scheduler.py index d628b10a8..bc218172e 100644 --- a/src/memos/mem_scheduler/base_scheduler.py +++ b/src/memos/mem_scheduler/base_scheduler.py @@ -180,13 +180,6 @@ def __init__(self, config: BaseSchedulerConfig): self.current_user_id: UserID | str | None = None self.current_mem_cube_id: MemCubeID | str | None = None self.current_mem_cube: BaseMemCube | None = None - try: - self.components = init_components() - self.current_mem_cube: BaseMemCube = self.components["naive_mem_cube"] - except Exception: - logger.info( - "No environment available to initialize mem cube. Using fallback naive_mem_cube." - ) self._mem_cubes: dict[str, BaseMemCube] = {} self.auth_config_path: str | Path | None = self.config.get("auth_config_path", None)