diff --git a/.github/actions/cognee_setup/action.yml b/.github/actions/cognee_setup/action.yml index e79619221..869582585 100644 --- a/.github/actions/cognee_setup/action.yml +++ b/.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 diff --git a/.github/workflows/db_examples_tests.yml b/.github/workflows/db_examples_tests.yml index 552395e6a..ae61d2ab0 100644 --- a/.github/workflows/db_examples_tests.yml +++ b/.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: diff --git a/.github/workflows/e2e_tests.yml b/.github/workflows/e2e_tests.yml index 4f7c67314..8caeca5c3 100644 --- a/.github/workflows/e2e_tests.yml +++ b/.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: diff --git a/.github/workflows/examples_tests.yml b/.github/workflows/examples_tests.yml index f0ae430ec..277ed8c21 100644 --- a/.github/workflows/examples_tests.yml +++ b/.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: diff --git a/.github/workflows/graph_db_tests.yml b/.github/workflows/graph_db_tests.yml index 91a611168..fe8fec4f0 100644 --- a/.github/workflows/graph_db_tests.yml +++ b/.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' diff --git a/.github/workflows/relational_db_migration_tests.yml b/.github/workflows/relational_db_migration_tests.yml index 604964512..26fd7e150 100644 --- a/.github/workflows/relational_db_migration_tests.yml +++ b/.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: diff --git a/.github/workflows/reusable_notebook.yml b/.github/workflows/reusable_notebook.yml index 7ee7f61f3..8eaf7ce3a 100644 --- a/.github/workflows/reusable_notebook.yml +++ b/.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: diff --git a/.github/workflows/search_db_tests.yml b/.github/workflows/search_db_tests.yml index a2ab00754..c468bd1c6 100644 --- a/.github/workflows/search_db_tests.yml +++ b/.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: diff --git a/.github/workflows/vector_db_tests.yml b/.github/workflows/vector_db_tests.yml index faa570868..a82194db3 100644 --- a/.github/workflows/vector_db_tests.yml +++ b/.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: