postgres-aiops 0.10.2__tar.gz → 0.10.3__tar.gz

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/.claude-plugin/plugin.json

@@ -2,7 +2,7 @@
   "$schema": "https://anthropic.com/claude-code/plugin.schema.json",
   "name": "postgres-aiops",
   "displayName": "Postgres AIops",
-  "version": "0.10.2",
+  "version": "0.10.3",
   "description": "Governed PostgreSQL DBA ops: slow-query RCA, bloat/vacuum & blocking-lock analysis; 35 MCP tools.",
   "author": {
     "name": "Wei Zhou",

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/.mcp.json

@@ -5,7 +5,7 @@
       "command": "uvx",
       "args": [
         "--from",
-        "postgres-aiops==0.10.2",
+        "postgres-aiops==0.10.3",
         "postgres-aiops-mcp"
       ]
     }

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/CHANGELOG.md

@@ -1,5 +1,14 @@
 # Changelog
 
+## v0.10.3 — 2026-09-15
+
+### Fixed
+- CLI commands that read the engine without calling an MCP tool now write an
+  audit row, as every MCP call does: they run through the same `@governed_tool`
+  harness, budget and runaway guard included. Before, an operator's CLI reads
+  left no trace in `audit.db`, contrary to the documented guarantee that MCP
+  and CLI are audited alike. A test fails if any engine command escapes it.
+
 ## v0.10.2 — 2026-09-12
 
 ### Changed

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: postgres-aiops
-Version: 0.10.2
+Version: 0.10.3
 Summary: Governed AI-ops for PostgreSQL DBA operations: slow-query RCA, bloat/vacuum & blocking-lock analysis with a built-in governance harness (audit, budget, undo, risk tiers)
 Author-email: Wei Zhou <zhouwei008@gmail.com>
 License-Expression: MIT

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/openclaw.plugin.json

@@ -11,7 +11,7 @@
       "command": "uvx",
       "args": [
         "--from",
-        "postgres-aiops==0.10.2",
+        "postgres-aiops==0.10.3",
         "postgres-aiops-mcp"
       ]
     }

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zw008/postgres-aiops",
-  "version": "0.10.2",
+  "version": "0.10.3",
   "description": "Governed PostgreSQL DBA ops: slow-query RCA, bloat/vacuum & blocking-lock analysis; 35 MCP tools.",
   "private": true,
   "license": "MIT",

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/_common.py

@@ -180,3 +180,43 @@ def double_confirm(action: str, resource: str) -> None:
         f"Confirm 2/2: really {action} '{resource}'? This may be irreversible.",
         abort=True,
     )
+
+
+def audited(fn: Callable) -> Callable:
+    """Audit a CLI command that reaches the engine without calling an MCP tool.
+
+    MCP tools write their audit row through ``@governed_tool``; a CLI command that
+    called the ops layer directly used to write none, so an operator's CLI reads
+    left no trail. This runs the command through the same harness: one audit row per
+    invocation (tool = the command function, params = its options), plus the budget
+    and runaway guard. ``typer.Exit`` is how a command ends: exit code 0 is audited
+    ``ok``, a non-zero code ``error``, and the exit is re-raised unchanged. Any other
+    exception is audited ``error`` and propagates to ``cli_errors`` as before.
+    """
+    import functools as _functools
+
+    from postgres_aiops.governance import governed_tool
+
+    @_functools.wraps(fn)
+    def body(*args: Any, **kwargs: Any) -> Any:
+        try:
+            fn(*args, **kwargs)
+        except typer.Exit as exc:
+            code = exc.exit_code
+            outcome: dict[str, Any] = {"_exitCode": code}
+            if code:
+                outcome["error"] = f"command exited with code {code}"
+            return outcome
+        return None
+
+    governed = governed_tool(risk_level="low")(body)
+
+    @_functools.wraps(fn)
+    def wrapper(*args: Any, **kwargs: Any) -> Any:
+        result = governed(*args, **kwargs)
+        if isinstance(result, dict) and "_exitCode" in result:
+            raise typer.Exit(result["_exitCode"])
+        return None
+
+    wrapper._is_audited_cli = True  # type: ignore[attr-defined]
+    return wrapper

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/activity.py

@@ -7,7 +7,7 @@ from typing import Annotated
 
 import typer
 
