chore: import upstream snapshot with attribution

integrations/azure_sql/tools/azure_sql_current_queries_tool/__init__.py

@@ -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),
+    )

integrations/azure_sql/tools/azure_sql_resource_stats_tool/__init__.py

@@ -0,0 +1,44 @@
+"""Azure SQL Resource Stats Tool."""
+
+from typing import Any
+
+from core.tool_framework.tool_decorator import tool
+from integrations.azure_sql import (
+    azure_sql_extract_params,
+    azure_sql_is_available,
+    get_resource_stats,
+    resolve_azure_sql_config,
+)
+
+
+@tool(
+    name="get_azure_sql_resource_stats",
+    description="Retrieve Azure SQL Database resource utilization history (CPU, IO, log throughput, memory) with throttling risk assessment.",
+    source="azure_sql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Diagnosing DTU/vCore throttling on Azure SQL Database",
+        "Identifying resource saturation causing query timeouts",
+        "Reviewing historical resource trends to determine if tier upgrade is needed",
+    ],
+    is_available=azure_sql_is_available,
+    injected_params=("server",),
+    extract_params=azure_sql_extract_params,
+)
+def get_azure_sql_resource_stats(
+    server: str,
+    database: str | None = None,
+    port: int = 1433,
+    minutes: int = 30,
+) -> dict[str, Any]:
+    """Fetch resource utilization stats from an Azure SQL Database instance."""
+    _db_defaulted = database is None
+    if database is None:
+        database = "master"
+    config = resolve_azure_sql_config(server=server, database=database, port=port)
+    result = get_resource_stats(config, minutes=minutes)
+    if _db_defaulted:
+        result["default_db_warning"] = (
+            "WARNING: No database was specified; defaulted to 'master'. Results may not reflect application data."
+        )
+    return result

integrations/azure_sql/tools/azure_sql_server_status_tool/__init__.py

@@ -0,0 +1,43 @@
+"""Azure SQL Server Status Tool."""
+
+from typing import Any
+
+from core.tool_framework.tool_decorator import tool
+from integrations.azure_sql import (
+    azure_sql_extract_params,
+    azure_sql_is_available,
+    get_server_status,
+    resolve_azure_sql_config,
+)
+
+
+@tool(
+    name="get_azure_sql_server_status",
+    description="Retrieve Azure SQL Database server metrics including service tier, resource utilization, connections, and database size.",
+    source="azure_sql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Checking Azure SQL Database health during an incident",
+        "Identifying DTU/vCore throttling or resource exhaustion",
+        "Reviewing service tier and connection saturation",
+    ],
+    is_available=azure_sql_is_available,
+    injected_params=("server",),
+    extract_params=azure_sql_extract_params,
+)
+def get_azure_sql_server_status(
+    server: str,
+    database: str | None = None,
+    port: int = 1433,
+) -> dict[str, Any]:
+    """Fetch server status metrics from an Azure SQL Database instance."""
+    _db_defaulted = database is None
+    if database is None:
+        database = "master"
+    config = resolve_azure_sql_config(server=server, database=database, port=port)
+    result = get_server_status(config)
+    if _db_defaulted:
+        result["default_db_warning"] = (
+            "WARNING: No database was specified; defaulted to 'master'. Results may not reflect application data."
+        )
+    return result

integrations/azure_sql/tools/azure_sql_slow_queries_tool/__init__.py

@@ -0,0 +1,45 @@
+"""Azure SQL Slow 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_slow_queries,
+    resolve_azure_sql_config,
+)
+
+
+@tool(
+    name="get_azure_sql_slow_queries",
+    description=(
+        "Retrieve slow query statistics from Azure SQL Database query stats DMV,"
+        " ordered by average elapsed time."
+    ),
+    source="azure_sql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Identifying queries with high average execution time",
+        "Finding resource-intensive queries causing DTU throttling",
+        "Reviewing query performance trends for capacity planning",
+    ],
+    is_available=azure_sql_is_available,
+    injected_params=("server",),
+    extract_params=azure_sql_extract_params,
+)
+def get_azure_sql_slow_queries(
+    server: str,
+    database: str | None = None,
+    port: int = 1433,
+    threshold_ms: int = 1000,
+) -> dict[str, Any]:
+    """Fetch slow query statistics 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_slow_queries(config, threshold_ms=threshold_ms),
+    )

integrations/azure_sql/tools/azure_sql_wait_stats_tool/__init__.py

@@ -0,0 +1,43 @@
+"""Azure SQL Wait Stats Tool."""
+
+from typing import Any
+
+from core.tool_framework.tool_decorator import tool
+from integrations.azure_sql import (
+    azure_sql_extract_params,
+    azure_sql_is_available,
+    get_wait_stats,
+    resolve_azure_sql_config,
+)
+
+
+@tool(
+    name="get_azure_sql_wait_stats",
+    description="Retrieve top wait statistics from Azure SQL Database to diagnose throttling, lock contention, IO bottlenecks, and network issues.",
+    source="azure_sql",
+    surfaces=("investigation", "chat"),
+    use_cases=[
+        "Identifying the most impactful wait types during an incident",
+        "Diagnosing lock contention or IO bottlenecks",
+        "Understanding resource governance limits on Azure SQL",
+    ],
+    is_available=azure_sql_is_available,
+    injected_params=("server",),
+    extract_params=azure_sql_extract_params,
+)
+def get_azure_sql_wait_stats(
+    server: str,
+    database: str | None = None,
+    port: int = 1433,
+) -> dict[str, Any]:
+    """Fetch wait statistics from an Azure SQL Database instance."""
+    _db_defaulted = database is None
+    if database is None:
+        database = "master"
+    config = resolve_azure_sql_config(server=server, database=database, port=port)
+    result = get_wait_stats(config)
+    if _db_defaulted:
+        result["default_db_warning"] = (
+            "WARNING: No database was specified; defaulted to 'master'. Results may not reflect application data."
+        )
+    return result