add args

Author: PacificDou

roboflow/adapters/deploymentapi.py

@@ -7,14 +7,16 @@ class DeploymentApiError(Exception):
     pass
 
 
-def add_deployment(api_key, machine_type, deployment_name, inference_version):
+def add_deployment(api_key, machine_type, duration, delete_on_expiration, deployment_name, inference_version):
     url = f"{DEDICATED_DEPLOYMENT_URL}/add"
     response = requests.post(
         url,
         json={
             "api_key": api_key,
             # "security_level": security_level,
             "machine_type": machine_type,
+            "duration": duration,
+            "delete_on_expiration": delete_on_expiration,
             "deployment_name": deployment_name,
             "inference_version": inference_version,
         },

roboflow/deployment.py

@@ -29,6 +29,14 @@ def add_deployment_parser(subparsers):
     deployment_add_parser.add_argument(
         "-m", dest="machine_type", help="machine type, run `roboflow deployment machine_type` to see available options"
     )
+    deployment_add_parser.add_argument(
+        "-t", dest="duration", help="duration, how long you want to keep the deployment (unit: hour, default: 3)",
+        type=float, default=3
+    )
+    deployment_add_parser.add_argument(
+        "-e", dest="delete_on_expiration", help="delete when expired (default: True)",
+        type=bool, default=True
+    )
     deployment_add_parser.add_argument(
         "-n", dest="deployment_name", help="deployment name, must contain 3-10 lowercase characters"
     )
@@ -60,6 +68,8 @@ def add_deployment(args):
         api_key,
         # args.security_level,
         args.machine_type,
+        args.duration,
+        args.delete_on_expiration,
         args.deployment_name,
         args.inference_version,
     )