feat(llm): expand graph extraction service APIs

.github/workflows/auto-pr-comment.yml

@@ -28,7 +28,6 @@ jobs:
       pull-requests: write
     steps:
       - name: Add review comment
-        # This workflow is only enabled in fork repositories to request AI review.
         uses: peter-evans/create-or-update-comment@v4
         with:
           issue-number: ${{ github.event.pull_request.number }}

.github/workflows/hugegraph-llm.yml

@@ -26,98 +26,34 @@ on:
       - 'release-*'
   pull_request:
 
-permissions:
-  contents: read
-
-env:
-  UV_VERSION: "0.9.22"
-
 jobs:
-  llm-unit-contract:
+  build:
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
         python-version: ["3.10", "3.11"]
 
     steps:
-    - uses: actions/checkout@v6
-      with:
-        persist-credentials: false
-
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v6
-      with:
-        python-version: ${{ matrix.python-version }}
-
-    - name: Install uv
+    - name: Prepare HugeGraph Server Environment
       run: |
-        curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
-        echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-        "$HOME/.local/bin/uv" --version
-
-    - name: Cache dependencies
-      uses: actions/cache@v5
-      with:
-        path: |
-          ~/.cache/uv
-          ~/nltk_data
-        key: ${{ runner.os }}-uv-${{ matrix.python-version }}-${{ hashFiles('**/pyproject.toml', 'uv.lock') }}
-        restore-keys: |
-          ${{ runner.os }}-uv-${{ matrix.python-version }}-
+        docker run -d --name=graph -p 8080:8080 -e PASSWORD=admin hugegraph/hugegraph:1.5.0
+        sleep 10
 
-    - name: Install dependencies
-      run: |
-        uv sync --extra llm --extra dev
-        uv run python -c "import nltk; nltk.download('stopwords'); nltk.download('punkt')"
-
-    - name: Run unit tests
-      env:
-        SKIP_EXTERNAL_SERVICES: true
-      run: |
-        uv run pytest hugegraph-llm/src/tests -m "not integration and not hugegraph and not smoke and not external" --cov=hugegraph_llm --cov-fail-under=34 --cov-report=term --cov-report=xml:llm-unit-contract.xml -v --tb=short --durations=20
-
-    - name: Upload coverage artifact
-      uses: actions/upload-artifact@v7
-      with:
-        name: llm-unit-contract-coverage-${{ matrix.python-version }}
-        path: llm-unit-contract.xml
-        if-no-files-found: error
-
-  llm-hugegraph-boundary:
-    runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        python-version: ["3.10", "3.11"]
-
-    services:
-      hugegraph:
-        image: hugegraph/hugegraph:1.7.0
-        env:
-          PASSWORD: admin
-        options: --health-cmd="curl -f http://localhost:8080/versions || exit 1" --health-interval=10s --health-timeout=5s --health-retries=8
-        ports:
-          - 8080:8080
-
-    steps:
-    - uses: actions/checkout@v6
-      with:
-        persist-credentials: false
+    - uses: actions/checkout@v4
 
     - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v6
+      uses: actions/setup-python@v5
       with:
         python-version: ${{ matrix.python-version }}
 
     - name: Install uv
       run: |
-        curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
-        echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-        "$HOME/.local/bin/uv" --version
+        curl -LsSf https://astral.sh/uv/install.sh | sh
+        echo "$HOME/.cargo/bin" >> $GITHUB_PATH
 
     - name: Cache dependencies
-      uses: actions/cache@v5
+      uses: actions/cache@v4
       with:
         path: |
           ~/.cache/uv
@@ -131,89 +67,16 @@ jobs:
         uv sync --extra llm --extra dev
         uv run python -c "import nltk; nltk.download('stopwords'); nltk.download('punkt')"
 
-    - name: Run HugeGraph boundary tests
+    - name: Run unit tests
+      working-directory: hugegraph-llm
       env:
