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

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/.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.3",
+  "version": "0.11.0",
   "description": "Governed PostgreSQL DBA ops: slow-query RCA, bloat/vacuum & blocking-lock analysis; 35 MCP tools.",
   "author": {
     "name": "Wei Zhou",

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

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

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

@@ -1,5 +1,25 @@
 # Changelog
 
+## v0.11.0 — 2026-10-15
+
+### Added
+- MCP tool annotations derived from each tool's governance `risk_level`, so
+  clients can gate approval from `tools/list` alone. `low` tools advertise
+  `readOnlyHint: true`; every other tool advertises `readOnlyHint: false` and
+  `destructiveHint: true`. The `medium` tier includes deletes and hard
+  power-offs, so a tool is never advertised as non-destructive on the strength
+  of its tier. `idempotentHint` only repeats an explicit declaration. The exact
+  level is published in `_meta` under `io.github.AIops-tools/riskLevel`. These
+  are hints for the client; audit and undo are unchanged.
+
+### Fixed
+- `blocking_lock_chain_rca`: the chain root's state was dropped and a missing
+  query became `""`. For a backend that is `idle in transaction`, `rootQuery` is
+  the last statement it ran, not a running one, so the recommended
+  `cancel_query` had nothing to cancel. Roots now carry `rootState` and
+  `rootIdle` (true for any `idle*` state, including plain `idle` with session-level locks), a missing query stays `null`, and the action for an
+  idle root points at `terminate_backend` (which rolls the transaction back).
+
 ## v0.10.3 — 2026-09-15
 
 ### Fixed

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: postgres-aiops
-Version: 0.10.3
+Version: 0.11.0
 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.3 → postgres_aiops-0.11.0}/RELEASE_NOTES.md

@@ -1,5 +1,8 @@
 # Release notes — postgres-aiops 0.6.0
 
+> Historical — this file covers the 0.6.0 release only and is not updated.
+> For every release since, see [CHANGELOG.md](CHANGELOG.md) and the GitHub releases.
+
 Previous release: 0.5.0.
 
 ## BREAKING — the authorization layer is removed

postgres_aiops-0.11.0/mcp_server/_annotations.py

@@ -0,0 +1,66 @@
+"""Derive MCP tool annotations from the governance risk level.
+
+Every tool already declares ``risk_level`` on ``@governed_tool``; this module
+publishes it to MCP clients as the standard ``ToolAnnotations`` hints plus the
+exact level under ``_meta``, so a client can gate approval without reading our
+docstrings.
+
+Only the safe direction is ever asserted:
+
+- ``low`` → ``readOnlyHint: true``. Backed by the smoke-test invariant that
+  ``risk_level == "low"`` exactly when the docstring says ``[READ]``.
+- anything else → ``readOnlyHint: false, destructiveHint: true``. The
+  ``medium`` tier is NOT non-destructive — it holds deletes, hard power-offs,
+  state kills and stat resets — so claiming ``destructiveHint: false`` from the
+  tier would tell a client it may skip approval for a delete. Over-warning
+  costs a confirmation prompt; under-warning costs data.
+- ``idempotentHint`` only echoes an explicit ``@governed_tool(idempotent=True)``
+  declaration; it is never inferred.
+- ``openWorldHint`` is left unset (spec default ``true``): every tool talks to
+  an external system.
+
+This file is identical across the AIops line; keep it that way.
+"""
+
+from typing import Any
+
+from mcp.types import ToolAnnotations
+
+RISK_META_KEY = "io.github.AIops-tools/riskLevel"
+RISK_LEVELS = frozenset({"low", "medium", "high", "critical"})
+
+
+def annotations_for(risk_level: str, idempotent: bool) -> ToolAnnotations:
+    """Return the MCP hints that a tool of ``risk_level`` may honestly claim."""
+    if risk_level not in RISK_LEVELS:
+        raise ValueError(
+            f"unknown risk_level {risk_level!r}; expected one of {sorted(RISK_LEVELS)}"
+        )
+    if risk_level == "low":
+        return ToolAnnotations(read_only_hint=True)
+    return ToolAnnotations(
+        read_only_hint=False,
+        destructive_hint=True,
+        idempotent_hint=idempotent,
+    )
+
+
+def apply_risk_annotations(server: Any) -> None:
+    """Replace every registered tool with a copy carrying its risk annotations.
+
+    Fails fast on a tool without ``@governed_tool`` metadata: an unannotated
+    tool would be advertised with the spec defaults, i.e. as a destructive
+    write with no stated risk level — wrong for a read, silent for a write.
+    """
+    tools = server._tool_manager._tools  # same private registry undo.py relies on
+    for name, tool in list(tools.items()):
+        risk_level = getattr(tool.fn, "_risk_level", None)
+        if risk_level is None:
+            raise RuntimeError(f"tool {name!r} has no @governed_tool risk_level")
+        idempotent = bool(getattr(tool.fn, "_idempotent", False))
+        tools[name] = tool.model_copy(
+            update={
+                "annotations": annotations_for(risk_level, idempotent),
+                "meta": {**(tool.meta or {}), RISK_META_KEY: risk_level},
+            }
+        )

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/mcp_server/server.py

