diff --git a/projects/fal/src/fal/cli/apps.py b/projects/fal/src/fal/cli/apps.py index fa98f053..439fb43e 100644 --- a/projects/fal/src/fal/cli/apps.py +++ b/projects/fal/src/fal/cli/apps.py @@ -128,6 +128,7 @@ def _scale(args): and args.min_concurrency is None and args.request_timeout is None and args.startup_timeout is None + and args.machine_types is None ): args.console.log("No parameters for update were provided, ignoring.") return @@ -140,6 +141,7 @@ def _scale(args): min_concurrency=args.min_concurrency, request_timeout=args.request_timeout, startup_timeout=args.startup_timeout, + machine_types=args.machine_types, ) table = _apps_table([alias_info]) @@ -188,6 +190,13 @@ def _add_scale_parser(subparsers, parents): type=int, help="Startup timeout (seconds).", ) + parser.add_argument( + "--machine-type", + type=str, + action="append", + dest="machine_types", + help="Machine type.", + ) parser.set_defaults(func=_scale) diff --git a/projects/fal/src/fal/sdk.py b/projects/fal/src/fal/sdk.py index b214ce90..3eedef62 100644 --- a/projects/fal/src/fal/sdk.py +++ b/projects/fal/src/fal/sdk.py @@ -582,7 +582,9 @@ def update_application( min_concurrency: int | None = None, request_timeout: int | None = None, startup_timeout: int | None = None, + machine_types: list[str] | None = None, ) -> AliasInfo: + print(machine_types) request = isolate_proto.UpdateApplicationRequest( application_name=application_name, keep_alive=keep_alive, @@ -591,6 +593,7 @@ def update_application( min_concurrency=min_concurrency, request_timeout=request_timeout, startup_timeout=startup_timeout, + machine_types=machine_types, ) res: isolate_proto.UpdateApplicationResult = self.stub.UpdateApplication( request