initial version

Author: PacificDou

roboflow/adapters/rfapi.py

@@ -6,7 +6,7 @@
 import requests
 from requests_toolbelt.multipart.encoder import MultipartEncoder
 
-from roboflow.config import API_URL, DEFAULT_BATCH_NAME, DEFAULT_JOB_NAME
+from roboflow.config import API_URL, DEFAULT_BATCH_NAME, DEFAULT_JOB_NAME, ROBOFLOW_RUNNER_URL
 from roboflow.util import image_utils
 
 
@@ -205,3 +205,42 @@ def _save_annotation_error(image_id, response):
     else:
         errmsg += f"bad response: {response.status_code}: {responsejson}"
     return UploadError(errmsg)
+
+
+def add_runner(api_key, security_level, cloud_provider, machine_type, runner_name, inference_version):
+    url = f"{ROBOFLOW_RUNNER_URL}/create"
+    response = requests.post(url, json={'api_key': api_key, 'security_level': security_level, 
+                                        'cloud_provider': cloud_provider, 'machine_type': machine_type, 
+                                        'runner_name': runner_name, 'inference_version': inference_version})
+    if response.status_code != 200:
+        raise RoboflowError(response.text)
+    result = response.json()
+    return result
+    
+
+def get_runner(api_key, runner_id):
+    url = f"{ROBOFLOW_RUNNER_URL}/get"
+    response = requests.get(url, json={'api_key': api_key, 'runner_id': runner_id})
+    if response.status_code != 200:
+        raise RoboflowError(response.text)
+    result = response.json()
+    return result
+
+
+def list_runner(api_key):
+    url = f"{ROBOFLOW_RUNNER_URL}/list"
+    response = requests.get(url, json={'api_key': api_key})
+    if response.status_code != 200:
+        raise RoboflowError(response.text)
+    result = response.json()
+    return result
+
+
+def delete_runner(api_key, runner_id):
+    url = f"{ROBOFLOW_RUNNER_URL}/delete"
+    response = requests.post(url, json={'api_key': api_key, 'runner_id': runner_id})
+    if response.status_code != 200:
+        raise RoboflowError(response.text)
+    result = response.json()
+    return result
+

roboflow/config.py

@@ -63,6 +63,8 @@ def get_conditional_configuration_variable(key, default):
 CLIP_FEATURIZE_URL = get_conditional_configuration_variable("CLIP_FEATURIZE_URL", "CLIP FEATURIZE URL NOT IN ENV")
 OCR_URL = get_conditional_configuration_variable("OCR_URL", "OCR URL NOT IN ENV")
 
+ROBOFLOW_RUNNER_URL = get_conditional_configuration_variable("ROBOFLOW_RUNNER_URL", "https://runner.svc.roboflow.one")
+
 DEMO_KEYS = ["coco-128-sample", "chess-sample-only-api-key"]
 
 TYPE_CLASSICATION = "classification"

roboflow/roboflowpy.py

@@ -161,6 +161,23 @@ def infer(args):
     print(group)
 
 
+def add_runner(args):
+    runner_json = rfapi.add_runner(args.api_key, args.security_level, args.cloud_provider, args.machine_type, args.runner_name, args.inference_version)
+    print(json.dumps(runner_json, indent=2))
+
+def get_runner(args):
+    runner_json = rfapi.get_runner(args.api_key, args.runner_id)
+    print(json.dumps(runner_json, indent=2))
+
+def list_runner(args):
+    runner_json = rfapi.list_runner(args.vapi_key)
+    print(json.dumps(runner_json, indent=2))
+
+def delete_runner(args):
+    runner_json = rfapi.delete_runner(args.api_key, args.runner_id)
+    print(json.dumps(runner_json, indent=2))
+
+
 def _argparser():
     parser = argparse.ArgumentParser(description="Welcome to the roboflow CLI: computer vision at your fingertips 🪄")
     subparsers = parser.add_subparsers(title="subcommands")
@@ -173,6 +190,7 @@ def _argparser():
     _add_workspaces_parser(subparsers)
     _add_upload_model_parser(subparsers)
     _add_get_workspace_project_version_parser(subparsers)
+    _add_runner_parser(subparsers)
 
     return parser
 
@@ -458,6 +476,37 @@ def _add_login_parser(subparsers):
     login_parser.set_defaults(func=login)
 
 
+def _add_runner_parser(subparsers):
+    runner_parser = subparsers.add_parser(
+        "runner",
+        help="runner related commands.  type 'roboflow runner' to see detailed command help",
+    )
+    runner_subparsers = runner_parser.add_subparsers(title="runner subcommands")
+    runner_add_parser = runner_subparsers.add_parser("add", help="create a new runner")
+    runner_get_parser = runner_subparsers.add_parser("get", help="show detailed info for a runner")
+    runner_list_parser = runner_subparsers.add_parser("list", help="list runners in a workspace")
+    runner_delete_parser = runner_subparsers.add_parser("delete", help="delete a runner")
+
+    runner_add_parser.set_defaults(func=add_runner)
+    runner_add_parser.add_argument("api_key", help="api key")
+    runner_add_parser.add_argument("security_level", help="security level (protected)", default="protected")
+    runner_add_parser.add_argument("cloud_provider", help="cloud provider (gcp)", default="gcp")
+    runner_add_parser.add_argument("machine_type", help="machine type (n1-standard-4-cpu, g2-standard-8-cpu)", default="n1-standard-4-cpu")
+    runner_add_parser.add_argument("runner_name", help="runner name")
+    runner_add_parser.add_argument("inference_version", help="inference server version (docker image tag)", default="latest")
+    
+    runner_get_parser.set_defaults(func=get_runner)
+    runner_get_parser.add_argument("api_key", help="api key")
+    runner_get_parser.add_argument("runner_id", help="runner id")
+    
+    runner_list_parser.set_defaults(func=list_runner)
+    runner_list_parser.add_argument("api_key", help="api key")
+    
+    runner_delete_parser.set_defaults(func=delete_runner)
+    runner_delete_parser.add_argument("api_key", help="api key")
+    runner_delete_parser.add_argument("runner_id", help="runner id")
+
+
 def main():
     parser = _argparser()
     args = parser.parse_args()