update deploy to retry if deployment failed

Author: jtcorbett

src/mcp_agent/cli/cloud/commands/deploy/main.py

@@ -27,12 +27,14 @@
 from mcp_agent.cli.secrets.processor import (
     process_config_secrets,
 )
+from mcp_agent.cli.utils.retry import retry_async_with_exponential_backoff, RetryError
 from mcp_agent.cli.utils.ux import (
     console,
     print_deployment_header,
     print_error,
     print_info,
     print_success,
+    print_warning,
 )
 
 from .wrangler_wrapper import wrangler_deploy
@@ -84,6 +86,13 @@ def deploy_config(
         help="API key for authentication. Defaults to MCP_API_KEY environment variable.",
         envvar=ENV_API_KEY,
     ),
+    retry_count: int = typer.Option(
+        3,
+        "--retry-count",
+        help="Number of retries on deployment failure.",
+        min=1,
+        max=10,
+    ),
 ) -> str:
     """Deploy an MCP agent using the specified configuration.
 
@@ -101,6 +110,7 @@ def deploy_config(
         non_interactive: Never prompt for reusing or updating secrets or existing apps; reuse existing where possible
         api_url: API base URL
         api_key: API key for authentication
+        retry_count: Number of retries on deployment failure
 
     Returns:
         Newly-deployed MCP App ID
@@ -119,11 +129,13 @@ def deploy_config(
 
         if not effective_api_url:
             raise CLIError(
-                "MCP_API_BASE_URL environment variable or --api-url option must be set."
+                "MCP_API_BASE_URL environment variable or --api-url option must be set.",
+                retriable=False
             )
         if not effective_api_key:
             raise CLIError(
-                "Must be logged in to deploy. Run 'mcp-agent login', set MCP_API_KEY environment variable or specify --api-key option."
+                "Must be logged in to deploy. Run 'mcp-agent login', set MCP_API_KEY environment variable or specify --api-key option.",
+                retriable=False
             )
         print_info(f"Using API at {effective_api_url}")
 
@@ -166,7 +178,8 @@ def deploy_config(
                     )
         except UnauthenticatedError as e:
             raise CLIError(
-                "Invalid API key for deployment. Run 'mcp-agent login' or set MCP_API_KEY environment variable with new API key."
+                "Invalid API key for deployment. Run 'mcp-agent login' or set MCP_API_KEY environment variable with new API key.",
+                retriable=False
             ) from e
         except Exception as e:
             raise CLIError(f"Error checking or creating app: {str(e)}") from e
@@ -237,47 +250,108 @@ def deploy_config(
             )
         )
 
-        wrangler_deploy(
+        # Execute deployment with retry logic
+        app = run_async(_deploy_with_retry(
             app_id=app_id,
             api_key=effective_api_key,
             project_dir=config_dir,
-        )
+            mcp_app_client=mcp_app_client,
+            retry_count=retry_count,
+        ))
+
+        print_info(f"App ID: {app_id}")
+        if app.appServerInfo:
+            status = (
+                "ONLINE"
+                if app.appServerInfo.status == "APP_SERVER_STATUS_ONLINE"
+                else "OFFLINE"
+            )
+            print_info(f"App URL: {app.appServerInfo.serverUrl}")
+            print_info(f"App Status: {status}")
+        return app_id
+
+    except Exception as e:
+        if settings.VERBOSE:
+            import traceback
+
+            typer.echo(traceback.format_exc())
+        raise CLIError(f"Deployment failed: {str(e)}") from e
+
+
+async def _deploy_with_retry(
+    app_id: str,
+    api_key: str,
+    project_dir: Path,
+    mcp_app_client: MCPAppClient,
+    retry_count: int,
+):
+    """Execute the deployment operations with retry logic.
+
+    Args:
+        app_id: The application ID
+        api_key: API key for authentication
+        project_dir: Directory containing the project files
+        mcp_app_client: MCP App client for API calls
+        retry_count: Number of retry attempts
+
+    Returns:
+        Deployed app information
+    """
+    attempt = 0
+
+    async def _perform_deployment():
+        nonlocal attempt
+        attempt += 1
+
+        attempt_suffix = f" (attempt {attempt}/{retry_count})" if retry_count > 1 else ""
 
         with Progress(
             SpinnerColumn(spinner_name="arrow3"),
             TextColumn("[progress.description]{task.description}"),
         ) as progress:
-            task = progress.add_task("Deploying MCP App bundle...", total=None)
-
+            # Bundle with wrangler
+            bundle_task = progress.add_task(f"Bundling MCP Agent{attempt_suffix}...", total=None)
             try:
-                app = run_async(
-                    mcp_app_client.deploy_app(
-                        app_id=app_id,
-                    )
+                wrangler_deploy(
+                    app_id=app_id,
+                    api_key=api_key,
+                    project_dir=project_dir,
                 )
-                progress.update(task, description="✅ MCP App deployed successfully!")
-                print_info(f"App ID: {app_id}")
-
-                if app.appServerInfo:
-                    status = (
-                        "ONLINE"
-                        if app.appServerInfo.status == "APP_SERVER_STATUS_ONLINE"
-                        else "OFFLINE"
-                    )
-                    print_info(f"App URL: {app.appServerInfo.serverUrl}")
-                    print_info(f"App Status: {status}")
-                return app_id
+                progress.update(bundle_task, description=f"✅ Bundled successfully{attempt_suffix}")
+            except Exception as e:
+                progress.update(bundle_task, description=f"❌ Bundling failed{attempt_suffix}")
+                if attempt < retry_count:
+                    print_warning(f"Bundling failed on attempt {attempt}/{retry_count}: {str(e)}")
+                raise
 
+            # Deploy app
+            deploy_task = progress.add_task(f"Deploying MCP App bundle{attempt_suffix}...", total=None)
+            try:
+                app = await mcp_app_client.deploy_app(app_id=app_id)
+                progress.update(deploy_task, description=f"✅ MCP App deployed successfully{attempt_suffix}!")
+                return app
             except Exception as e:
-                progress.update(task, description="❌ Deployment failed")
-                raise e
+                progress.update(deploy_task, description=f"❌ Deployment failed{attempt_suffix}")
+                if attempt < retry_count:
+                    print_warning(f"Deployment failed on attempt {attempt}/{retry_count}: {str(e)}")
+                raise
 
-    except Exception as e:
-        if settings.VERBOSE:
-            import traceback
+    # Execute deployment with retry logic
+    if retry_count > 1:
+        print_info(f"Deployment configured with up to {retry_count} attempts")
 
-            typer.echo(traceback.format_exc())
-        raise CLIError(f"Deployment failed: {str(e)}") from e
+    try:
+        return await retry_async_with_exponential_backoff(
+            _perform_deployment,
+            max_attempts=retry_count,
+            initial_delay=1.0,
+            backoff_multiplier=2.0,
+            max_delay=30.0,
+        )
+    except RetryError as e:
+        attempts_text = "attempts" if retry_count > 1 else "attempt"
+        print_error(f"Deployment failed after {retry_count} {attempts_text}")
+        raise CLIError(f"Deployment failed after {retry_count} {attempts_text}. Last error: {e.original_error}") from e.original_error
 
 
 def get_config_files(config_dir: Path) -> tuple[Path, Optional[Path], Optional[Path]]:
@@ -293,7 +367,8 @@ def get_config_files(config_dir: Path) -> tuple[Path, Optional[Path], Optional[P
     config_file = config_dir / MCP_CONFIG_FILENAME
     if not config_file.exists():
         raise CLIError(
-            f"Configuration file '{MCP_CONFIG_FILENAME}' not found in {config_dir}"
+            f"Configuration file '{MCP_CONFIG_FILENAME}' not found in {config_dir}",
+            retriable=False
         )
 
     secrets_file: Optional[Path] = None