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,45 @@
+"""Azure SQL Current Queries 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.azure_sql import (
+    azure_sql_extract_params,
+    azure_sql_is_available,
+    get_current_queries,
+    resolve_azure_sql_config,
+)
+
+
+@tool(
+    name="get_azure_sql_current_queries",
+    description=(
+        "Retrieve currently running queries on Azure SQL Database above a duration"
+        " threshold, including wait types and resource usage."
+    ),
+    source="azure_sql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Identifying long-running queries causing lock contention",
+        "Diagnosing blocking chains during an Azure SQL incident",
+        "Finding queries consuming excessive CPU or IO",
+    ],
+    is_available=azure_sql_is_available,
+    injected_params=("server",),
+    extract_params=azure_sql_extract_params,
+)
+def get_azure_sql_current_queries(
+    server: str,
+    database: str | None = None,
+    port: int = 1433,
+    threshold_seconds: int = 1,
+) -> dict[str, Any]:
+    """Fetch currently running queries from an Azure SQL Database instance."""
+    return call_db_tool_with_default_db_warning(
+        database=database,
+        default_db_name="master",
+        config_resolver=resolve_azure_sql_config,
+        resolver_kwargs={"server": server, "port": port},
+        db_caller=lambda config: get_current_queries(config, threshold_seconds=threshold_seconds),
+    )