mcp-instana 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

src/{client/infrastructure_analyze_mcp_tools.py → infrastructure/infrastructure_analyze.py}

@@ -4,26 +4,39 @@ Infrastructure Analyze MCP Tools Module
 This module provides infrastructure analysis-specific MCP tools for Instana monitoring.
 """
 
+import logging
 import sys
-import traceback
-from typing import Dict, Any, Optional, List, Union
-from datetime import datetime
+from typing import Any, Dict, Optional, Union
 
 # Import the necessary classes from the SDK
 try:
-    from instana_client.api.infrastructure_analyze_api import InfrastructureAnalyzeApi
+    from instana_client.api.infrastructure_analyze_api import (
+        InfrastructureAnalyzeApi,
+    )
     from instana_client.api_client import ApiClient
     from instana_client.configuration import Configuration
-    from instana_client.models.get_available_metrics_query import GetAvailableMetricsQuery
-    from instana_client.models.get_available_plugins_query import GetAvailablePluginsQuery
-    from instana_client.models.get_infrastructure_query import GetInfrastructureQuery
-    from instana_client.models.get_infrastructure_groups_query import GetInfrastructureGroupsQuery
+    from instana_client.models.get_available_metrics_query import (
+        GetAvailableMetricsQuery,
+    )
+    from instana_client.models.get_available_plugins_query import (
+        GetAvailablePluginsQuery,
+    )
+    from instana_client.models.get_infrastructure_groups_query import (
+        GetInfrastructureGroupsQuery,
+    )
+    from instana_client.models.get_infrastructure_query import (
+        GetInfrastructureQuery,
+    )
 except ImportError as e:
-    print(f"Error importing Instana SDK: {e}", file=sys.stderr)
-    traceback.print_exc(file=sys.stderr)
+    import logging
+    logger = logging.getLogger(__name__)
+    logger.error(f"Error importing Instana SDK: {e}", exc_info=True)
     raise
 
-from .instana_client_base import BaseInstanaClient, register_as_tool
+from src.core.utils import BaseInstanaClient, register_as_tool, with_header_auth
+
+# Configure logger for this module
+logger = logging.getLogger(__name__)
 
 # Helper function for debug printing
 def debug_print(*args, **kwargs):
@@ -32,39 +45,22 @@ def debug_print(*args, **kwargs):
 
 class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
     """Tools for infrastructure analysis in Instana MCP."""
-    
+
     def __init__(self, read_token: str, base_url: str):
         """Initialize the Infrastructure Analyze MCP tools client."""
         super().__init__(read_token=read_token, base_url=base_url)
-        
-        try:
-            
-            # Configure the API client with the correct base URL and authentication
-            configuration = Configuration()
-            configuration.host = base_url
-            configuration.api_key['ApiKeyAuth'] = read_token
-            configuration.api_key_prefix['ApiKeyAuth'] = 'apiToken'
-            
-            # Create an API client with this configuration
-            api_client = ApiClient(configuration=configuration)
-            
-            # Initialize the Instana SDK's InfrastructureAnalyzeApi with our configured client
-            self.analyze_api = InfrastructureAnalyzeApi(api_client=api_client)
-        except Exception as e:
-            debug_print(f"Error initializing InfrastructureAnalyzeApi: {e}")
-            traceback.print_exc(file=sys.stderr)
-            raise
-    
+
     @register_as_tool
-    async def get_available_metrics(self, 
-                                payload: Optional[Union[Dict[str, Any], str]] = None,
-                                ctx=None) -> Dict[str, Any]:
+    @with_header_auth(InfrastructureAnalyzeApi)
+    async def get_available_metrics(self,
+                                    payload: Optional[Union[Dict[str, Any], str]] = None,
+                                    ctx=None, api_client=None) -> Dict[str, Any]:
         """
         Get available metrics for infrastructure monitoring.
-        
+
         This tool retrieves information about available metrics for a specific entity type.
         You can use this to discover what metrics are available for monitoring different components in your environment.
-        
+
         Sample payload:
         {
             "timeFrame": {
@@ -80,68 +76,71 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
             "query": "",
             "type": "jvmRuntimePlatform"
         }
-        
+
         Args:
             payload: Complete request payload as a dictionary or a JSON string
             ctx: The MCP context (optional)
-                
+
         Returns:
             Dictionary containing available metrics or error information
         """
         try:
-            debug_print(f"get_available_metrics called with payload={payload}")
-            
+            logger.debug(f"get_available_metrics called with payload={payload}")
+
             # If payload is a string, try to parse it as JSON
             if isinstance(payload, str):
-                debug_print(f"Payload is a string, attempting to parse")
+                logger.debug("Payload is a string, attempting to parse")
                 try:
                     import json
                     try:
                         parsed_payload = json.loads(payload)
-                        debug_print(f"Successfully parsed payload as JSON")
+                        logger.debug("Successfully parsed payload as JSON")
                         request_body = parsed_payload
                     except json.JSONDecodeError as e:
-                        debug_print(f"JSON parsing failed: {e}, trying with quotes replaced")
-                        
+                        logger.error(f"JSON parsing failed: {e}, trying with quotes replaced")
+
                         # Try replacing single quotes with double quotes
                         fixed_payload = payload.replace("'", "\"")
                         try:
                             parsed_payload = json.loads(fixed_payload)
-                            debug_print(f"Successfully parsed fixed JSON")
+                            logger.debug("Successfully parsed fixed JSON")
                             request_body = parsed_payload
                         except json.JSONDecodeError:
                             # Try as Python literal
                             import ast
                             try:
                                 parsed_payload = ast.literal_eval(payload)
-                                debug_print(f"Successfully parsed payload as Python literal")
+                                logger.debug("Successfully parsed payload as Python literal")
                                 request_body = parsed_payload
                             except (SyntaxError, ValueError) as e2:
-                                debug_print(f"Failed to parse payload string: {e2}")
+                                logger.error(f"Failed to parse payload string: {e2}")
                                 return {"error": f"Invalid payload format: {e2}", "payload": payload}
                 except Exception as e:
-                    debug_print(f"Error parsing payload string: {e}")
+                    logger.error(f"Error parsing payload string: {e}")
                     return {"error": f"Failed to parse payload: {e}", "payload": payload}
             else:
                 # If payload is already a dictionary, use it directly
-                debug_print(f"Using provided payload dictionary")
+                logger.debug("Using provided payload dictionary")
                 request_body = payload
-            
-            debug_print(f"Final request body: {request_body}")
-            
+
+            logger.debug(f"Final request body: {request_body}")
+
             # Import the GetAvailableMetricsQuery class
             try:
-                from instana_client.models.get_available_metrics_query import GetAvailableMetricsQuery
-                debug_print("Successfully imported GetAvailableMetricsQuery")
+                from instana_client.models.get_available_metrics_query import (
+                    GetAvailableMetricsQuery,
+                )
+                logger.debug("Successfully imported GetAvailableMetricsQuery")
             except ImportError as e:
-                debug_print(f"Error importing GetAvailableMetricsQuery: {e}")
-                return {"error": f"Failed to import GetAvailableMetricsQuery: {str(e)}"}
-            
+                error_msg = f"Error importing GetAvailableMetricsQuery: {e!s}"
+                logger.error(error_msg)
+                return {"error": error_msg}
+
             # Create a GetAvailableMetricsQuery object from the request body
             try:
                 # Extract parameters from the request body
                 query_params = {}
-                
+
                 # Handle timeFrame
                 if request_body and "timeFrame" in request_body:
                     time_frame = {}
@@ -152,54 +151,54 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
                     if "windowSize" in request_body["timeFrame"]:
                         time_frame["windowSize"] = request_body["timeFrame"]["windowSize"]
                     query_params["timeFrame"] = time_frame
-                
+
                 # Handle other parameters
                 if request_body and "query" in request_body:
                     query_params["query"] = request_body["query"]
-                
+
                 if request_body and "type" in request_body:
                     query_params["type"] = request_body["type"]
-                    
+
                 if request_body and "tagFilterExpression" in request_body:
                     query_params["tagFilterExpression"] = request_body["tagFilterExpression"]
-                
-                debug_print(f"Creating GetAvailableMetricsQuery with params: {query_params}")
+
+                logger.debug(f"Creating GetAvailableMetricsQuery with params: {query_params}")
                 query_object = GetAvailableMetricsQuery(**query_params)
-                debug_print(f"Successfully created query object: {query_object}")
+                logger.debug(f"Successfully created query object: {query_object}")
             except Exception as e:
-                debug_print(f"Error creating GetAvailableMetricsQuery: {e}")
-                return {"error": f"Failed to create query object: {str(e)}"}
-            
+                logger.error(f"Error creating GetAvailableMetricsQuery: {e}")
+                return {"error": f"Failed to create query object: {e!s}"}
+
             # Call the get_available_metrics method from the SDK with the query object
-            debug_print("Calling get_available_metrics with query object")
-            result = self.analyze_api.get_available_metrics(get_available_metrics_query=query_object)
-            
+            logger.debug("Calling get_available_metrics with query object")
+            result = api_client.get_available_metrics(get_available_metrics_query=query_object)
+
             # Convert the result to a dictionary
             if hasattr(result, 'to_dict'):
                 result_dict = result.to_dict()
             else:
                 # If it's already a dict or another format, use it as is
                 result_dict = result
-                
-            debug_print(f"Result from get_available_metrics: {result_dict}")
+
+            logger.debug(f"Result from get_available_metrics: {result_dict}")
             return result_dict
         except Exception as e:
-            debug_print(f"Error in get_available_metrics: {e}")
-            traceback.print_exc(file=sys.stderr)
-            return {"error": f"Failed to get available metrics: {str(e)}"}
+            logger.error(f"Error in get_available_metrics: {e}", exc_info=True)
+            return {"error": f"Failed to get available metrics: {e!s}"}
 
     @register_as_tool
-    async def get_entities(self, 
-                        payload: Optional[Union[Dict[str, Any], str]] = None,
-                        ctx=None) -> Dict[str, Any]:
+    @with_header_auth(InfrastructureAnalyzeApi)
+    async def get_entities(self,
+                           payload: Optional[Union[Dict[str, Any], str]] = None,
+                           ctx=None, api_client=None) -> Dict[str, Any]:
         """
         Get infrastructure entities for a given entity type along with requested metrics.
-        
+
         We want to know the memory used and no of blocked threads for entity named JVMruntimeplatform for last 1 hour in Instana. can you help us get the details ?
-        
+
         This tool retrieves entities of a specific type (e.g., hosts, processes, containers) along with
         their metrics. You can filter the results using tag filters and paginate through large result sets.
-        
+
         Sample payload:
         {
             "tagFilterExpression": {
@@ -224,95 +223,95 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
                 {"metric": "threads.blocked", "granularity": 600000, "aggregation": "MEAN"}
             ]
         }
-        
+
         Args:
             payload: Complete request payload as a dictionary or a JSON string
             ctx: The MCP context (optional)
-                
+
         Returns:
             Dictionary containing infrastructure entities and their metrics or error information
         """
         try:
-            debug_print(f"get_entities called with payload={payload}")
-            
+            logger.debug(f"get_entities called with payload={payload}")
+
             # If payload is a string, try to parse it as JSON
             if isinstance(payload, str):
-                debug_print(f"Payload is a string, attempting to parse")
+                logger.debug("Payload is a string, attempting to parse")
                 try:
                     import json
                     try:
                         parsed_payload = json.loads(payload)
-                        debug_print(f"Successfully parsed payload as JSON")
+                        logger.debug("Successfully parsed payload as JSON")
                         request_body = parsed_payload
                     except json.JSONDecodeError as e:
-                        debug_print(f"JSON parsing failed: {e}, trying with quotes replaced")
-                        
+                        logger.warning(f"JSON parsing failed: {e}, trying with quotes replaced")
+
                         # Try replacing single quotes with double quotes
                         fixed_payload = payload.replace("'", "\"")
                         try:
                             parsed_payload = json.loads(fixed_payload)
-                            debug_print(f"Successfully parsed fixed JSON")
+                            logger.debug("Successfully parsed fixed JSON")
                             request_body = parsed_payload
                         except json.JSONDecodeError:
                             # Try as Python literal
                             import ast
                             try:
                                 parsed_payload = ast.literal_eval(payload)
-                                debug_print(f"Successfully parsed payload as Python literal")
+                                logger.debug("Successfully parsed payload as Python literal")
                                 request_body = parsed_payload
                             except (SyntaxError, ValueError) as e2:
-                                debug_print(f"Failed to parse payload string: {e2}")
+                                logger.error(f"Failed to parse payload string: {e2}")
                                 return {"error": f"Invalid payload format: {e2}", "payload": payload}
                 except Exception as e:
-                    debug_print(f"Error parsing payload string: {e}")
+                    logger.error(f"Error parsing payload string: {e}")
                     return {"error": f"Failed to parse payload: {e}", "payload": payload}
             else:
                 # If payload is already a dictionary, use it directly
-                debug_print(f"Using provided payload dictionary")
+                logger.debug("Using provided payload dictionary")
                 request_body = payload
-            
-            debug_print(f"Final request body: {request_body}")
-            
+
+            logger.debug(f"Final request body: {request_body}")
+
             # Create the GetInfrastructureQuery object
             try:
                 # Create the query object directly from the request body
-                get_infra_query = GetInfrastructureQuery(**request_body)
-                debug_print("Successfully created GetInfrastructureQuery object")
+                get_infra_query = GetInfrastructureQuery(**request_body) #type: ignore
+                logger.debug("Successfully created GetInfrastructureQuery object")
             except Exception as model_error:
                 error_msg = f"Failed to create GetInfrastructureQuery object: {model_error}"
-                debug_print(error_msg)
+                logger.error(error_msg)
                 return {"error": error_msg, "request_body": request_body}
-            
+
             # Call the get_entities method from the SDK
-            debug_print("Calling API method get_entities")
-            result = self.analyze_api.get_entities(
+            logger.debug("Calling API method get_entities")
+            result = api_client.get_entities(
                 get_infrastructure_query=get_infra_query
             )
-            
+
             # Convert the result to a dictionary
             if hasattr(result, 'to_dict'):
                 result_dict = result.to_dict()
             else:
                 # If it's already a dict or another format, use it as is
                 result_dict = result
-                
-            debug_print(f"Result from get_entities: {result_dict}")
+
+            logger.debug(f"Result from get_entities: {result_dict}")
             return result_dict
         except Exception as e:
-            debug_print(f"Error in get_entities: {e}")
-            traceback.print_exc(file=sys.stderr)
-            return {"error": f"Failed to get entities: {str(e)}"}
+            logger.error(f"Error in get_entities: {e}", exc_info=True)
+            return {"error": f"Failed to get entities: {e!s}"}
 
     @register_as_tool
-    async def get_aggregated_entity_groups(self, 
-                                        payload: Optional[Union[Dict[str, Any], str]] = None,
-                                        ctx=None) -> Dict[str, Any]:
+    @with_header_auth(InfrastructureAnalyzeApi)
+    async def get_aggregated_entity_groups(self,
+                                           payload: Optional[Union[Dict[str, Any], str]] = None,
+                                           ctx=None, api_client=None) -> Dict[str, Any]:
         """
         Get grouped infrastructure entities with aggregated metrics.
-        
+
         This tool groups entities of a specific type by specified tags and aggregates metrics for these groups.
         For example, you can group hosts by their region and get average CPU usage per region.
-        
+
         Sample payload:
         {
             "timeFrame": {
@@ -340,115 +339,116 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
                 "direction": "ASC"
             }
         }
-        
+
         Args:
             payload: Complete request payload as a dictionary or a JSON string
             ctx: The MCP context (optional)
-                
+
         Returns:
             Dictionary containing grouped entities and their aggregated metrics or error information
         """
         try:
-            debug_print(f"get_aggregated_entity_groups called with payload={payload}")
-            
+            logger.debug(f"get_aggregated_entity_groups called with payload={payload}")
+
             # If no payload is provided, return an error
             if not payload:
                 return {"error": "Payload is required for this operation"}
-            
+
             # If payload is a string, try to parse it as JSON
             if isinstance(payload, str):
-                debug_print(f"Payload is a string, attempting to parse")
+                logger.debug("Payload is a string, attempting to parse")
                 try:
                     import json
                     try:
                         parsed_payload = json.loads(payload)
-                        debug_print(f"Successfully parsed payload as JSON")
+                        logger.debug("Successfully parsed payload as JSON")
                         request_body = parsed_payload
                     except json.JSONDecodeError as e:
-                        debug_print(f"JSON parsing failed: {e}, trying with quotes replaced")
-                        
+                        logger.warning(f"JSON parsing failed: {e}, trying with quotes replaced")
+
                         # Try replacing single quotes with double quotes
                         fixed_payload = payload.replace("'", "\"")
                         try:
                             parsed_payload = json.loads(fixed_payload)
-                            debug_print(f"Successfully parsed fixed JSON")
+                            logger.debug("Successfully parsed fixed JSON")
                             request_body = parsed_payload
                         except json.JSONDecodeError:
                             # Try as Python literal
                             import ast
                             try:
                                 parsed_payload = ast.literal_eval(payload)
-                                debug_print(f"Successfully parsed payload as Python literal")
+                                logger.debug("Successfully parsed payload as Python literal")
                                 request_body = parsed_payload
                             except (SyntaxError, ValueError) as e2:
-                                debug_print(f"Failed to parse payload string: {e2}")
+                                logger.error(f"Failed to parse payload string: {e2}")
                                 return {"error": f"Invalid payload format: {e2}", "payload": payload}
                 except Exception as e:
-                    debug_print(f"Error parsing payload string: {e}")
+                    logger.error(f"Error parsing payload string: {e}")
                     return {"error": f"Failed to parse payload: {e}", "payload": payload}
             else:
                 # If payload is already a dictionary, use it directly
-                debug_print(f"Using provided payload dictionary")
+                logger.debug("Using provided payload dictionary")
                 request_body = payload
-            
-            debug_print(f"Final request body: {request_body}")
-            
+
+            logger.debug(f"Final request body: {request_body}")
+
             # Create the GetInfrastructureGroupsQuery object
             try:
                 # Import the model class
-                from instana_client.models.get_infrastructure_groups_query import GetInfrastructureGroupsQuery
-                
+                from instana_client.models.get_infrastructure_groups_query import (
+                    GetInfrastructureGroupsQuery,
+                )
+
                 # Create the query object
                 get_groups_query = GetInfrastructureGroupsQuery(**request_body)
-                debug_print("Successfully created GetInfrastructureGroupsQuery object")
+                logger.debug("Successfully created GetInfrastructureGroupsQuery object")
             except Exception as model_error:
                 error_msg = f"Failed to create GetInfrastructureGroupsQuery object: {model_error}"
-                debug_print(error_msg)
+                logger.debug(error_msg)
                 return {"error": error_msg, "request_body": request_body}
-            
+
             # Call the get_entity_groups method from the SDK
-            debug_print("Calling API method get_entity_groups")
+            logger.debug("Calling API method get_entity_groups")
             try:
                 # Use the without_preload_content version to get the raw response
-                response = self.analyze_api.get_entity_groups_without_preload_content(
+                response = api_client.get_entity_groups_without_preload_content(
                     get_infrastructure_groups_query=get_groups_query
                 )
-                
+
                 # Check if the response was successful
                 if response.status != 200:
                     error_message = f"Failed to get entity groups: HTTP {response.status}"
-                    debug_print(error_message)
+                    logger.debug(error_message)
                     return {"error": error_message}
-                
+
                 # Read the response content
                 response_text = response.data.decode('utf-8')
-                
+
                 # Parse the response as JSON
                 import json
                 result_dict = json.loads(response_text)
-                
-                debug_print(f"Successfully parsed raw response")
-                
+
+                logger.debug("Successfully parsed raw response")
+
                 # Create a summarized version of the results
                 return self._summarize_entity_groups_result(result_dict, request_body)
             except Exception as api_error:
                 error_msg = f"API call failed: {api_error}"
-                debug_print(error_msg)
+                logger.error(error_msg)
                 return {"error": error_msg}
-                
+
         except Exception as e:
-            debug_print(f"Error in get_aggregated_entity_groups: {e}")
-            traceback.print_exc(file=sys.stderr)
-            return {"error": f"Failed to get aggregated entity groups: {str(e)}"}
+            logger.error(f"Error in get_aggregated_entity_groups: {e}", exc_info=True)
+            return {"error": f"Failed to get aggregated entity groups: {e!s}"}
 
     def _summarize_entity_groups_result(self, result_dict, query_body):
         """
         Create a summarized version of the entity groups result.
-        
+
         Args:
             result_dict: The full API response
             query_body: The query body used to make the request
-            
+
         Returns:
             A summarized version of the results
         """
@@ -456,15 +456,15 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
             # Check if there's an error in the result
             if isinstance(result_dict, dict) and "error" in result_dict:
                 return result_dict
-            
+
             # Extract the group by tag
             group_by_tag = None
             if "groupBy" in query_body and isinstance(query_body["groupBy"], list) and len(query_body["groupBy"]) > 0:
                 group_by_tag = query_body["groupBy"][0]
-            
+
             # Extract host names if available
             host_names = []
-            
+
             # Process each item in the results
             if "items" in result_dict and isinstance(result_dict["items"], list):
                 for item in result_dict["items"]:
@@ -480,39 +480,39 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
                         else:
                             # Convert other types to string
                             host_name = str(tag_value)
-                        
+
                         if host_name not in host_names:
                             host_names.append(host_name)
-            
+
             # Sort host names alphabetically
             host_names.sort()
-            
+
             # Create the exact format requested
             summary = {
                 "hosts": host_names,
                 "count": len(host_names),
                 "summary": f"Found {len(host_names)} hosts: {', '.join(host_names)}"
             }
-            
+
             return summary
         except Exception as e:
-            debug_print(f"Error in _summarize_entity_groups_result: {e}")
-            traceback.print_exc(file=sys.stderr)
+            logger.error(f"Error in _summarize_entity_groups_result: {e}", exc_info=True)
             # If summarization fails, return an error message
             return {
-                "error": f"Failed to summarize results: {str(e)}"
+                "error": f"Failed to summarize results: {e!s}"
             }
 
     @register_as_tool
-    async def get_available_plugins(self, 
-                                payload: Optional[Union[Dict[str, Any], str]] = None,
-                                ctx=None) -> Dict[str, Any]:
+    @with_header_auth(InfrastructureAnalyzeApi)
+    async def get_available_plugins(self,
+                                    payload: Optional[Union[Dict[str, Any], str]] = None,
+                                    ctx=None, api_client=None) -> Dict[str, Any]:
         """
         Get available plugins for infrastructure monitoring.
-        
+
         This tool retrieves information about available plugins for infrastructure monitoring.
         You can use this to discover what types of entities can be monitored in your environment.
-        
+
         Sample payload:
         {
             "timeFrame": {
@@ -527,68 +527,70 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
             "query": "java",
             "offline": false
         }
-        
+
         Args:
             payload: Complete request payload as a dictionary or a JSON string
             ctx: The MCP context (optional)
-                
+
         Returns:
             Dictionary containing available plugins or error information
         """
         try:
-            debug_print(f"get_available_plugins called with payload={payload}")
-            
+            logger.debug(f"get_available_plugins called with payload={payload}")
+
             # If payload is a string, try to parse it as JSON
             if isinstance(payload, str):
-                debug_print(f"Payload is a string, attempting to parse")
+                logger.debug("Payload is a string, attempting to parse")
                 try:
                     import json
                     try:
                         parsed_payload = json.loads(payload)
-                        debug_print(f"Successfully parsed payload as JSON")
+                        logger.debug("Successfully parsed payload as JSON")
                         request_body = parsed_payload
                     except json.JSONDecodeError as e:
-                        debug_print(f"JSON parsing failed: {e}, trying with quotes replaced")
-                        
+                        logger.warning(f"JSON parsing failed: {e}, trying with quotes replaced")
+
                         # Try replacing single quotes with double quotes
                         fixed_payload = payload.replace("'", "\"")
                         try:
                             parsed_payload = json.loads(fixed_payload)
-                            debug_print(f"Successfully parsed fixed JSON")
+                            logger.debug("Successfully parsed fixed JSON")
                             request_body = parsed_payload
                         except json.JSONDecodeError:
                             # Try as Python literal
                             import ast
                             try:
                                 parsed_payload = ast.literal_eval(payload)
-                                debug_print(f"Successfully parsed payload as Python literal")
+                                logger.debug("Successfully parsed payload as Python literal")
                                 request_body = parsed_payload
                             except (SyntaxError, ValueError) as e2:
-                                debug_print(f"Failed to parse payload string: {e2}")
+                                logger.error(f"Failed to parse payload string: {e2}")
                                 return {"error": f"Invalid payload format: {e2}", "payload": payload}
                 except Exception as e:
-                    debug_print(f"Error parsing payload string: {e}")
+                    logger.error(f"Error parsing payload string: {e}")
                     return {"error": f"Failed to parse payload: {e}", "payload": payload}
             else:
                 # If payload is already a dictionary, use it directly
-                debug_print(f"Using provided payload dictionary")
+                logger.debug("Using provided payload dictionary")
                 request_body = payload
-            
-            debug_print(f"Final request body: {request_body}")
-            
+
+            logger.debug(f"Final request body: {request_body}")
+
             # Import the GetAvailablePluginsQuery class
             try:
-                from instana_client.models.get_available_plugins_query import GetAvailablePluginsQuery
-                debug_print("Successfully imported GetAvailablePluginsQuery")
+                from instana_client.models.get_available_plugins_query import (
+                    GetAvailablePluginsQuery,
+                )
+                logger.debug("Successfully imported GetAvailablePluginsQuery")
             except ImportError as e:
-                debug_print(f"Error importing GetAvailablePluginsQuery: {e}")
-                return {"error": f"Failed to import GetAvailablePluginsQuery: {str(e)}"}
-            
+                logger.error(f"Error importing GetAvailablePluginsQuery: {e}")
+                return {"error": f"Failed to import GetAvailablePluginsQuery: {e!s}"}
+
             # Create a GetAvailablePluginsQuery object from the request body
             try:
                 # Extract parameters from the request body
                 query_params = {}
-                
+
                 # Handle timeFrame
                 if request_body and "timeFrame" in request_body:
                     time_frame = {}
@@ -597,38 +599,37 @@ class InfrastructureAnalyzeMCPTools(BaseInstanaClient):
                     if "windowSize" in request_body["timeFrame"]:
                         time_frame["windowSize"] = request_body["timeFrame"]["windowSize"]
                     query_params["timeFrame"] = time_frame
-                
+
                 # Handle other parameters
                 if request_body and "query" in request_body:
                     query_params["query"] = request_body["query"]
-                
+
                 if request_body and "offline" in request_body:
                     query_params["offline"] = request_body["offline"]
-                    
+
                 if request_body and "tagFilterExpression" in request_body:
                     query_params["tagFilterExpression"] = request_body["tagFilterExpression"]
-                
-                debug_print(f"Creating GetAvailablePluginsQuery with params: {query_params}")
+
+                logger.debug(f"Creating GetAvailablePluginsQuery with params: {query_params}")
                 query_object = GetAvailablePluginsQuery(**query_params)
-                debug_print(f"Successfully created query object: {query_object}")
+                logger.debug(f"Successfully created query object: {query_object}")
             except Exception as e:
-                debug_print(f"Error creating GetAvailablePluginsQuery: {e}")
-                return {"error": f"Failed to create query object: {str(e)}"}
-            
+                logger.error(f"Error creating GetAvailablePluginsQuery: {e}")
+                return {"error": f"Failed to create query object: {e!s}"}
+
             # Call the get_available_plugins method from the SDK with the query object
-            debug_print("Calling get_available_plugins with query object")
-            result = self.analyze_api.get_available_plugins(get_available_plugins_query=query_object)
-            
+            logger.debug("Calling get_available_plugins with query object")
+            result = api_client.get_available_plugins(get_available_plugins_query=query_object)
+
             # Convert the result to a dictionary
             if hasattr(result, 'to_dict'):
                 result_dict = result.to_dict()
             else:
                 # If it's already a dict or another format, use it as is
                 result_dict = result
-                
-            debug_print(f"Result from get_available_plugins: {result_dict}")
+
+            logger.debug(f"Result from get_available_plugins: {result_dict}")
             return result_dict
         except Exception as e:
-            debug_print(f"Error in get_available_plugins: {e}")
-            traceback.print_exc(file=sys.stderr)
-            return {"error": f"Failed to get available plugins: {str(e)}"}
+            logger.error(f"Error in get_available_plugins: {e}", exc_info=True)
+            return {"error": f"Failed to get available plugins: {e!s}"}