Merge branch 'feat/evaluation_doc_qa' of github.com:CaralHsi/MemOSRealPublic into feat/evaluation_doc_qa

Author: CaralHsi

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):

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)