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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: coderag-ai
-Version: 0.3.2
+Version: 0.3.3
 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.2 → coderag_ai-0.3.3}/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.2"
+version = "0.3.3"
 description = "Token-efficient, structure-aware Code Intelligence + RAG for private source repositories."
 readme = "README.md"
 requires-python = ">=3.12"

{coderag_ai-0.3.2 → coderag_ai-0.3.3}/src/coderag/cli/main.py

@@ -45,19 +45,17 @@ def localdb_start(
         raise typer.Exit(1) from None
     console.print(f"[green]PostgreSQL running.[/]\n\n  export CODERAG_DATABASE_URL='{url}'\n")
     if migrate:
-        import os
-        import subprocess
+        # Uses Alembic's Python API against the packaged migrations — an installed
+        # wheel has neither the `alembic` console script on PATH nor alembic.ini.
+        from coderag.db.migrate import upgrade_to_head
 
-        env = dict(os.environ, CODERAG_DATABASE_URL=url)
-        r = subprocess.run(["alembic", "upgrade", "head"], env=env,
-                           capture_output=True, text=True)
-        if r.returncode == 0:
+        try:
+            upgrade_to_head(url)
             console.print("[green]Migrations applied.[/]")
-        else:
-            console.print(
-                "[yellow]Could not run migrations automatically "
-                "(run `alembic upgrade head` from a repo checkout).[/]"
-            )
+        except Exception as exc:
+            console.print(f"[red]Migrations failed:[/] {exc}")
+            console.print("[yellow]Retry with:[/] coderag migrate")
+            raise typer.Exit(1) from None
 
 
 @localdb_app.command("stop")
@@ -82,6 +80,23 @@ def localdb_status(
     console.print(f"[green]running (pid {pid})[/]" if pid else "[yellow]not running[/]")
 
 
+@app.command()
+def migrate(
+    database_url: str | None = typer.Option(
+        None, "--database-url", help="Defaults to CODERAG_DATABASE_URL."
+    ),
+) -> None:
+    """Create/upgrade the database schema (works from a pip/pipx install)."""
+    from coderag.db.migrate import current_revision, upgrade_to_head
+
+    try:
+        url = upgrade_to_head(database_url)
+    except Exception as exc:
+        console.print(f"[red]Migration failed:[/] {exc}")
+        raise typer.Exit(1) from None
+    console.print(f"[green]Database is up to date[/] (revision {current_revision(url)}).")
+
+
 @app.command()
 def index(
     path: str = typer.Argument(..., help="Path to the repository to index."),

coderag_ai-0.3.3/src/coderag/db/migrate.py

@@ -0,0 +1,51 @@
+"""Run database migrations from an installed package.
+
+The `alembic` console script and `alembic.ini` are only available in a source
+checkout. When CodeRAG is installed as a wheel (pipx/pip), neither is on PATH, so
+migrations are driven through Alembic's Python API with the migration directory
+resolved from the installed package.
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+from coderag.core.config import get_settings
+
+
+def migrations_path() -> Path:
+    """Filesystem path of the packaged Alembic migration directory."""
+    return Path(__file__).resolve().parent / "migrations"
+
+
+def _config(url: str):
+    from alembic.config import Config
+
+    cfg = Config()
+    cfg.set_main_option("script_location", str(migrations_path()))
+    # Escape '%' so ConfigParser interpolation doesn't choke on URL-encoded values.
+    cfg.set_main_option("sqlalchemy.url", url.replace("%", "%%"))
+    return cfg
+
+
+def upgrade_to_head(url: str | None = None) -> str:
+    """Upgrade the database to the latest revision. Returns the URL used."""
+    from alembic import command
+
+    target = url or get_settings().database_url
+    command.upgrade(_config(target), "head")
+    return target
+
+
+def current_revision(url: str | None = None) -> str | None:
+    """Return the revision the database is currently stamped at (None if empty)."""
+    from alembic.migration import MigrationContext
+    from sqlalchemy import create_engine
+
+    target = url or get_settings().database_url
+    engine = create_engine(target)
+    try:
+        with engine.connect() as conn:
+            return MigrationContext.configure(conn).get_current_revision()
+    finally:
+        engine.dispose()

coderag_ai-0.3.3/tests/test_phase15_migrate.py

@@ -0,0 +1,43 @@
+"""Migrations must work from an installed wheel (no alembic.ini, no CLI on PATH)."""
+
+from __future__ import annotations
+
+import uuid
+
+import pytest
+from sqlalchemy import create_engine, text
+
+from coderag.db.migrate import current_revision, migrations_path, upgrade_to_head
+
+pytestmark = pytest.mark.db
+
+
+def test_migrations_are_packaged():
+    """The migration scripts ship inside the package, not just the source tree."""
+    path = migrations_path()
+    assert (path / "env.py").is_file()
+    versions = list((path / "versions").glob("*.py"))
+    assert versions, "no migration revisions packaged"
+
+
+def test_upgrade_to_head_without_alembic_ini(pg_uri: str):
+    """Regression: `localdb start` used to shell out to an `alembic` binary that
+    isn't on PATH in a pipx install. Migrations now run via the Python API."""
+    dbname = "migapi_" + uuid.uuid4().hex[:8]
+    admin = create_engine(pg_uri, isolation_level="AUTOCOMMIT")
+    with admin.connect() as conn:
+        conn.execute(text(f'CREATE DATABASE "{dbname}"'))
+    admin.dispose()
+
+    url = pg_uri.replace("/postgres?", f"/{dbname}?")
+    assert current_revision(url) is None          # fresh database
+    upgrade_to_head(url)
+    assert current_revision(url) is not None      # stamped after upgrade
+
+    eng = create_engine(url)
+    with eng.connect() as conn:
+        tables = set(conn.execute(text(
+            "SELECT table_name FROM information_schema.tables WHERE table_schema='public'"
+        )).scalars())
+    eng.dispose()
+    assert {"repositories", "symbols", "symbol_embeddings"} <= tables