feat(sdk): Enable containerizing v2 Python components

sdk/python/kfp/cli/cli.py

@@ -12,28 +12,43 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import click
 import logging
 import sys
+
+import click
+import typer
+
 from kfp._client import Client
 from kfp.cli.run import run
 from kfp.cli.pipeline import pipeline
 from kfp.cli.diagnose_me_cli import diagnose_me
 from kfp.cli.experiment import experiment
 from kfp.cli.output import OutputFormat
+from kfp.cli import components
+
 
 @click.group()
 @click.option('--endpoint', help='Endpoint of the KFP API service to connect.')
 @click.option('--iap-client-id', help='Client ID for IAP protected endpoint.')
-@click.option('-n', '--namespace', default='kubeflow', show_default=True,
+@click.option('-n',
+              '--namespace',
+              default='kubeflow',
+              show_default=True,
               help='Kubernetes namespace to connect to the KFP API.')
-@click.option('--other-client-id', help='Client ID for IAP protected endpoint to obtain the refresh token.')
-@click.option('--other-client-secret', help='Client ID for IAP protected endpoint to obtain the refresh token.')
-@click.option('--output', type=click.Choice(list(map(lambda x: x.name, OutputFormat))),
-              default=OutputFormat.table.name, show_default=True,
+@click.option(
+    '--other-client-id',
+    help='Client ID for IAP protected endpoint to obtain the refresh token.')
+@click.option(
+    '--other-client-secret',
+    help='Client ID for IAP protected endpoint to obtain the refresh token.')
+@click.option('--output',
+              type=click.Choice(list(map(lambda x: x.name, OutputFormat))),
+              default=OutputFormat.table.name,
+              show_default=True,
               help='The formatting style for command output.')
 @click.pass_context
-def cli(ctx, endpoint, iap_client_id, namespace, other_client_id, other_client_secret, output):
+def cli(ctx, endpoint, iap_client_id, namespace, other_client_id,
+        other_client_secret, output):
     """kfp is the command line interface to KFP service.
 
     Feature stage:
@@ -43,7 +58,8 @@ def cli(ctx, endpoint, iap_client_id, namespace, other_client_id, other_client_s
     if ctx.invoked_subcommand == 'diagnose_me':
         # Do not create a client for diagnose_me
         return
-    ctx.obj['client'] = Client(endpoint, iap_client_id, namespace, other_client_id, other_client_secret)
+    ctx.obj['client'] = Client(endpoint, iap_client_id, namespace,
+                               other_client_id, other_client_secret)
     ctx.obj['namespace'] = namespace
     ctx.obj['output'] = output
 
@@ -52,8 +68,9 @@ def main():
     logging.basicConfig(format='%(message)s', level=logging.INFO)
     cli.add_command(run)
     cli.add_command(pipeline)
-    cli.add_command(diagnose_me,'diagnose_me')
+    cli.add_command(diagnose_me, 'diagnose_me')
     cli.add_command(experiment)
+    cli.add_command(typer.main.get_command(components.app))
     try:
         cli(obj={}, auto_envvar_prefix='KFP')
     except Exception as e: