fix: Add type annotations to CLI functions

Co-Authored-By: Aaron <AJ> Steers <[email protected]>

Author: devin-ai-integration[bot]

airbyte_cdk/cli/commands/cli.py

@@ -6,7 +6,7 @@
 
 
 @click.group()
-def cli():
+def cli() -> None:
     """Airbyte CDK command-line interface."""
     pass
 

airbyte_cdk/cli/commands/image.py

@@ -17,59 +17,51 @@
 
 
 @click.group()
-def image():
+def image() -> None:
     """Commands for working with connector Docker images."""
     pass
 
 
 @image.command()
-@click.argument("connector_dir", type=click.Path(exists=True, file_okay=False, dir_okay=True, path_type=Path))
-@click.option(
-    "--tag", 
-    default="dev", 
-    help="Tag to apply to the built image (default: dev)"
+@click.argument(
+    "connector_dir", type=click.Path(exists=True, file_okay=False, dir_okay=True, path_type=Path)
 )
+@click.option("--tag", default="dev", help="Tag to apply to the built image (default: dev)")
 @click.option(
     "--platform",
     type=click.Choice(["linux/amd64", "linux/arm64"]),
     default="linux/amd64",
     help="Platform to build for (default: linux/amd64)",
 )
-@click.option(
-    "--no-verify", 
-    is_flag=True, 
-    help="Skip verification of the built image"
-)
-@click.option(
-    "--verbose", "-v", 
-    is_flag=True, 
-    help="Enable verbose logging"
-)
-def build(connector_dir, tag, platform, no_verify, verbose):
+@click.option("--no-verify", is_flag=True, help="Skip verification of the built image")
+@click.option("--verbose", "-v", is_flag=True, help="Enable verbose logging")
+def build(connector_dir: Path, tag: str, platform: str, no_verify: bool, verbose: bool) -> None:
     """Build a connector Docker image.
 
     This command builds a Docker image for a connector, using either
     the connector's Dockerfile or a base image specified in the metadata.
     """
     set_up_logging(verbose)
-    
+
     if not verify_docker_installation():
-        click.echo("Docker is not installed or not running. Please install Docker and try again.", err=True)
+        click.echo(
+            "Docker is not installed or not running. Please install Docker and try again.", err=True
+        )
         sys.exit(1)
-        
+
     try:
         metadata = read_metadata(connector_dir)
         click.echo(f"Connector: {metadata.dockerRepository}")
         click.echo(f"Version: {metadata.dockerImageTag}")
-        
+
         language = infer_connector_language(metadata, connector_dir)
         click.echo(f"Detected connector language: {language}")
-        
+
         if metadata.connectorBuildOptions and metadata.connectorBuildOptions.baseImage:
             image_name = build_from_base_image(connector_dir, metadata, tag, platform)
         else:
             image_name = build_from_dockerfile(connector_dir, metadata, tag, platform)
-            
+
         if not no_verify:
             if verify_image(image_name):
                 click.echo(f"Build completed successfully: {image_name}")
@@ -80,10 +72,11 @@ def build(connector_dir, tag, platform, no_verify, verbose):
         else:
             click.echo(f"Build completed successfully (without verification): {image_name}")
             sys.exit(0)
-            
+
     except Exception as e:
         click.echo(f"Error: {str(e)}", err=True)
         if verbose:
             import traceback
+
             click.echo(traceback.format_exc(), err=True)
         sys.exit(1)

airbyte_cdk/cli/entrypoint/cli.py

@@ -3,6 +3,6 @@
 from airbyte_cdk.cli.commands.cli import cli
 
 
-def run():
+def run() -> None:
     """Run the Airbyte CDK CLI."""
     cli(prog_name="airbyte-cdk")