code: def create_container_image(
      image: str, base_image: Optional[str]
  ) -> str:
      if image == kubeflow_dag_runner.DEFAULT_KUBEFLOW_TFX_IMAGE:
          sys.exit(
              "Default image for KubeflowDagRunner given and used with "
              "--build-image flag. If you want to use your custom image, please "
              "specify the image name that will be built at the "
              "KubeflowDagRunnerConfig. Otherwise, do not use --build-image "
              "flag."
          )
      built_image = builder.build(
          target_image=image, base_image=base_image
      )
      click.echo(f'New container image "{built_image}" was built.')
      return built_image
  
imports: import functools
  import os
  import sys
  import time
  from typing import Any, Dict, Optional
  import click
  import kfp
  from tfx.orchestration.kubeflow import kubeflow_dag_runner
  from tfx.tools.cli import labels
  from tfx.tools.cli.container_builder import builder
  from tfx.tools.cli.handler import base_handler
  from tfx.tools.cli.handler import kubeflow_dag_runner_patcher
label_desc: Use image node_id 1 for calling a global function with example usage: create_container_image(image, base_image) and returns: built_image