-from postgres_aiops.cli._common import TargetOption, cli_errors, console, get_connection
+from postgres_aiops.cli._common import TargetOption, audited, cli_errors, console, get_connection
 
 activity_app = typer.Typer(
     name="activity",
@@ -18,6 +18,7 @@ activity_app = typer.Typer(
 
 @activity_app.command("list")
 @cli_errors
+@audited
 def activity_list(
     state: Annotated[str | None, typer.Option("--state", help="Filter by state")] = None,
     target: TargetOption = None,
@@ -31,6 +32,7 @@ def activity_list(
 
 @activity_app.command("long")
 @cli_errors
+@audited
 def activity_long(
     min_seconds: Annotated[int, typer.Option("--min-seconds", help="Minimum age")] = 60,
     target: TargetOption = None,
@@ -44,6 +46,7 @@ def activity_long(
 
 @activity_app.command("locks")
 @cli_errors
+@audited
 def activity_locks(target: TargetOption = None) -> None:
     """List held/awaited locks."""
     from postgres_aiops.ops import activity as ops

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/analyze.py

@@ -8,6 +8,7 @@ import typer
 
 from postgres_aiops.cli._common import (
     TargetOption,
+    audited,
     cli_errors,
     get_connection,
     print_result,
@@ -22,6 +23,7 @@ analyze_app = typer.Typer(
 
 @analyze_app.command("slow-query")
 @cli_errors
+@audited
 def analyze_slow_query(
     explain_sql: Annotated[str | None, typer.Option("--explain", help="SQL to EXPLAIN")] = None,
     limit: Annotated[int, typer.Option("--limit", help="Statements to consider")] = 20,
@@ -41,6 +43,7 @@ def analyze_slow_query(
 
 @analyze_app.command("bloat-vacuum")
 @cli_errors
+@audited
 def analyze_bloat_vacuum(
     limit: Annotated[int, typer.Option("--limit", help="Tables to consider")] = 50,
     target: TargetOption = None,
@@ -58,6 +61,7 @@ def analyze_bloat_vacuum(
 
 @analyze_app.command("blocking")
 @cli_errors
+@audited
 def analyze_blocking(target: TargetOption = None) -> None:
     """Build the blocking-lock chain and name the root blocker."""
     from postgres_aiops.ops import activity, analysis

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/index.py

@@ -9,6 +9,7 @@ import typer
 
 from postgres_aiops.cli._common import (
     TargetOption,
+    audited,
     cli_errors,
     console,
     get_connection,
@@ -24,6 +25,7 @@ index_app = typer.Typer(
 
 @index_app.command("unused")
 @cli_errors
+@audited
 def index_unused(target: TargetOption = None) -> None:
     """Non-unique, non-primary indexes with zero scans."""
     from postgres_aiops.ops import indexes as ops
@@ -34,6 +36,7 @@ def index_unused(target: TargetOption = None) -> None:
 
 @index_app.command("missing")
 @cli_errors
+@audited
 def index_missing(target: TargetOption = None) -> None:
     """Tables with heavy sequential scans (missing-index hints)."""
     from postgres_aiops.ops import indexes as ops
@@ -44,6 +47,7 @@ def index_missing(target: TargetOption = None) -> None:
 
 @index_app.command("bloat")
 @cli_errors
+@audited
 def index_bloat(
     limit: Annotated[int, typer.Option("--limit", help="Indexes to inspect")] = 50,
     target: TargetOption = None,
@@ -57,6 +61,7 @@ def index_bloat(
 
 @index_app.command("invalid")
 @cli_errors
+@audited
 def index_invalid(target: TargetOption = None) -> None:
     """Invalid and duplicate indexes."""
     from postgres_aiops.ops import indexes as ops

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/overview.py

@@ -4,10 +4,11 @@ from __future__ import annotations
 
 import json
 
-from postgres_aiops.cli._common import TargetOption, cli_errors, console, get_connection
+from postgres_aiops.cli._common import TargetOption, audited, cli_errors, console, get_connection
 
 
 @cli_errors
+@audited
 def overview_cmd(target: TargetOption = None) -> None:
     """One-shot cluster health: version, connections, long queries, bloat, replication."""
     from postgres_aiops.ops import overview as ops

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/query.py

@@ -10,6 +10,7 @@ import typer
 from postgres_aiops.cli._common import (
     DryRunOption,
     TargetOption,
+    audited,
     checked,
     cli_errors,
     console,
@@ -28,6 +29,7 @@ query_app = typer.Typer(
 
 @query_app.command("top")
 @cli_errors
+@audited
 def query_top(
     order_by: Annotated[
         str, typer.Option("--order-by", help="total_time|mean_time|calls|rows|io")
@@ -44,6 +46,7 @@ def query_top(
 
 @query_app.command("explain")
 @cli_errors
+@audited
 def query_explain(
     sql: Annotated[str, typer.Argument(help="A single SQL statement to EXPLAIN")],
     analyze: Annotated[

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/replication.py

@@ -6,7 +6,7 @@ import json
 
 import typer
 
-from postgres_aiops.cli._common import TargetOption, cli_errors, console, get_connection
+from postgres_aiops.cli._common import TargetOption, audited, cli_errors, console, get_connection
 
 repl_app = typer.Typer(
     name="repl",
@@ -17,6 +17,7 @@ repl_app = typer.Typer(
 
 @repl_app.command("status")
 @cli_errors
+@audited
 def repl_status(target: TargetOption = None) -> None:
     """Connected standbys and replay lag."""
     from postgres_aiops.ops import replication as ops
@@ -27,6 +28,7 @@ def repl_status(target: TargetOption = None) -> None:
 
 @repl_app.command("slots")
 @cli_errors
+@audited
 def repl_slots(target: TargetOption = None) -> None:
     """Replication slots (inactive slots flagged)."""
     from postgres_aiops.ops import replication as ops
@@ -37,6 +39,7 @@ def repl_slots(target: TargetOption = None) -> None:
 
 @repl_app.command("wal")
 @cli_errors
+@audited
 def repl_wal(target: TargetOption = None) -> None:
     """WAL position, level and archiver health."""
     from postgres_aiops.ops import replication as ops

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/server.py

@@ -7,7 +7,7 @@ from typing import Annotated
 
 import typer
 
-from postgres_aiops.cli._common import TargetOption, cli_errors, console, get_connection
+from postgres_aiops.cli._common import TargetOption, audited, cli_errors, console, get_connection
 
 server_app = typer.Typer(
     name="server",
@@ -18,6 +18,7 @@ server_app = typer.Typer(
 
 @server_app.command("version")
 @cli_errors
+@audited
 def server_version(target: TargetOption = None) -> None:
     """Server version, uptime and recovery state."""
     from postgres_aiops.ops import server as ops
@@ -28,6 +29,7 @@ def server_version(target: TargetOption = None) -> None:
 
 @server_app.command("settings")
 @cli_errors
+@audited
 def server_settings(
     pattern: Annotated[str | None, typer.Argument(help="Name substring filter")] = None,
     target: TargetOption = None,
@@ -41,6 +43,7 @@ def server_settings(
 
 @server_app.command("extensions")
 @cli_errors
+@audited
 def server_extensions(target: TargetOption = None) -> None:
     """Installed extensions."""
     from postgres_aiops.ops import server as ops
@@ -51,6 +54,7 @@ def server_extensions(target: TargetOption = None) -> None:
 
 @server_app.command("databases")
 @cli_errors
+@audited
 def server_databases(target: TargetOption = None) -> None:
     """Databases with owner, encoding and size."""
     from postgres_aiops.ops import server as ops
@@ -61,6 +65,7 @@ def server_databases(target: TargetOption = None) -> None:
 
 @server_app.command("roles")
 @cli_errors
+@audited
 def server_roles(target: TargetOption = None) -> None:
     """Roles and their attributes."""
     from postgres_aiops.ops import server as ops

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/postgres_aiops/cli/table.py

@@ -8,6 +8,7 @@ import typer
 
 from postgres_aiops.cli._common import (
     TargetOption,
+    audited,
     cli_errors,
     get_connection,
     print_result,
@@ -22,6 +23,7 @@ table_app = typer.Typer(
 
 @table_app.command("sizes")
 @cli_errors
+@audited
 def table_sizes(
     limit: Annotated[int, typer.Option("--limit", help="Tables to return")] = 20,
     target: TargetOption = None,
@@ -35,6 +37,7 @@ def table_sizes(
 
 @table_app.command("bloat")
 @cli_errors
+@audited
 def table_bloat(
     limit: Annotated[int, typer.Option("--limit", help="Tables to inspect")] = 50,
     target: TargetOption = None,
@@ -48,6 +51,7 @@ def table_bloat(
 
 @table_app.command("autovacuum")
 @cli_errors
+@audited
 def table_autovacuum(
     limit: Annotated[int, typer.Option("--limit", help="Tables to inspect")] = 50,
     target: TargetOption = None,

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "postgres-aiops"
-version = "0.10.2"
+version = "0.10.3"
 description = "Governed AI-ops for PostgreSQL DBA operations: slow-query RCA, bloat/vacuum & blocking-lock analysis with a built-in governance harness (audit, budget, undo, risk tiers)"
 readme = "README.md"
 license = "MIT"

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/server.json

@@ -7,12 +7,12 @@
     "url": "https://github.com/AIops-tools/Postgres-AIops",
     "source": "github"
   },
-  "version": "0.10.2",
+  "version": "0.10.3",
   "packages": [
     {
       "registryType": "pypi",
       "identifier": "postgres-aiops",
-      "version": "0.10.2",
+      "version": "0.10.3",
       "transport": {
         "type": "stdio"
       }

postgres_aiops-0.10.3/tests/test_cli_audited.py

@@ -0,0 +1,131 @@
+"""Every CLI command that reaches the engine is audited.
+
+MCP tools write their audit row through ``@governed_tool``. CLI commands that called
+the ops layer directly wrote none: on 2026-09-15 a read command of every tool in the
+line, run in a scratch home, left no audit.db while the harness itself worked. Each
+such command is now wrapped in ``audited`` (``cli/_common.py``); commands that call an
+MCP tool are audited by that tool. This pins the helper's behaviour and that no
+engine command escapes it.
+"""
+
+from __future__ import annotations
+
+import ast
+import inspect
+import sqlite3
+import textwrap
+
+import pytest
+import typer
+from typer.main import get_command
+
+from postgres_aiops.cli import app
+from postgres_aiops.cli._common import audited
+
+pytestmark = pytest.mark.unit
+
+#: Local-only commands: they never reach the engine (``undo`` goes through MCP tools).
+EXEMPT = {"init", "doctor", "secret", "mcp", "undo"}
+
+
+def _leaves() -> list[tuple[list[str], object]]:
+    found: list[tuple[list[str], object]] = []
+
+    def walk(cmd, path):
+        if hasattr(cmd, "list_commands"):
+            for name, sub in cmd.commands.items():
+                walk(sub, path + [name])
+        else:
+            found.append((path, cmd))
+
+    walk(get_command(app), [])
+    return found
+
+
+def _chain(fn):
+    seen = []
+    while fn is not None and fn not in seen:
+        seen.append(fn)
+        fn = getattr(fn, "__wrapped__", None)
+    return seen
+
+
+def _calls_an_mcp_tool(fn) -> bool:
+    source = inspect.getsource(fn)
+    if "mcp_server" in source:
+        return True
+    names = set()
+    for node in ast.parse(open(inspect.getsourcefile(fn)).read()).body:
+        if isinstance(node, ast.ImportFrom) and (node.module or "").startswith("mcp_server"):
+            names |= {a.asname or a.name for a in node.names}
+    used = {n.id for n in ast.walk(ast.parse(textwrap.dedent(source))) if isinstance(n, ast.Name)}
+    return bool(names & used)
+
+
+def test_every_engine_command_is_audited():
+    commands = [(path, cmd) for path, cmd in _leaves() if path and path[0] not in EXEMPT]
+    assert commands, "no commands found: the walker is broken, not the CLI clean"
+    missing = [
+        " ".join(path) for path, cmd in commands
+        if not any(getattr(f, "_is_audited_cli", False) for f in _chain(cmd.callback))
+        and not _calls_an_mcp_tool(inspect.unwrap(cmd.callback))
+    ]
+    assert not missing, f"CLI commands that leave no audit row: {missing}"
+
+
+@pytest.fixture
+def home(tmp_path, monkeypatch):
+    import postgres_aiops.governance.audit as audit_mod
+    import postgres_aiops.governance.budget as budget_mod
+
+    monkeypatch.setenv("POSTGRES_AIOPS_HOME", str(tmp_path))
+    audit_mod.reset_engine()
+    budget_mod.reset_budget()
+    yield tmp_path
+    audit_mod.reset_engine()
+    budget_mod.reset_budget()
+
+
+def _rows(home) -> list[tuple[str, str]]:
+    con = sqlite3.connect(home / "audit.db")
+    try:
+        return con.execute("SELECT tool, status FROM audit_log ORDER BY id").fetchall()
+    finally:
+        con.close()
+
+
+def test_audited_records_success_exit_codes_and_exceptions(home):
+    @audited
+    def cmd_ok(target=None):
+        return None
+
+    @audited
+    def cmd_exit_zero(target=None):
+        raise typer.Exit()
+
+    @audited
+    def cmd_exit_one(target=None):
+        raise typer.Exit(1)
+
+    @audited
+    def cmd_raises(target=None):
+        raise RuntimeError("engine unreachable")
+
+    cmd_ok()
+    with pytest.raises(typer.Exit) as zero:
+        cmd_exit_zero()
+    assert zero.value.exit_code == 0
+    with pytest.raises(typer.Exit) as one:
+        cmd_exit_one()
+    assert one.value.exit_code == 1
+    with pytest.raises(RuntimeError, match="engine unreachable"):
+        cmd_raises()
+    assert _rows(home) == [("cmd_ok", "ok"), ("cmd_exit_zero", "ok"),
+                           ("cmd_exit_one", "error"), ("cmd_raises", "error")]
+
+
+def test_audited_keeps_the_command_signature_for_typer():
+    def cmd(limit: int = 5, target: str | None = None) -> None:
+        return None
+
+    assert list(inspect.signature(audited(cmd)).parameters) == ["limit", "target"]

{postgres_aiops-0.10.2 → postgres_aiops-0.10.3}/uv.lock

@@ -492,7 +492,7 @@ wheels = [
 
 [[package]]
 name = "postgres-aiops"
-version = "0.10.2"
+version = "0.10.3"
 source = { editable = "." }
 dependencies = [
     { name = "cryptography" },