diff --git a/.github/workflows/contrib-graph-rag-tests.yml b/.github/workflows/contrib-graph-rag-tests.yml index 5733d3365..dd9c9d8a1 100644 --- a/.github/workflows/contrib-graph-rag-tests.yml +++ b/.github/workflows/contrib-graph-rag-tests.yml @@ -95,7 +95,7 @@ jobs: uv pip install --system pytest - name: Install Neo4j and Llama-index when on linux run: | - uv pip install --system -e .[neo4j,test] + uv pip install --system -e .[neo4j_graph_rag,test] - name: Set AUTOGEN_USE_DOCKER based on OS shell: bash run: | @@ -109,7 +109,6 @@ jobs: run: | uv pip install --system pytest-cov>=5 bash scripts/test-skip-llm.sh test/agentchat/contrib/graph_rag/test_neo4j_graph_rag.py - bash scripts/test.sh -m "neo4j" - name: Upload coverage to Codecov uses: codecov/codecov-action@v3 with: diff --git a/.github/workflows/test-with-optional-deps.yml b/.github/workflows/test-with-optional-deps.yml index 17916dbc6..b8aeebabe 100644 --- a/.github/workflows/test-with-optional-deps.yml +++ b/.github/workflows/test-with-optional-deps.yml @@ -52,6 +52,14 @@ jobs: runs-on: ${{ matrix.os }} env: AUTOGEN_USE_DOCKER: ${{ matrix.os != 'ubuntu-latest' && 'False' }} + services: + neo4j: + image: neo4j:latest + ports: + - 7687:7687 + - 7474:7474 + env: + NEO4J_AUTH: neo4j/password strategy: fail-fast: false matrix: @@ -66,7 +74,7 @@ jobs: - "retrievechat-mongodb" - "retrievechat-qdrant" - "graph-rag-falkor-db" - # - "neo4j" + - "neo4j" - "twilio" - "interop" # exclude: