|
16 | 16 | import ads.opctl.model.cli |
17 | 17 | import ads.opctl.spark.cli |
18 | 18 | from ads.common import auth as authutil |
| 19 | +from ads.common.auth import AuthType |
19 | 20 | from ads.opctl.cmds import activate as activate_cmd |
20 | 21 | from ads.opctl.cmds import cancel as cancel_cmd |
21 | 22 | from ads.opctl.cmds import configure as configure_cmd |
|
29 | 30 | from ads.opctl.cmds import run_diagnostics as run_diagnostics_cmd |
30 | 31 | from ads.opctl.cmds import watch as watch_cmd |
31 | 32 | from ads.opctl.config.merger import ConfigMerger |
32 | | -from ads.opctl.constants import ( |
33 | | - BACKEND_NAME, |
34 | | - DEFAULT_MODEL_FOLDER, |
35 | | - RESOURCE_TYPE, |
36 | | - RUNTIME_TYPE, |
37 | | -) |
| 33 | +from ads.opctl.constants import (BACKEND_NAME, DEFAULT_MODEL_FOLDER, |
| 34 | + RESOURCE_TYPE, RUNTIME_TYPE) |
38 | 35 | from ads.opctl.utils import build_image as build_image_cmd |
39 | 36 | from ads.opctl.utils import publish_image as publish_image_cmd |
40 | 37 | from ads.opctl.utils import suppress_traceback |
@@ -618,6 +615,13 @@ def init(debug: bool, **kwargs: Dict[str, Any]) -> None: |
618 | 615 | required=False, |
619 | 616 | help="When the `inference_server='triton'`, the name of the model to invoke. This can only be used when model deployment id is passed in. For the other cases, it will be ignored.", |
620 | 617 | ) |
| 618 | +@click.option( |
| 619 | + "--auth", |
| 620 | + "-a", |
| 621 | + help="authentication method", |
| 622 | + type=click.Choice(AuthType.values()), |
| 623 | + default=None, |
| 624 | +) |
621 | 625 | @click.option("--debug", "-d", help="set debug mode", is_flag=True, default=False) |
622 | 626 | def predict(**kwargs): |
623 | 627 | """ |
|
0 commit comments