tests: add CI

.github/workflows/test.yml

@@ -0,0 +1,54 @@
+name: test
+
+on:
+  push:
+    branches:
+      - main
+      - dev
+    paths-ignore:
+      - '**/*.md'
+      - '**/*.ipynb'
+  pull_request:
+    branches:
+      - main
+      - dev
+    paths-ignore:
+      - '**/*.md'
+      - '**/*.ipynb'
+
+jobs:
+  test:
+    name: Tests on ${{ matrix.os }} for ${{ matrix.python-version }}
+    strategy:
+      matrix:
+        python-version: [3.9]
+        os: [ubuntu-latest]
+    runs-on: ${{ matrix.os }}
+    timeout-minutes: 10
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -r requirements.txt
+          pip install -r requirements-dev.txt
+      - name: Lint with flake8
+        run: |
+          flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+      - name: Build and Test
+        run: |
+          python -m pytest -o log_cli=true -o log_cli_level="INFO" -v ./
+    #   - name: Check codecov file
+    #     id: check_files
+    #     uses: andstor/file-existence-action@v1
+    #     with:
+    #       files: './coverage.xml'
+    #   - name: Upload coverage from test to Codecov
+    #     uses: codecov/codecov-action@v2
+    #     with:
+    #       file: ./coverage.xml
+    #       token: ${{ secrets.CODECOV_TOKEN }}

nano_graphrag/graphrag.py

@@ -266,4 +266,9 @@ async def _insert_done(self):
         await asyncio.gather(*tasks)
 
     async def _query_done(self):
-        pass
+        tasks = []
+        for storage_inst in [self.llm_response_cache]:
+            if storage_inst is None:
+                continue
+            tasks.append(cast(StorageNameSpace, storage_inst).index_done_callback())
+        await asyncio.gather(*tasks)

requirements-dev.txt

@@ -0,0 +1,3 @@
+sentence_transformers
+flake8
+pytest