cherry-pick fdcb034d

lightrag/kg/postgres_impl.py

@@ -77,6 +77,9 @@ class PostgreSQLDB:
         # Server settings
         self.server_settings = config.get("server_settings")
 
+        # Statement LRU cache size
+        self.statement_cache_size = int(config.get("statement_cache_size"))
+
         if self.user is None or self.password is None or self.database is None:
             raise ValueError("Missing database user, password, or database")
 
@@ -161,9 +164,13 @@ class PostgreSQLDB:
                 "port": self.port,
                 "min_size": 1,
                 "max_size": self.max,
-                "statement_cache_size": 0,
+                "statement_cache_size": self.statement_cache_size,
             }
 
+            logger.info(
+                f"PostgreSQL, statement LRU cache size set as: {self.statement_cache_size}"
+            )
+
             # Add SSL configuration if provided
             ssl_context = self._create_ssl_context()
             if ssl_context is not None:
@@ -1392,9 +1399,13 @@ class ClientManager:
             ),
             # Server settings for Supabase
             "server_settings": os.environ.get(
-                "POSTGRES_SERVER_OPTIONS",
+                "POSTGRES_SERVER_SETTINGS",
                 config.get("postgres", "server_options", fallback=None),
             ),
+            "statement_cache_size": os.environ.get(
+                "POSTGRES_STATEMENT_CACHE_SIZE",
+                config.get("postgres", "statement_cache_size", fallback=None),
+            ),
         }
 
     @classmethod