coderag-ai 0.3.1__tar.gz → 0.3.2__tar.gz

{coderag_ai-0.3.1 → coderag_ai-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: coderag-ai
-Version: 0.3.1
+Version: 0.3.2
 Summary: Token-efficient, structure-aware Code Intelligence + RAG for private source repositories.
 Project-URL: Homepage, https://github.com/galacticsurfer/coderag
 Project-URL: Repository, https://github.com/galacticsurfer/coderag

{coderag_ai-0.3.1 → coderag_ai-0.3.2}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "hatchling.build"
 # command are both `coderag`. Plain "coderag" is taken on PyPI by an unrelated project,
 # and "code-rag" is rejected as too similar (PyPI collapses -/_/. when comparing names).
 name = "coderag-ai"
-version = "0.3.1"
+version = "0.3.2"
 description = "Token-efficient, structure-aware Code Intelligence + RAG for private source repositories."
 readme = "README.md"
 requires-python = ">=3.12"

{coderag_ai-0.3.1 → coderag_ai-0.3.2}/src/coderag/localdb.py

@@ -49,8 +49,27 @@ def start(pgdata: Path | str | None = None) -> str:
     path.parent.mkdir(parents=True, exist_ok=True)
     # cleanup_mode=None -> the postmaster keeps running after we exit.
     server = pgserver.get_server(str(path), cleanup_mode=None)
-    server.psql("CREATE EXTENSION IF NOT EXISTS vector;")
-    return sqlalchemy_url(server.get_uri())
+    url = sqlalchemy_url(server.get_uri())
+    _ensure_vector_extension(url)
+    return url
+
+
+def _ensure_vector_extension(url: str) -> None:
+    """Create the pgvector extension over a normal connection.
+
+    Deliberately does NOT use pgserver's bundled ``psql`` binary: on macOS that
+    executable can fail to run (exit 127), which would break `localdb start` even
+    though the server itself is healthy. psycopg is already a dependency, so we
+    talk to the server the same way the rest of the application does.
+    """
+    from sqlalchemy import create_engine, text
+
+    engine = create_engine(url)
+    try:
+        with engine.begin() as conn:
+            conn.execute(text("CREATE EXTENSION IF NOT EXISTS vector"))
+    finally:
+        engine.dispose()
 
 
 def stop(pgdata: Path | str | None = None, timeout: float = 15.0) -> None:

coderag_ai-0.3.2/tests/test_phase14_localdb.py

@@ -0,0 +1,72 @@
+"""Rootless local Postgres helper (no Docker/sudo) — lifecycle correctness."""
+
+from __future__ import annotations
+
+import pytest
+
+pytest.importorskip("pgserver")
+
+from coderag import localdb  # noqa: E402
+
+
+def test_status_does_not_start_the_server(tmp_path):
+    """A status check must never start a server (regression: it used to)."""
+    pgdata = tmp_path / "pgdata"
+    assert localdb.status(pgdata) is None       # not running, nothing created
+    assert not (pgdata / "postmaster.pid").exists()
+
+
+def test_start_status_stop_lifecycle(tmp_path):
+    pgdata = tmp_path / "pgdata"
+    url = localdb.start(pgdata)
+    try:
+        assert url.startswith("postgresql+psycopg://")
+        pid = localdb.status(pgdata)
+        assert pid and pid > 0
+    finally:
+        localdb.stop(pgdata)
+    assert localdb.status(pgdata) is None       # stays stopped
+
+
+def test_stale_pidfile_reports_not_running(tmp_path):
+    pgdata = tmp_path / "pgdata"
+    pgdata.mkdir()
+    (pgdata / "postmaster.pid").write_text("999999999\n")  # impossible pid
+    assert localdb.status(pgdata) is None
+
+
+def test_start_does_not_shell_out_to_psql(tmp_path, monkeypatch):
+    """Regression: the bundled `psql` binary can fail on macOS (exit 127).
+
+    `start()` must create the pgvector extension over a normal DB connection, so
+    the command works even when that executable is unusable.
+    """
+    import pgserver
+
+    real_get_server = pgserver.get_server
+
+    def guarded(pgdata, cleanup_mode=None):
+        server = real_get_server(pgdata, cleanup_mode=cleanup_mode)
+
+        def boom(*a, **k):
+            raise AssertionError("start() must not invoke the bundled psql binary")
+
+        monkeypatch.setattr(server, "psql", boom, raising=False)
+        return server
+
+    monkeypatch.setattr(pgserver, "get_server", guarded)
+
+    pgdata = tmp_path / "pgdata"
+    url = localdb.start(pgdata)
+    try:
+        from sqlalchemy import create_engine, text
+
+        engine = create_engine(url)
+        with engine.connect() as conn:
+            got = conn.execute(
+                text("SELECT extversion FROM pg_extension WHERE extname='vector'")
+            ).scalar()
+        engine.dispose()
+        assert got, "pgvector extension should have been created"
+    finally:
+        localdb.stop(pgdata)

coderag_ai-0.3.1/tests/test_phase14_localdb.py

@@ -1,35 +0,0 @@
-"""Rootless local Postgres helper (no Docker/sudo) — lifecycle correctness."""
-
-from __future__ import annotations
-
-import pytest
-
-pytest.importorskip("pgserver")
-
-from coderag import localdb  # noqa: E402
-
-
-def test_status_does_not_start_the_server(tmp_path):
-    """A status check must never start a server (regression: it used to)."""
-    pgdata = tmp_path / "pgdata"
-    assert localdb.status(pgdata) is None       # not running, nothing created
-    assert not (pgdata / "postmaster.pid").exists()
-
-
-def test_start_status_stop_lifecycle(tmp_path):
-    pgdata = tmp_path / "pgdata"
-    url = localdb.start(pgdata)
-    try:
-        assert url.startswith("postgresql+psycopg://")
-        pid = localdb.status(pgdata)
-        assert pid and pid > 0
-    finally:
-        localdb.stop(pgdata)
-    assert localdb.status(pgdata) is None       # stays stopped
-
-
-def test_stale_pidfile_reports_not_running(tmp_path):
-    pgdata = tmp_path / "pgdata"
-    pgdata.mkdir()
-    (pgdata / "postmaster.pid").write_text("999999999\n")  # impossible pid
-    assert localdb.status(pgdata) is None