@@ -14,6 +14,7 @@ License: MIT
 import logging
 import os
 
+from mcp_server._annotations import apply_risk_annotations
 from mcp_server._shared import _safe_error, mcp, tool_errors
 
 # Importing the tool modules registers every @mcp.tool() onto the shared
@@ -30,6 +31,10 @@ from mcp_server.tools import (  # noqa: F401 — side effects
     undo,
 )
 
+# Publish each tool's governance risk level as MCP annotations. Runs after
+# every tool module has registered, so no tool is advertised unannotated.
+apply_risk_annotations(mcp)
+
 __all__ = ["mcp", "main", "_safe_error", "tool_errors"]
 
 logger = logging.getLogger(__name__)

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

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

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

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

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/postgres_aiops/ops/analysis.py

@@ -201,6 +201,22 @@ def _descendants(root: int, children: dict[int, list[int]]) -> set[int]:
     return seen
 
 
+def _lock_root_action(pid: int, idle: bool | None) -> str:
+    if idle:
+        return (
+            f"Backend {pid} heads the chain but is not running a statement (see "
+            "rootState: an open transaction, or session-level locks): rootQuery is "
+            "the last statement it ran, not a running one. cancel_query has nothing "
+            f"to cancel; terminate_backend({pid}) ends the session, rolling back any "
+            "transaction and releasing its locks — confirm with its owner first."
+        )
+    return (
+        f"Backend {pid} is the head of the chain — investigate its transaction; "
+        f"cancel_query({pid}) to stop its statement, or terminate_backend({pid}) to "
+        "end the session and release the locks."
+    )
+
+
 def blocking_lock_chain_rca(pairs: list[dict]) -> dict:
     """[READ] Build the wait-for tree from blocking pairs and name the root blocker.
 
@@ -217,7 +233,8 @@ def blocking_lock_chain_rca(pairs: list[dict]) -> dict:
         return {"blockedBackends": 0, "roots": [], "note": "No blocking detected."}
 
     children: dict[int, list[int]] = {}
-    blocking_query: dict[int, str] = {}
+    blocking_query: dict[int, str | None] = {}
+    blocking_state: dict[int, str | None] = {}
     blocked_pids: set[int] = set()
     blocking_pids: set[int] = set()
     for e in edges:
@@ -225,7 +242,8 @@ def blocking_lock_chain_rca(pairs: list[dict]) -> dict:
         children.setdefault(g, []).append(b)
         blocked_pids.add(b)
         blocking_pids.add(g)
-        blocking_query.setdefault(g, e.get("blockingQuery") or "")
+        blocking_query.setdefault(g, e.get("blockingQuery"))
+        blocking_state.setdefault(g, e.get("blockingState"))
 
     root_pids = [pid for pid in blocking_pids if pid not in blocked_pids]
     if not root_pids:
@@ -242,16 +260,18 @@ def blocking_lock_chain_rca(pairs: list[dict]) -> dict:
     roots = []
     for pid in root_pids:
         blocked = _descendants(pid, children)
+        state = blocking_state.get(pid)
+        # Any idle state means no statement is running: "idle in transaction"
+        # holds row locks, plain "idle" can still hold session-level advisory locks.
+        idle = None if state is None else str(state).startswith("idle")
         roots.append({
             "rootPid": pid,
             "blockedCount": len(blocked),
             "blockedPids": sorted(blocked),
-            "rootQuery": blocking_query.get(pid, ""),
-            "action": (
-                f"Backend {pid} is the head of the chain — investigate its "
-                "transaction; cancel_query(pid) to stop its statement, or "
-                "terminate_backend(pid) to end the session and release the locks."
-            ),
+            "rootQuery": blocking_query.get(pid),
+            "rootState": state,
+            "rootIdle": idle,
+            "action": _lock_root_action(pid, idle),
         })
     roots.sort(key=lambda r: r["blockedCount"], reverse=True)
     return {

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "postgres-aiops"
-version = "0.10.3"
+version = "0.11.0"
 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.3 → postgres_aiops-0.11.0}/server.json

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

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/skills/postgres-aiops/references/agent-guardrails.md

@@ -28,7 +28,7 @@ recorded on the audit row; they are never required and never block a call.
 ## What still needs a prompt
 
 These are model-behaviour problems the harness cannot fix from the outside.
-Copy this into your agent's system prompt:
+Recommended setup for a local model (adapt it to your agent client):
 
 ```text
 You operate a PostgreSQL server through the postgres-aiops MCP tools.

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/skills/postgres-aiops/references/capabilities.md

@@ -32,7 +32,7 @@
 | `wal_status` | WAL fns + `pg_stat_archiver` | currentLsn, walLevel, max/minWalSize, archiver |
 | `slow_query_rca` | pg_stat_statements + EXPLAIN | worst{}, findings[] (cited cause/action) |
 | `bloat_and_vacuum_analysis` | table-bloat rows | recommendations[] (cited reasons + action) |
-| `blocking_lock_chain_rca` | `pg_blocking_pids` pairs | roots[], worstRootPid, deadlockSuspected |
+| `blocking_lock_chain_rca` | `pg_blocking_pids` pairs | roots[] (rootQuery, rootState, rootIdle — true for any `idle*` state; rootQuery is then the last statement run, not a running one — and action), worstRootPid, deadlockSuspected |
 | `undo_list` | local undo store | recorded, not-yet-applied reversible writes: undoId, ts, originalTool, inverseTool, note |
 
 The flagship analyses accept injected records (`statements=` / `tables=` /

{postgres_aiops-0.10.3 → postgres_aiops-0.11.0}/tests/test_analysis.py

@@ -82,3 +82,48 @@ def test_blocking_chain_detects_cycle():
 def test_blocking_chain_no_blocking():
     out = analysis.blocking_lock_chain_rca([])
     assert out["blockedBackends"] == 0
+
+
+# ── lock-chain root that is idle inside an open transaction ──────────────────
+
+
+def _pg_pairs(state, query="UPDATE t SET v=1 WHERE id=1"):
+    return [{"blockedPid": 201, "blockingPid": 100, "blockedQuery": "DELETE FROM t",
+             "blockingQuery": query, "blockingState": state}]
+
+
+def test_idle_in_transaction_root_is_named_and_cancel_is_ruled_out():
+    from postgres_aiops.ops import analysis as a
+
+    (root,) = a.blocking_lock_chain_rca(_pg_pairs("idle in transaction"))["roots"]
+    assert root["rootState"] == "idle in transaction"
+    assert root["rootIdle"] is True
+    # The query is the LAST statement it ran, not a running one.
+    assert "last statement" in root["action"]
+    assert "terminate_backend(100)" in root["action"]
+    assert "nothing to cancel" in root["action"]
+
+
+def test_active_root_offers_cancel_first():
+    from postgres_aiops.ops import analysis as a
+
+    (root,) = a.blocking_lock_chain_rca(_pg_pairs("active"))["roots"]
+    assert root["rootIdle"] is False
+    assert "cancel_query(100)" in root["action"]
+
+
+def test_missing_root_query_stays_null():
+    from postgres_aiops.ops import analysis as a
+
+    (root,) = a.blocking_lock_chain_rca(_pg_pairs(None, query=None))["roots"]
+    assert root["rootQuery"] is None
+    assert root["rootState"] is None
+    assert root["rootIdle"] is None  # state unknown
+
+
+def test_plain_idle_root_with_session_locks_is_not_offered_cancel():
+    from postgres_aiops.ops import analysis as a
+
+    (root,) = a.blocking_lock_chain_rca(_pg_pairs("idle"))["roots"]
+    assert root["rootIdle"] is True
+    assert "terminate_backend(100)" in root["action"]

postgres_aiops-0.11.0/tests/test_tool_annotations.py

@@ -0,0 +1,128 @@
+"""Invariants for the MCP annotations derived from ``risk_level``.
+
+Asserted on ``list_tools()`` — the payload a client actually receives — not on
+the internal registry, so a regression in how annotations reach the wire fails
+here too. This file is identical across the AIops line.
+"""
+
+import asyncio
+import re
+from types import SimpleNamespace
+
+import pytest
+from mcp.server import MCPServer
+
+import mcp_server.server as server
+from mcp_server._annotations import (
+    RISK_META_KEY,
+    annotations_for,
+    apply_risk_annotations,
+)
+
+
+@pytest.fixture(scope="module")
+def listed():
+    return asyncio.run(server.mcp.list_tools())
+
+
+@pytest.fixture(scope="module")
+def registry():
+    return server.mcp._tool_manager._tools
+
+
+def test_every_tool_is_annotated(listed):
+    assert listed, "no tools listed"
+    missing = [t.name for t in listed if t.annotations is None or not t.meta]
+    assert not missing, f"tools advertised without annotations: {missing}"
+
+
+def test_meta_carries_the_declared_risk_level(listed, registry):
+    wrong = [t.name for t in listed if t.meta.get(RISK_META_KEY) != registry[t.name].fn._risk_level]
+    assert not wrong, f"_meta risk level differs from @governed_tool: {wrong}"
+
+
+def test_read_only_hint_true_exactly_for_low_risk(listed):
+    wrong = [
+        t.name for t in listed if t.annotations.read_only_hint != (t.meta[RISK_META_KEY] == "low")
+    ]
+    assert not wrong, f"readOnlyHint disagrees with risk_level: {wrong}"
+
+
+def test_no_write_claims_to_be_non_destructive(listed):
+    # medium holds deletes and hard power-offs; the tier cannot prove safety.
+    wrong = [
+        t.name
+        for t in listed
+        if not t.annotations.read_only_hint and t.annotations.destructive_hint is not True
+    ]
+    assert not wrong, f"writes advertised as non-destructive: {wrong}"
+
+
+def test_idempotent_hint_only_echoes_an_explicit_declaration(listed, registry):
+    wrong = [
+        t.name
+        for t in listed
+        if t.annotations.idempotent_hint and not registry[t.name].fn._idempotent
+    ]
+    assert not wrong, f"idempotentHint inferred without a declaration: {wrong}"
+
+
+def test_open_world_hint_is_never_narrowed(listed):
+    assert all(t.annotations.open_world_hint in (None, True) for t in listed)
+
+
+@pytest.mark.parametrize("level", ["low", "medium", "high", "critical"])
+def test_mapping_per_level(level):
+    hints = annotations_for(level, idempotent=False)
+    if level == "low":
+        assert hints.read_only_hint is True
+        assert hints.destructive_hint is None
+    else:
+        assert hints.read_only_hint is False
+        assert hints.destructive_hint is True
+        assert hints.idempotent_hint is False
+
+
+def test_declared_idempotent_write_is_echoed():
+    assert annotations_for("medium", idempotent=True).idempotent_hint is True
+
+
+def test_unknown_risk_level_is_rejected():
+    with pytest.raises(ValueError, match="unknown risk_level"):
+        annotations_for("Low", idempotent=False)
+
+
+def test_ungoverned_tool_fails_fast():
+    fake_tool = SimpleNamespace(fn=lambda: None, meta=None)
+    fake = SimpleNamespace(_tool_manager=SimpleNamespace(_tools={"rogue": fake_tool}))
+    with pytest.raises(RuntimeError, match="rogue"):
+        apply_risk_annotations(fake)
+
+
+def test_meta_key_is_the_published_contract():
+    # Clients match on this literal; it must be a valid, non-reserved _meta key.
+    assert RISK_META_KEY == "io.github.AIops-tools/riskLevel"
+    prefix, name = RISK_META_KEY.split("/")
+    label = re.compile(r"^[A-Za-z]([A-Za-z0-9-]*[A-Za-z0-9])?$")
+    labels = prefix.split(".")
+    assert all(label.match(part) for part in labels)
+    assert not {"mcp", "modelcontextprotocol"} & {part.lower() for part in labels}
+    assert re.match(r"^[A-Za-z0-9]([A-Za-z0-9._-]*[A-Za-z0-9])?$", name)
+
+
+def test_existing_meta_survives_on_a_real_tool():
+    fresh = MCPServer("annotations-test")
+
+    def probe() -> str:
+        return "ok"
+
+    probe._risk_level = "medium"
+    probe._idempotent = True
+    fresh.tool(meta={"x": 1})(probe)
+    apply_risk_annotations(fresh)
+
+    (listed_tool,) = asyncio.run(fresh.list_tools())
+    assert listed_tool.meta == {"x": 1, RISK_META_KEY: "medium"}
+    assert listed_tool.annotations.destructive_hint is True
+    assert listed_tool.annotations.idempotent_hint is True
+    assert fresh._tool_manager._tools["probe"].fn is probe

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

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