-        HUGEGRAPH_REQUIRED: true
-        HUGEGRAPH_URL: http://127.0.0.1:8080
-        HUGEGRAPH_GRAPH: hugegraph
-        HUGEGRAPH_USER: admin
-        HUGEGRAPH_PASSWORD: admin
-        SKIP_EXTERNAL_SERVICES: false
-      run: |
-        uv run pytest hugegraph-llm/src/tests -m "integration and hugegraph" -v --tb=short
-
-    - name: Dump HugeGraph diagnostics
-      if: failure()
-      run: |
-        docker ps -a
-        container_id=$(docker ps -aq --filter ancestor=hugegraph/hugegraph:1.7.0 | head -n 1)
-        if [ -n "$container_id" ]; then
-          docker logs "$container_id"
-        fi
-
-  llm-core-smoke:
-    runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        python-version: ["3.10", "3.11"]
-
-    services:
-      hugegraph:
-        image: hugegraph/hugegraph:1.7.0
-        env:
-          PASSWORD: admin
-        options: --health-cmd="curl -f http://localhost:8080/versions || exit 1" --health-interval=10s --health-timeout=5s --health-retries=8
-        ports:
-          - 8080:8080
-
-    steps:
-    - uses: actions/checkout@v6
-      with:
-        persist-credentials: false
-
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v6
-      with:
-        python-version: ${{ matrix.python-version }}
-
-    - name: Install uv
-      run: |
-        curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
-        echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-        "$HOME/.local/bin/uv" --version
-
-    - name: Cache dependencies
-      uses: actions/cache@v5
-      with:
-        path: |
-          ~/.cache/uv
-          ~/nltk_data
-        key: ${{ runner.os }}-uv-${{ matrix.python-version }}-${{ hashFiles('**/pyproject.toml', 'uv.lock') }}
-        restore-keys: |
-          ${{ runner.os }}-uv-${{ matrix.python-version }}-
-
-    - name: Install dependencies
+        SKIP_EXTERNAL_SERVICES: true
       run: |
-        uv sync --extra llm --extra dev
-        uv run python -c "import nltk; nltk.download('stopwords'); nltk.download('punkt')"
+        uv run pytest src/tests/config/ src/tests/document/ src/tests/middleware/ src/tests/operators/ src/tests/models/ src/tests/indices/ src/tests/test_utils.py -v --tb=short
 
-    - name: Run deterministic core smoke tests
+    - name: Run integration tests
+      working-directory: hugegraph-llm
       env:
-        HUGEGRAPH_REQUIRED: true
-        HUGEGRAPH_URL: http://127.0.0.1:8080
-        HUGEGRAPH_GRAPH: hugegraph
-        HUGEGRAPH_USER: admin
-        HUGEGRAPH_PASSWORD: admin
-        SKIP_EXTERNAL_SERVICES: false
-      run: |
-        uv run pytest hugegraph-llm/src/tests/integration -m "smoke" -v --tb=short
-
-    - name: Dump HugeGraph diagnostics
-      if: failure()
+        SKIP_EXTERNAL_SERVICES: true
       run: |
-        docker ps -a
-        container_id=$(docker ps -aq --filter ancestor=hugegraph/hugegraph:1.7.0 | head -n 1)
-        if [ -n "$container_id" ]; then
-          docker logs "$container_id"
-        fi
+        uv run pytest src/tests/integration/test_graph_rag_pipeline.py src/tests/integration/test_kg_construction.py src/tests/integration/test_rag_pipeline.py -v --tb=short

.github/workflows/hugegraph-python-client.yml

@@ -7,96 +7,39 @@ on:
       - 'release-*'
   pull_request:
 
-permissions:
-  contents: read
-
-env:
-  UV_VERSION: "0.9.22"
-
 jobs:
-  client-unit-contract:
-    runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        python-version: ["3.10", "3.11"]
-
-    steps:
-      - uses: actions/checkout@v6
-        with:
-          persist-credentials: false
-
-      - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v6
-        with:
-          python-version: ${{ matrix.python-version }}
-
-      - name: Install uv
-        run: |
-          curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
-          echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-          "$HOME/.local/bin/uv" --version
-
-      - name: Cache dependencies
-        id: cache-deps
-        uses: actions/cache@v5
-        with:
-          path: |
-            ~/.cache/uv
-          key: ${{ runner.os }}-uv-${{ matrix.python-version }}-${{ hashFiles('**/pyproject.toml') }}
-          restore-keys: |
-            ${{ runner.os }}-uv-${{ matrix.python-version }}-
-
-      - name: Install dependencies with uv sync
-        run: |
-          uv sync --extra python-client --extra dev
-
-      - name: Run unit and contract tests
-        run: |
-          uv run pytest hugegraph-python-client/src/tests -m "unit or contract" --cov=pyhugegraph --cov-fail-under=45 --cov-report=term --cov-report=xml:client-unit-contract.xml -v --tb=short --durations=20
-
-      - name: Upload coverage artifact
-        uses: actions/upload-artifact@v7
-        with:
-          name: client-unit-contract-coverage-${{ matrix.python-version }}
-          path: client-unit-contract.xml
-          if-no-files-found: error
-
-  client-hugegraph-integration:
+  build:
     runs-on: ubuntu-latest
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     services:
       hugegraph:
         image: hugegraph/hugegraph:1.7.0
         env:
           PASSWORD: admin
