Fix nested tool calls

Author: cbbayburt

src/mcp_server_uyuni/server.py

@@ -214,7 +214,6 @@ async def get_list_of_active_systems(ctx: Context) -> List[Dict[str, Any]]:
     return await _get_list_of_active_systems(ctx.get_state('token'))
 
 async def _get_list_of_active_systems(token: str) -> List[Dict[str, Union[str, int]]]:
-
     async with httpx.AsyncClient(verify=UYUNI_MCP_SSL_VERIFY) as client:
         systems_data_result = await _call_uyuni_api(
             client=client,
@@ -452,6 +451,9 @@ async def check_system_updates(system_identifier: Union[str, int], ctx: Context)
     log_string = f"Checking pending updates for system {system_identifier}"
     logger.info(log_string)
     await ctx.info(log_string)
+    return await _check_system_updates(system_identifier, ctx)
+
+async def _check_system_updates(system_identifier: Union[str, int], ctx: Context) -> Dict[str, Any]:
     token = ctx.get_state('token')
     system_id = await _resolve_system_id(system_identifier, token)
     default_error_response = {
@@ -492,7 +494,7 @@ async def check_system_updates(system_identifier: Union[str, int], ctx: Context)
             unscheduled_errata_call
         )
         relevant_updates_list, unscheduled_updates_list = results
-        
+
         if not isinstance(relevant_updates_list, list) or not isinstance(unscheduled_updates_list, list):
             logger.error(
                 f"API calls for system {system_id} did not return lists as expected. "
@@ -521,7 +523,7 @@ async def check_system_updates(system_identifier: Union[str, int], ctx: Context)
                 update_details['application_status'] = 'Pending'
             else:
                 update_details['application_status'] = 'Queued'
-            
+
             # Initialize and fetch CVEs
             update_details['cves'] = []
             if advisory_name:
@@ -575,7 +577,7 @@ async def check_all_systems_for_updates(ctx: Context) -> List[Dict[str, Any]]:
     await ctx.info(log_string)
 
     systems_with_updates = []
-    active_systems = await get_list_of_active_systems(ctx) # Get the list of all systems
+    active_systems = await _get_list_of_active_systems(ctx.get_state('token')) # Get the list of all systems
 
     if not active_systems:
         print("Warning: No active systems found or failed to retrieve system list.")
@@ -593,7 +595,7 @@ async def check_all_systems_for_updates(ctx: Context) -> List[Dict[str, Any]]:
 
         print(f"Checking updates for system: {system_name} (ID: {system_id})")
         # Use the existing check_system_updates tool
-        update_check_result = await check_system_updates(system_id, ctx)
+        update_check_result = await _check_system_updates(system_id, ctx)
 
         if update_check_result.get('has_pending_updates', False):
             # If the system has updates, add its info and update details to the result list
@@ -637,7 +639,7 @@ async def schedule_apply_pending_updates_to_system(system_identifier: Union[str,
     token = ctx.get_state('token')
 
     # 1. Use check_system_updates to get relevant errata
-    update_info = await check_system_updates(system_identifier, ctx)
+    update_info = await _check_system_updates(system_identifier, ctx)
 
     if not update_info or not update_info.get('has_pending_updates'):
         print(f"No pending updates found for system {system_identifier}, or an error occurred while fetching update information.")