fix(seg-fault): attempt

Author: Daniele Briggi

.devcontainer/devcontainer.json

@@ -1,18 +1,6 @@
 {
     "name": "Python 3.10",
     "image": "mcr.microsoft.com/devcontainers/python:3.10",
-    "runArgs": [
-        "--runtime",
-        "nvidia",
-        "--gpus",
-        "all",
-        // optional but make sure CUDA workloads are available
-        "--env",
-        "NVIDIA_VISIBLE_DEVICES=all",
-        // optional but make sure CUDA workloads are available
-        "--env",
-        "NVIDIA_DRIVER_CAPABILITIES=compute,utility"
-    ],
     "customizations": {
         "vscode": {
             "extensions": [
@@ -26,13 +14,5 @@
                 "hbenl.vscode-test-explorer"
             ]
         }
-    },
-    "hostRequirements": {
-        "gpu": "optional"
-    },
-    "remoteEnv": {
-        // optional but make sure CUDA workloads are available
-        "NVIDIA_VISIBLE_DEVICES": "all",
-        "NVIDIA_DRIVER_CAPABILITIES": "compute,utility"
     }
 }

.devcontainer/py3.11/devcontainer.json

@@ -0,0 +1,38 @@
+{
+    "name": "Python 3.11",
+    "image": "mcr.microsoft.com/devcontainers/python:3.11",
+    "runArgs": [
+        "--runtime",
+        "nvidia",
+        "--gpus",
+        "all",
+        // optional but make sure CUDA workloads are available
+        "--env",
+        "NVIDIA_VISIBLE_DEVICES=all",
+        // optional but make sure CUDA workloads are available
+        "--env",
+        "NVIDIA_DRIVER_CAPABILITIES=compute,utility"
+    ],
+    "customizations": {
+        "vscode": {
+            "extensions": [
+                "ms-python.black-formatter",
+                "ms-python.flake8",
+                "ms-python.isort",
+                "ms-python.vscode-pylance",
+                "ms-python.python",
+                "ms-python.debugpy",
+                "ms-python.vscode-python-envs",
+                "hbenl.vscode-test-explorer"
+            ]
+        }
+    },
+    "hostRequirements": {
+        "gpu": "optional"
+    },
+    "remoteEnv": {
+        // optional but make sure CUDA workloads are available
+        "NVIDIA_VISIBLE_DEVICES": "all",
+        "NVIDIA_DRIVER_CAPABILITIES": "compute,utility"
+    }
+}

.github/workflows/test.yaml

@@ -49,6 +49,10 @@ jobs:
 
     - name: Test
       # Using default directory for models
+      # COVERAGE_CORE=pytrace: Workaround for Python 3.11 segfault with SQLite extensions + C tracer
+      # See: https://github.com/nedbat/coveragepy/issues/1665
+      env:
+        COVERAGE_CORE: ${{ matrix.python-version == '3.11' && 'pytrace' || '' }}
       run: |
         pytest --cov --cov-branch --cov-report=xml -v -m "not slow" ./tests
 

src/sqlite_rag/models/document_result.py

@@ -42,8 +42,9 @@ def get_preview(
             return self.chunk_content[:max_chars]
 
         # Sort by start_offset to maintain document order
-        top_sentences.sort(
-            key=lambda s: s.start_offset if s.start_offset is not None else -1
+        top_sentences = sorted(
+            top_sentences,
+            key=lambda s: s.start_offset if s.start_offset is not None else -1,
         )
 
         preview_parts = []

tests/conftest.py

@@ -1,5 +1,6 @@
 import sqlite3
 import tempfile
+from collections.abc import Generator
 
 import pytest
 
@@ -26,7 +27,7 @@ def db_conn():
 
 
 @pytest.fixture
-def engine(db_conn) -> Engine:
+def engine(db_conn) -> Generator[Engine, None, None]:
     conn, settings = db_conn
 
     engine = Engine(
@@ -39,4 +40,11 @@ def engine(db_conn) -> Engine:
     engine.quantize()
     engine.create_new_context()
 
-    return engine
+    yield engine
+
+    # Cleanup resources to prevent segfaults in Python 3.11
+    # Must explicitly free resources before garbage collection
+    try:
+        engine.close()
+    except Exception:
+        pass