-        options: --health-cmd="curl -f http://localhost:8080/versions || exit 1" --health-interval=10s --health-timeout=5s --health-retries=8
+        options: --health-cmd="curl -f http://localhost:8080/versions || exit 1" --health-interval=10s --health-timeout=5s --health-retries=5
         ports:
           - 8080:8080
 
     steps:
-      - uses: actions/checkout@v6
-        with:
-          persist-credentials: false
+      - uses: actions/checkout@v4
 
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v6
+        uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
 
       - name: Install uv
         run: |
-          curl -LsSf "https://astral.sh/uv/${UV_VERSION}/install.sh" | sh
-          echo "$HOME/.local/bin" >> "$GITHUB_PATH"
-          "$HOME/.local/bin/uv" --version
+          curl -LsSf https://astral.sh/uv/install.sh | sh
+          echo "$HOME/.cargo/bin" >> $GITHUB_PATH
 
       - name: Cache dependencies
         id: cache-deps
-        uses: actions/cache@v5
+        uses: actions/cache@v4
         with:
           path: |
             ~/.cache/uv
@@ -108,21 +51,13 @@ jobs:
         run: |
           uv sync --extra python-client --extra dev
 
-      - name: Run HugeGraph integration tests
-        env:
-          HUGEGRAPH_REQUIRED: true
-          HUGEGRAPH_URL: http://127.0.0.1:8080
-          HUGEGRAPH_GRAPH: hugegraph
-          HUGEGRAPH_USER: admin
-          HUGEGRAPH_PASSWORD: admin
+      - name: Test example
         run: |
-          uv run pytest hugegraph-python-client/src/tests -m "integration and hugegraph" -v --tb=short
+          ls -al
+          export PYTHONPATH=$(pwd)/hugegraph-python-client/src && echo ${PYTHONPATH}
+          uv run python hugegraph-python-client/src/pyhugegraph/example/hugegraph_example.py
 
-      - name: Dump HugeGraph diagnostics
-        if: failure()
+      - name: Test with pytest
         run: |
-          docker ps -a
-          container_id=$(docker ps -aq --filter ancestor=hugegraph/hugegraph:1.7.0 | head -n 1)
-          if [ -n "$container_id" ]; then
-            docker logs "$container_id"
-          fi
+          uv run pytest
+        working-directory: hugegraph-python-client

.github/workflows/ruff.yml

@@ -49,43 +49,3 @@ jobs:
       - name: Lint code with Ruff
         run: |
           uv run ruff check .
-
-  ty-check:
-    runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        python-version: ["3.10", "3.11"]   # hugegraph-llm requires <3.12
-
-    steps:
-      - uses: actions/checkout@v4
-
-      - name: Setup Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v5
-        with:
-          python-version: ${{ matrix.python-version }}
-
-      - name: Install uv
-        run: |
-          curl -LsSf https://astral.sh/uv/install.sh | sh
-          echo "$HOME/.cargo/bin" >> $GITHUB_PATH
-
-      - name: Cache dependencies
-        id: cache-deps
-        uses: actions/cache@v4
-        with:
-          path: |
-            ~/.cache/uv
-            ~/.cache/pip
-          key: ${{ runner.os }}-uv-${{ matrix.python-version }}-${{ hashFiles('**/pyproject.toml') }}
-          restore-keys: |
-            ${{ runner.os }}-uv-${{ matrix.python-version }}-
-
-      - name: Install dev + runtime extras
-        run: |
-          uv sync --extra dev --extra llm --extra python-client
-
-      - name: Type check with ty (non-blocking)
-        run: uv run ty check hugegraph-llm/src hugegraph-python-client/src
-        continue-on-error: true
-        # TODO: extend scope to hugegraph-ml once heavy optional deps (DGL, PyTorch) are handled
-        # TODO: add graph-mcp once the module is introduced