fix: wait for postgres to be ready before tests

.github/workflows/test_common.yml

@@ -98,6 +98,12 @@ jobs:
           mkdir .cognee_system
           echo $(pwd)/.cognee_system
 
+      - name: Wait for PostgreSQL to be ready
+        run: |
+          echo "Waiting for PostgreSQL to be ready..."
+          until pg_isready -h localhost -p 5432 -U cognee; do
+            sleep 1
+          done
 
       - name: Run tests
         run: poetry run pytest tests/

.github/workflows/test_neo4j.yml

@@ -63,13 +63,6 @@ jobs:
           virtualenvs-in-project: true
           installer-parallel: true
 
-      - name: Wait for PostgreSQL to be ready
-        run: |
-          echo "Waiting for PostgreSQL to be ready..."
-          until pg_isready -h localhost -p 5432 -U cognee; do
-            sleep 1
-          done
-
       - name: Install dependencies
         run: poetry install --no-interaction
 
@@ -78,6 +71,13 @@ jobs:
           mkdir .cognee_system
           echo $(pwd)/.cognee_system
 
+      - name: Wait for PostgreSQL to be ready
+        run: |
+          echo "Waiting for PostgreSQL to be ready..."
+          until pg_isready -h localhost -p 5432 -U cognee; do
+            sleep 1
+          done
+
       - name: Run default Neo4j
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}

.github/workflows/test_qdrant.yml

@@ -71,6 +71,13 @@ jobs:
           mkdir .cognee_system
           echo $(pwd)/.cognee_system
 
+      - name: Wait for PostgreSQL to be ready
+        run: |
+          echo "Waiting for PostgreSQL to be ready..."
+          until pg_isready -h localhost -p 5432 -U cognee; do
+            sleep 1
+          done
+
       - name: Run default Qdrant
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}

.github/workflows/test_weaviate.yml

@@ -71,6 +71,12 @@ jobs:
           mkdir .cognee_system
           echo $(pwd)/.cognee_system
 
+      - name: Wait for PostgreSQL to be ready
+        run: |
+          echo "Waiting for PostgreSQL to be ready..."
+          until pg_isready -h localhost -p 5432 -U cognee; do
+            sleep 1
+
       - name: Run default Weaviate
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}

cognee/infrastructure/databases/graph/neo4j_driver/adapter.py

@@ -21,7 +21,8 @@ class Neo4jAdapter(GraphDBInterface):
     ):
         self.driver = driver or AsyncGraphDatabase.driver(
             graph_database_url,
-            auth = (graph_database_username, graph_database_password)
+            auth = (graph_database_username, graph_database_password),
+            max_connection_lifetime = 120
         )
 
     async def close(self) -> None: