chore: import upstream snapshot with attribution

2026-07-13 13:10:45 +08:00
commit 4b6817381b
3933 changed files with 525247 additions and 0 deletions
@@ -0,0 +1,41 @@
+"""PostgreSQL Replication Status Tool."""
+
+from typing import Any
+
+from core.tool_framework.tool_decorator import tool
+from core.tool_framework.utils.sql_wrapper import call_db_tool_with_default_db_warning
+from integrations.postgresql import (
+    get_replication_status,
+    postgresql_extract_params,
+    postgresql_is_available,
+    resolve_postgresql_config,
+)
+
+
+@tool(
+    name="get_postgresql_replication_status",
+    description="Retrieve PostgreSQL replication status including replica lag, WAL positions, and streaming status.",
+    source="postgresql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Investigating replication lag issues during database incidents",
+        "Checking replica health and synchronization status",
+        "Monitoring WAL streaming and replica connectivity problems",
+    ],
+    is_available=postgresql_is_available,
+    injected_params=("host",),
+    extract_params=postgresql_extract_params,
+)
+def get_postgresql_replication_status(
+    host: str,
+    database: str | None = None,
+    port: int = 5432,
+) -> dict[str, Any]:
+    """Fetch replication status from a PostgreSQL primary server."""
+    return call_db_tool_with_default_db_warning(
+        database=database,
+        default_db_name="postgres",
+        config_resolver=resolve_postgresql_config,
+        resolver_kwargs={"host": host, "port": port},
+        db_caller=get_replication_status,
+    )