|
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