From 3ae2043e7b71614d7f488bd34850bba9ae6e52df Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20MANSUY?= Date: Thu, 4 Dec 2025 19:18:14 +0800 Subject: [PATCH] cherry-pick fdcb034d --- lightrag/kg/postgres_impl.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index 8fc9d590..4e6658b9 100644 --- a/lightrag/kg/postgres_impl.py +++ b/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