mcp-instana 0.3.1__py3-none-any.whl → 0.7.0__py3-none-any.whl

src/application/application_topology.py

@@ -55,65 +55,65 @@ class ApplicationTopologyMCPTools(BaseInstanaClient):
             logger.error(f"Error initializing ApplicationTopologyMCPTools: {e}", exc_info=True)
             raise
 
-    @register_as_tool(
-        title="Get Application Topology",
-        annotations=ToolAnnotations(readOnlyHint=True, destructiveHint=False)
-    )
-    async def get_application_topology(self,
-                              window_size: Optional[int] = None,
-                              to_timestamp: Optional[int] = None,
-                              application_id: Optional[str] = None,
-                              application_boundary_scope: Optional[str] = None,
-                              ctx = None) -> Dict[str, Any]:
-        """
-        Get the service topology from Instana Server.
-        This tool retrieves services and connections (call paths) between them for calls in the scope given by the parameters.
-
-        Args:
-            window_size: Size of time window in milliseconds
-            to_timestamp: Timestamp since Unix Epoch in milliseconds of the end of the time window
-            application_id: Filter by application ID
-            application_boundary_scope: Filter by application scope, i.e., INBOUND or ALL. The default value is INBOUND.
-            ctx: Context information
-
-        Returns:
-            A dictionary containing the service topology data
-        """
-
-        try:
-            logger.debug("Fetching service topology data")
-
-            # Set default values if not provided
-            if not to_timestamp:
-                to_timestamp = int(datetime.now().timestamp() * 1000)
-
-            if not window_size:
-                window_size = 3600000  # Default to 1 hour in milliseconds
-
-            # Call the API with raw JSON response to avoid Pydantic validation issues
-            # Note: The SDK expects parameters in camelCase, but we use snake_case in Python
-            # The SDK will handle the conversion
-            result = self.topology_api.get_services_map_without_preload_content(
-                window_size=window_size,
-                to=to_timestamp,
-                application_id=application_id,
-                application_boundary_scope=application_boundary_scope
-            )
-
-            # Parse the JSON response manually
-            import json
-            try:
-                # The result from get_services_map_without_preload_content is a response object
-                # We need to read the response data and parse it as JSON
-                response_text = result.data.decode('utf-8')
-                result_dict = json.loads(response_text)
-                logger.debug("Successfully retrieved service topology data")
-                return result_dict
-            except (json.JSONDecodeError, AttributeError) as json_err:
-                error_message = f"Failed to parse JSON response: {json_err}"
-                logger.error(error_message)
-                return {"error": error_message}
-
-        except Exception as e:
-            logger.error(f"Error in get_application_topology: {e}", exc_info=True)
-            return {"error": f"Failed to get application topology: {e!s}"}
+    # @register_as_tool(
+    #     title="Get Application Topology",
+    #     annotations=ToolAnnotations(readOnlyHint=True, destructiveHint=False)
+    # )
+    # async def get_application_topology(self,
+    #                           window_size: Optional[int] = None,
+    #                           to_timestamp: Optional[int] = None,
+    #                           application_id: Optional[str] = None,
+    #                           application_boundary_scope: Optional[str] = None,
+    #                           ctx = None) -> Dict[str, Any]:
+    #     """
+    #     Get the service topology from Instana Server.
+    #     This tool retrieves services and connections (call paths) between them for calls in the scope given by the parameters.
+
+    #     Args:
+    #         window_size: Size of time window in milliseconds
+    #         to_timestamp: Timestamp since Unix Epoch in milliseconds of the end of the time window
+    #         application_id: Filter by application ID
+    #         application_boundary_scope: Filter by application scope, i.e., INBOUND or ALL. The default value is INBOUND.
+    #         ctx: Context information
+
+    #     Returns:
+    #         A dictionary containing the service topology data
+    #     """
+
+    #     try:
+    #         logger.debug("Fetching service topology data")
+
+    #         # Set default values if not provided
+    #         if not to_timestamp:
+    #             to_timestamp = int(datetime.now().timestamp() * 1000)
+
+    #         if not window_size:
+    #             window_size = 3600000  # Default to 1 hour in milliseconds
+
+    #         # Call the API with raw JSON response to avoid Pydantic validation issues
+    #         # Note: The SDK expects parameters in camelCase, but we use snake_case in Python
+    #         # The SDK will handle the conversion
+    #         result = self.topology_api.get_services_map_without_preload_content(
+    #             window_size=window_size,
+    #             to=to_timestamp,
+    #             application_id=application_id,
+    #             application_boundary_scope=application_boundary_scope
+    #         )
+
+    #         # Parse the JSON response manually
+    #         import json
+    #         try:
+    #             # The result from get_services_map_without_preload_content is a response object
+    #             # We need to read the response data and parse it as JSON
+    #             response_text = result.data.decode('utf-8')
+    #             result_dict = json.loads(response_text)
+    #             logger.debug("Successfully retrieved service topology data")
+    #             return result_dict
+    #         except (json.JSONDecodeError, AttributeError) as json_err:
+    #             error_message = f"Failed to parse JSON response: {json_err}"
+    #             logger.error(error_message)
+    #             return {"error": error_message}
+
+    #     except Exception as e:
+    #         logger.error(f"Error in get_application_topology: {e}", exc_info=True)
+    #         return {"error": f"Failed to get application topology: {e!s}"}

src/core/custom_dashboard_smart_router_tool.py

@@ -0,0 +1,135 @@
+"""
+Custom Dashboard Smart Router Tool
+
+This module provides a unified MCP tool that routes queries to the appropriate
+custom dashboard-specific tools for Instana monitoring.
+"""
+
+import logging
+from typing import Any, Dict, Optional
+
+from mcp.types import ToolAnnotations
+
+from src.core.utils import BaseInstanaClient, register_as_tool
+
+logger = logging.getLogger(__name__)
+
+
+class CustomDashboardSmartRouterMCPTool(BaseInstanaClient):
+    """
+    Smart router that routes queries to Custom Dashboard tools.
+    The LLM agent determines the appropriate operation based on query understanding.
+    """
+
+    def __init__(self, read_token: str, base_url: str):
+        """Initialize the Custom Dashboard Smart Router MCP tool."""
+        super().__init__(read_token=read_token, base_url=base_url)
+
+        # Initialize the custom dashboard tool client
+        from src.custom_dashboard.custom_dashboard_tools import CustomDashboardMCPTools
+
+        self.dashboard_client = CustomDashboardMCPTools(read_token, base_url)
+
+        logger.info("Custom Dashboard Smart Router initialized")
+
+    @register_as_tool(
+        title="Manage Instana Custom Dashboards",
+        annotations=ToolAnnotations(readOnlyHint=False, destructiveHint=False)
+    )
+    async def manage_custom_dashboards(
+        self,
+        operation: str,
+        params: Optional[Dict[str, Any]] = None,
+        ctx=None
+    ) -> Dict[str, Any]:
+        """
+        Unified Instana custom dashboard manager for CRUD operations.
+
+        Operations:
+        - "get_all": Get all custom dashboards
+        - "get": Get a specific dashboard by ID
+        - "create": Create a new custom dashboard
+        - "update": Update an existing custom dashboard
+        - "delete": Delete a custom dashboard
+        - "get_shareable_users": Get shareable users for a dashboard
+        - "get_shareable_api_tokens": Get shareable API tokens for a dashboard
+
+        Parameters (params dict):
+        - dashboard_id: Dashboard ID (required for get, update, delete, get_shareable_users, get_shareable_api_tokens)
+        - custom_dashboard: Dashboard configuration payload (required for create, update)
+
+        Args:
+            operation: Operation to perform
+            params: Operation-specific parameters (optional)
+            ctx: MCP context (internal)
+
+        Returns:
+            Dictionary with results from the appropriate tool
+
+        Examples:
+            # Get all dashboards
+            operation="get_all"
+
+            # Get specific dashboard
+            operation="get", params={"dashboard_id": "abc123"}
+
+            # Create dashboard
+            operation="create", params={"custom_dashboard": {"title": "My Dashboard", "widgets": []}}
+
+            # Update dashboard
+            operation="update", params={"dashboard_id": "abc123", "custom_dashboard": {"title": "Updated Dashboard"}}
+
+            # Delete dashboard
+            operation="delete", params={"dashboard_id": "abc123"}
+
+            # Get shareable users
+            operation="get_shareable_users", params={"dashboard_id": "abc123"}
+
+            # Get shareable API tokens
+            operation="get_shareable_api_tokens", params={"dashboard_id": "abc123"}
+        """
+        try:
+            logger.info(f"Custom Dashboard Smart Router received: operation={operation}")
+
+            # Initialize params if not provided
+            if params is None:
+                params = {}
+
+            # Validate operation
+            valid_operations = [
+                "get_all", "get", "create", "update", "delete",
+                "get_shareable_users", "get_shareable_api_tokens"
+            ]
+
+            if operation not in valid_operations:
+                return {
+                    "error": f"Invalid operation '{operation}'",
+                    "valid_operations": valid_operations
+                }
+
+            # Extract parameters
+            dashboard_id = params.get("dashboard_id")
+            custom_dashboard = params.get("custom_dashboard")
+
+            # Route to the dashboard client
+            logger.info(f"Routing to Custom Dashboard client for operation: {operation}")
+
+            result = await self.dashboard_client.execute_dashboard_operation(
+                operation=operation,
+                dashboard_id=dashboard_id,
+                custom_dashboard=custom_dashboard,
+                ctx=ctx
+            )
+
+            return {
+                "operation": operation,
+                "dashboard_id": dashboard_id if dashboard_id else None,
+                "results": result
+            }
+
+        except Exception as e:
+            logger.error(f"Error in custom dashboard smart router: {e}", exc_info=True)
+            return {
+                "error": f"Custom dashboard smart router error: {e!s}",
+                "operation": operation
+            }

src/core/server.py

@@ -20,6 +20,8 @@ from src.prompts import PROMPT_REGISTRY
 
 load_dotenv()
 
+from src.observability import task, workflow
+
 # Configure logging
 logging.basicConfig(
     level=logging.INFO,  # Default level, can be overridden
@@ -63,26 +65,30 @@ from fastmcp import FastMCP
 
 @dataclass
 class MCPState:
-    """State for the MCP server."""
+    """State for the MCP server with all tool categories."""
+    # Router tools
+    smart_router_client: Any = None
+    custom_dashboard_smart_router_client: Any = None
+
+    # Infrastructure - Only the new two-pass elicitation tool
+    infra_analyze_new_client: Any = None
+
+    # Events tools
     events_client: Any = None
-    infra_client: Any = None
-    app_resource_client: Any = None
-    app_metrics_client: Any = None
-    app_alert_client: Any = None
-    infra_catalog_client: Any = None
-    infra_topo_client: Any = None
-    infra_analyze_client: Any = None
-    infra_metrics_client: Any = None
-    app_catalog_client: Any = None
-    app_topology_client: Any = None
-    app_analyze_client: Any = None
-    app_settings_client: Any = None
-    app_global_alert_client: Any = None
+
+    # Automation tools
+    action_catalog_client: Any = None
+    action_history_client: Any = None
+
+    # Website tools
     website_metrics_client: Any = None
     website_catalog_client: Any = None
     website_analyze_client: Any = None
     website_configuration_client: Any = None
 
+    # Settings tools
+    custom_dashboard_client: Any = None
+
 # Global variables to store credentials for lifespan
 _global_token = None
 _global_base_url = None
@@ -234,33 +240,15 @@ async def execute_tool(tool_name: str, arguments: dict, clients_state) -> str:
 def get_client_categories():
     """Get client categories with lazy imports to avoid circular dependencies"""
     try:
-        from src.application.application_alert_config import ApplicationAlertMCPTools
-        from src.application.application_analyze import ApplicationAnalyzeMCPTools
-        from src.application.application_catalog import ApplicationCatalogMCPTools
-        from src.application.application_global_alert_config import (
-            ApplicationGlobalAlertMCPTools,
-        )
-        from src.application.application_metrics import ApplicationMetricsMCPTools
-        from src.application.application_resources import ApplicationResourcesMCPTools
-        from src.application.application_settings import ApplicationSettingsMCPTools
-        from src.application.application_topology import ApplicationTopologyMCPTools
         from src.automation.action_catalog import ActionCatalogMCPTools
         from src.automation.action_history import ActionHistoryMCPTools
-        from src.event.events_tools import AgentMonitoringEventsMCPTools
-        from src.infrastructure.infrastructure_analyze import (
-            InfrastructureAnalyzeMCPTools,
-        )
-        from src.infrastructure.infrastructure_catalog import (
-            InfrastructureCatalogMCPTools,
-        )
-        from src.infrastructure.infrastructure_metrics import (
-            InfrastructureMetricsMCPTools,
-        )
-        from src.infrastructure.infrastructure_resources import (
-            InfrastructureResourcesMCPTools,
+        from src.core.custom_dashboard_smart_router_tool import (
+            CustomDashboardSmartRouterMCPTool,
         )
-        from src.infrastructure.infrastructure_topology import (
-            InfrastructureTopologyMCPTools,
+        from src.core.smart_router_tool import SmartRouterMCPTool
+        from src.event.events_tools import AgentMonitoringEventsMCPTools
+        from src.infrastructure.infrastructure_analyze_new import (
+            InfrastructureAnalyzeOption2,
         )
         from src.settings.custom_dashboard_tools import CustomDashboardMCPTools
         from src.website.website_analyze import WebsiteAnalyzeMCPTools
@@ -272,22 +260,12 @@ def get_client_categories():
         return {}
 
     return {
-        "infra": [
-            ('infra_client', InfrastructureResourcesMCPTools),
-            ('infra_catalog_client', InfrastructureCatalogMCPTools),
-            ('infra_topo_client', InfrastructureTopologyMCPTools),
-            ('infra_analyze_client', InfrastructureAnalyzeMCPTools),
-            ('infra_metrics_client', InfrastructureMetricsMCPTools),
+        "router": [
+            ('smart_router_client', SmartRouterMCPTool),
+            ('custom_dashboard_smart_router_client', CustomDashboardSmartRouterMCPTool),
         ],
-        "app": [
-            ('app_resource_client', ApplicationResourcesMCPTools),
-            ('app_metrics_client', ApplicationMetricsMCPTools),
-            ('app_alert_client', ApplicationAlertMCPTools),
-            ('app_catalog_client', ApplicationCatalogMCPTools),
-            ('app_topology_client', ApplicationTopologyMCPTools),
-            ('app_analyze_client', ApplicationAnalyzeMCPTools),
-            ('app_settings_client', ApplicationSettingsMCPTools),
-            ('app_global_alert_client', ApplicationGlobalAlertMCPTools),
+        "infra": [
+            ('infra_analyze_new_client', InfrastructureAnalyzeOption2),
         ],
         "events": [
             ('events_client', AgentMonitoringEventsMCPTools),
@@ -310,79 +288,65 @@ def get_client_categories():
 def get_prompt_categories():
     """Get prompt categories organized by functionality"""
     # Import the class-based prompts
-    from src.prompts.application.application_alerts import ApplicationAlertsPrompts
-    from src.prompts.application.application_catalog import ApplicationCatalogPrompts
-    from src.prompts.application.application_metrics import ApplicationMetricsPrompts
-    from src.prompts.application.application_resources import (
-        ApplicationResourcesPrompts,
-    )
-    from src.prompts.application.application_settings import ApplicationSettingsPrompts
-    from src.prompts.application.application_topology import ApplicationTopologyPrompts
-    from src.prompts.infrastructure.infrastructure_analyze import (
-        InfrastructureAnalyzePrompts,
-    )
-    from src.prompts.infrastructure.infrastructure_catalog import (
-        InfrastructureCatalogPrompts,
-    )
-    from src.prompts.infrastructure.infrastructure_metrics import (
-        InfrastructureMetricsPrompts,
-    )
-    from src.prompts.infrastructure.infrastructure_resources import (
-        InfrastructureResourcesPrompts,
-    )
-    from src.prompts.infrastructure.infrastructure_topology import (
-        InfrastructureTopologyPrompts,
-    )
-    from src.prompts.settings.custom_dashboard import CustomDashboardPrompts
-    from src.prompts.website.website_analyze import WebsiteAnalyzePrompts
-    from src.prompts.website.website_catalog import WebsiteCatalogPrompts
-    from src.prompts.website.website_configuration import WebsiteConfigurationPrompts
-    from src.prompts.website.website_metrics import WebsiteMetricsPrompts
-
-    # Use the get_prompts method to get all prompts from the classes
-    infra_analyze_prompts = InfrastructureAnalyzePrompts.get_prompts()
-    infra_metrics_prompts = InfrastructureMetricsPrompts.get_prompts()
-    infra_catalog_prompts = InfrastructureCatalogPrompts.get_prompts()
-    infra_topology_prompts = InfrastructureTopologyPrompts.get_prompts()
-    infra_resources_prompts = InfrastructureResourcesPrompts.get_prompts()
-    app_resources_prompts = ApplicationResourcesPrompts.get_prompts()
+    try:
+        from src.prompts.application.application_alerts import ApplicationAlertsPrompts
+        from src.prompts.application.application_metrics import (
+            ApplicationMetricsPrompts,
+        )
+        from src.prompts.application.application_resources import (
+            ApplicationResourcesPrompts,
+        )
+        from src.prompts.application.application_settings import (
+            ApplicationSettingsPrompts,
+        )
+        from src.prompts.application.application_topology import (
+            ApplicationTopologyPrompts,
+        )
+        from src.prompts.events.events_tools import EventsPrompts
+        from src.prompts.settings.custom_dashboard import CustomDashboardPrompts
+        from src.prompts.website.website_analyze import WebsiteAnalyzePrompts
+        from src.prompts.website.website_catalog import WebsiteCatalogPrompts
+        from src.prompts.website.website_configuration import (
+            WebsiteConfigurationPrompts,
+        )
+        from src.prompts.website.website_metrics import WebsiteMetricsPrompts
+    except ImportError as e:
+        logger.warning(f"Could not import prompt classes: {e}")
+        return {}
+
+    # Get prompts from each class
+    app_alerts_prompts = ApplicationAlertsPrompts.get_prompts()
     app_metrics_prompts = ApplicationMetricsPrompts.get_prompts()
-    app_catalog_prompts = ApplicationCatalogPrompts.get_prompts()
+    app_resources_prompts = ApplicationResourcesPrompts.get_prompts()
     app_settings_prompts = ApplicationSettingsPrompts.get_prompts()
     app_topology_prompts = ApplicationTopologyPrompts.get_prompts()
-    app_alert_prompts = ApplicationAlertsPrompts.get_prompts()
-    website_metrics_prompts = WebsiteMetricsPrompts.get_prompts()
-    website_catalog_prompts = WebsiteCatalogPrompts.get_prompts()
+    events_prompts = EventsPrompts.get_prompts()
+    custom_dashboard_prompts = CustomDashboardPrompts.get_prompts()
     website_analyze_prompts = WebsiteAnalyzePrompts.get_prompts()
+    website_catalog_prompts = WebsiteCatalogPrompts.get_prompts()
     website_configuration_prompts = WebsiteConfigurationPrompts.get_prompts()
-    custom_dashboard_prompts = CustomDashboardPrompts.get_prompts()
+    website_metrics_prompts = WebsiteMetricsPrompts.get_prompts()
 
-    # Return the categories with their prompt groups
     return {
-        "infra": [
-            ('infra_resources_prompts', infra_resources_prompts),
-            ('infra_catalog_prompts', infra_catalog_prompts),
-            ('infra_topology_prompts', infra_topology_prompts),
-            ('infra_analyze_prompts', infra_analyze_prompts),
-            ('infra_metrics_prompts', infra_metrics_prompts),
-        ],
         "app": [
-            ('app_resources_prompts', app_resources_prompts),
-            ('app_metrics_prompts', app_metrics_prompts),
-            ('app_catalog_prompts', app_catalog_prompts),
-            ('app_settings_prompts', app_settings_prompts),
-            ('app_topology_prompts', app_topology_prompts),
-            ('app_alert_prompts', app_alert_prompts),
+            ("Application Alerts", app_alerts_prompts),
+            ("Application Metrics", app_metrics_prompts),
+            ("Application Resources", app_resources_prompts),
+            ("Application Settings", app_settings_prompts),
+            ("Application Topology", app_topology_prompts),
+        ],
+        "events": [
+            ("Events Tools", events_prompts),
         ],
         "website": [
-            ('website_metrics_prompts', website_metrics_prompts),
-            ('website_catalog_prompts', website_catalog_prompts),
-            ('website_analyze_prompts', website_analyze_prompts),
-            ('website_configuration_prompts', website_configuration_prompts),
+            ("Website Analyze", website_analyze_prompts),
+            ("Website Catalog", website_catalog_prompts),
+            ("Website Configuration", website_configuration_prompts),
+            ("Website Metrics", website_metrics_prompts),
         ],
         "settings": [
-            ('custom_dashboard_prompts', custom_dashboard_prompts),
-        ],
+            ("Custom Dashboard", custom_dashboard_prompts),
+        ]
     }
 
 def get_enabled_client_configs(enabled_categories: str):
@@ -404,6 +368,7 @@ def get_enabled_client_configs(enabled_categories: str):
             logger.warning(f"Unknown category '{category}'")
     return enabled_configs
 
+@workflow(name="instana_mcp_workflow")
 def main():
     """Main entry point for the MCP server."""
     try:
@@ -438,7 +403,7 @@ def main():
             "--tools",
             type=str,
             metavar='<categories>',
-            help="Comma-separated list of tool categories to enable (--tools infra,app,events,automation,website, settings). Also controls which prompts are enabled. If not provided, all tools and prompts are enabled."
+            help="Comma-separated list of tool categories to enable (--tools router,infra,app,events,automation,website,settings). Also controls which prompts are enabled. If not provided, all tools and prompts are enabled. Use 'router' for smart routing across app and infra metrics."
         )
         parser.add_argument(
             "--list-tools",
@@ -451,6 +416,16 @@ def main():
             default=int(os.getenv("PORT", "8080")),
             help="Port to listen on (default: 8080, can be overridden with PORT env var)"
         )
+        parser.add_argument(
+            "--api-token",
+            type=str,
+            help="Instana API token (overrides INSTANA_API_TOKEN env var)"
+        )
+        parser.add_argument(
+            "--base-url",
+            type=str,
+            help="Instana base URL (overrides INSTANA_BASE_URL env var)"
+        )
         # Check for help arguments before parsing
         if len(sys.argv) > 1 and any(arg in ['-h','--h','--help','-help'] for arg in sys.argv[1:]):
             # Check if help is combined with other arguments
@@ -485,7 +460,7 @@ def main():
         else:
             set_log_level(args.log_level)
 
-        all_categories = {"infra", "app", "events", "automation", "website", "settings"}
+        all_categories = {"router", "infra", "app", "events", "automation", "website", "settings"}
 
         # Handle --list-tools option
         if args.list_tools:
@@ -513,7 +488,7 @@ def main():
                 enabled = set(all_categories)
 
         if invalid:
-            logger.error(f"Error: Unknown category/categories: {', '.join(invalid)}. Available categories: infra, app, events, automation, website, settings")
+            logger.error(f"Error: Unknown category/categories: {', '.join(invalid)}. Available categories: router, infra, app, events, automation, website, settings")
             sys.exit(2)
 
         # Print enabled tools for user information
@@ -536,8 +511,9 @@ def main():
                 f"Total enabled tools: {len(enabled_tool_classes)}"
             )
 
-        # Get credentials from environment variables for stdio mode
-        INSTANA_API_TOKEN, INSTANA_BASE_URL = get_instana_credentials()
+        # Get credentials from command line args or environment variables
+        INSTANA_API_TOKEN = args.api_token if args.api_token else os.getenv("INSTANA_API_TOKEN", "")
+        INSTANA_BASE_URL = args.base_url if args.base_url else os.getenv("INSTANA_BASE_URL", "")
 
         if args.transport == "stdio" or args.transport is None:
             if not validate_credentials(INSTANA_API_TOKEN, INSTANA_BASE_URL):