From 47fd7ea10e7c94a6b1e25a4b2012d44ce4abf288 Mon Sep 17 00:00:00 2001 From: BukeLy Date: Thu, 20 Nov 2025 00:02:26 +0800 Subject: [PATCH] fix: add required connection retry configs to E2E tests Add missing connection retry configuration parameters: - connection_retry_attempts: 3 - connection_retry_backoff: 0.5 - connection_retry_backoff_max: 5.0 - pool_close_timeout: 5.0 These are required by PostgreSQLDB initialization. Issue: KeyError: 'connection_retry_attempts' in E2E tests --- tests/test_e2e_postgres_migration.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_e2e_postgres_migration.py b/tests/test_e2e_postgres_migration.py index 24e8b9c5..c6065d78 100644 --- a/tests/test_e2e_postgres_migration.py +++ b/tests/test_e2e_postgres_migration.py @@ -31,6 +31,11 @@ def pg_config(): "database": os.getenv("POSTGRES_DB", "lightrag_test"), "workspace": os.getenv("POSTGRES_WORKSPACE", "e2e_test"), "max_connections": 10, + # Connection retry configuration + "connection_retry_attempts": 3, + "connection_retry_backoff": 0.5, + "connection_retry_backoff_max": 5.0, + "pool_close_timeout": 5.0, }