attocode-code-intel 0.1.3__tar.gz → 0.1.4__tar.gz

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: attocode-code-intel
-Version: 0.1.3
+Version: 0.1.4
 Summary: Local code search and navigation for coding agents and people
 Project-URL: Homepage, https://github.com/eren23/attocode
 Project-URL: Repository, https://github.com/eren23/attocode

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "attocode-code-intel"
-version = "0.1.3"
+version = "0.1.4"
 description = "Local code search and navigation for coding agents and people"
 readme = "README.md"
 requires-python = ">=3.12"

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/src/attocode_intel/__init__.py

@@ -49,4 +49,4 @@ def __getattr__(name: str) -> Any:
     raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
 
 # The other copy is packages/code-intel/pyproject.toml. A test checks that both match.
-__version__ = "0.1.3"
+__version__ = "0.1.4"

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/src/attocode_intel/_internal/integrations/context/index_store.py

@@ -581,9 +581,11 @@ class IndexStore:
         return float(val) if val else None
 
     def close(self) -> None:
-        if self._conn:
-            self._conn.close()
-            self._conn = None
+        # Background hydration can be inside a write. Closing under it crashes sqlite3.
+        with self._lock:
+            if self._conn:
+                self._conn.close()
+                self._conn = None
 
     def __del__(self) -> None:
         self.close()

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/src/attocode_intel/_internal/integrations/context/vector_store.py

@@ -383,9 +383,10 @@ class VectorStore:
         """
         conn = self._get_conn()
         try:
-            row = conn.execute(
-                "SELECT value FROM store_metadata WHERE key = 'rotation_state'"
-            ).fetchone()
+            with self._lock:
+                row = conn.execute(
+                    "SELECT value FROM store_metadata WHERE key = 'rotation_state'"
+                ).fetchone()
         except sqlite3.OperationalError:
             # Pre-v2 store without a store_metadata row yet — treat as
             # no rotation.
@@ -413,9 +414,10 @@ class VectorStore:
         """
         conn = self._get_conn()
         try:
-            row = conn.execute(
-                "SELECT value FROM store_metadata WHERE key = '_rotator_cache_ver'"
-            ).fetchone()
+            with self._lock:
+                row = conn.execute(
+                    "SELECT value FROM store_metadata WHERE key = '_rotator_cache_ver'"
+                ).fetchone()
         except sqlite3.OperationalError:
             return
         if row is None:
@@ -787,9 +789,11 @@ class VectorStore:
 
     def close(self) -> None:
         """Close the database connection."""
-        if self._conn:
-            self._conn.close()
-            self._conn = None
+        # The background indexer can be inside a write. Closing under it crashes sqlite3.
+        with self._lock:
+            if self._conn:
+                self._conn.close()
+                self._conn = None
 
     def __del__(self) -> None:
         self.close()

{attocode_code_intel-0.1.3 → attocode_code_intel-0.1.4}/tests/test_navigation_evidence.py

@@ -198,6 +198,25 @@ def test_index_lookups_wait_for_a_background_writer():
         assert not reader.is_alive()
 
 
+
+@pytest.mark.parametrize("kind", ["index", "vector"])
+def test_close_waits_for_a_background_write(tmp_path, kind):
+    """Shutdown closed the store while hydration wrote to it, and sqlite3 crashed (R5)."""
+    from attocode_intel._internal.integrations.context.index_store import IndexStore
+    from attocode_intel._internal.integrations.context.vector_store import VectorStore
+
+    db = str(tmp_path / "store.db")
+    store = IndexStore(db_path=db) if kind == "index" else VectorStore(db_path=db, dimension=4)
+    closer = threading.Thread(target=store.close, daemon=True)
+    with store._lock:  # A writer is inside a statement.
+        closer.start()
+        closer.join(0.2)
+        assert closer.is_alive()
+        assert store._conn is not None
+    closer.join(5)
+    assert not closer.is_alive()
+    assert store._conn is None
+
 def test_http_indexer_resolves_relative_imports_and_import_roots():
     from attocode_intel.indexing.full_indexer import _resolve_import_path