fix: uv uninstalls rest of packages in some workflows

.github/actions/cognee_setup/action.yml

@@ -6,6 +6,10 @@ inputs:
     description: "Which Python version to use"
     required: false
     default: "3.11.x"
+  extra-dependencies:
+    description: "Additional extra dependencies to install (space-separated)"
+    required: false
+    default: ""
 
 runs:
   using: "composite"
@@ -23,4 +27,12 @@ runs:
 
     - name: Install dependencies
       shell: bash
-      run: uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra neo4j
+      run: |
+        EXTRA_ARGS=""
+        if [ -n "${{ inputs.extra-dependencies }}" ]; then
+          IFS=' ' read -r -a deps <<< "${{ inputs.extra-dependencies }}"
+          for extra in "${deps[@]}"; do
+            EXTRA_ARGS="$EXTRA_ARGS --extra $extra"
+          done
+        fi
+        uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra neo4j $EXTRA_ARGS

.github/workflows/db_examples_tests.yml

@@ -54,10 +54,6 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
-      - name: Install Neo4j extra
-        run: |
-          uv sync --extra neo4j
-
       - name: Run Neo4j Example
         env:
           ENV: dev
@@ -138,10 +134,7 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: ${{ inputs.python-version }}
-
-      - name: Install PGVector extra
-        run: |
-          uv sync --extra postgres
+          extra-dependencies: "postgres"
 
       - name: Run PGVector Example
         env:

.github/workflows/e2e_tests.yml

@@ -140,10 +140,7 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: '3.11.x'
-
-      - name: Install specific db dependency
-        run: |
-          uv sync --extra postgres
+          extra-dependencies: "postgres"
 
       - name: Run Deduplication Example
         env:
@@ -192,10 +189,10 @@ jobs:
           uses: ./.github/actions/cognee_setup
           with:
             python-version: '3.11.x'
+            extra-dependencies: "aws"
 
-        - name: Install specific S3 dependency
-          run: |
-            uv sync --extra aws
+        - name: Dependencies already installed
+          run: echo "Dependencies already installed in setup"
 
         - name: Run S3 Bucket Test
           env:

.github/workflows/examples_tests.yml

@@ -33,10 +33,7 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: '3.11.x'
-
-      - name: Install specific eval dependency
-        run: |
-          uv sync --extra deepeval
+          extra-dependencies: "deepeval"
 
       - name: Run Evaluation Framework Example
         env:

.github/workflows/graph_db_tests.yml

@@ -72,10 +72,6 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
-      - name: Install specific db dependency
-        run: |
-          uv sync --extra neo4j
-
       - name: Run default Neo4j
         env:
           ENV: 'dev'

.github/workflows/relational_db_migration_tests.yml

@@ -58,10 +58,10 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: '3.11.x'
+          extra-dependencies: "postgres"
 
       - name: Install specific db dependency
-        run: |
-          uv sync --extra postgres
+        run: echo "Dependencies already installed in setup"
 
       - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql)
         env:
@@ -120,10 +120,10 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: '3.11.x'
+          extra-dependencies: "postgres"
 
       - name: Install specific db dependency
-        run: |
-          uv sync --extra postgres
+        run: echo "Dependencies already installed in setup"
 
       - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql)
         env:
@@ -184,10 +184,10 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: '3.11.x'
+          extra-dependencies: "postgres"
 
       - name: Install specific db dependency
-        run: |
-          uv sync --extra postgres --extra neo4j
+        run: echo "Dependencies already installed in setup"
 
       - name: Run PostgreSQL Script to create test data (Chinook_PostgreSql.sql)
         env:

.github/workflows/reusable_notebook.yml

@@ -46,10 +46,7 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: ${{ inputs.python-version }}
-
-      - name: Install specific db dependency
-        run: |
-          uv sync --extra notebook
+          extra-dependencies: "notebook"
 
       - name: Execute Jupyter Notebook
         env:

.github/workflows/search_db_tests.yml

@@ -78,9 +78,8 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
-      - name: Install specific db dependency
-        run: |
-          uv sync --extra neo4j
+      - name: Dependencies already installed
+        run: echo "Dependencies already installed in setup"
 
       - name: Run Neo4j search Tests
         env:
@@ -129,9 +128,10 @@ jobs:
           uses: ./.github/actions/cognee_setup
           with:
             python-version: ${{ inputs.python-version }}
+            extra-dependencies: "postgres"
 
-        - name: Install dependencies
-          run: uv sync --extra postgres
+        - name: Dependencies already installed
+          run: echo "Dependencies already installed in setup"
 
         - name: Run Kuzu/PGVector/Postgres Tests
           env:
@@ -195,10 +195,10 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: ${{ inputs.python-version }}
+          extra-dependencies: "postgres"
 
-      - name: Install dependencies
-        run: |
-          uv sync --extra neo4j --extra postgres
+      - name: Dependencies already installed
+        run: echo "Dependencies already installed in setup"
 
       - name: Run Neo4j + PGVector + Postgres search Tests
         env:

.github/workflows/vector_db_tests.yml

@@ -87,10 +87,7 @@ jobs:
         uses: ./.github/actions/cognee_setup
         with:
           python-version: ${{ inputs.python-version }}
-
-      - name: Install specific db dependency
-        run: |
-          uv sync --extra postgres
+          extra-dependencies: "postgres"
 
       - name: Run PGVector Tests
         env: