allow a server to charge for api calls

packages/syft/src/syft/orchestra.py

@@ -175,6 +175,11 @@ def deploy_to_python(
     background_tasks: bool = False,
     debug: bool = False,
     migrate: bool = False,
+    payment_required: bool = False,
+    server_payment_handle: str | None = None,
+    payment_api: str | None = None,
+    compute_price_module_path: str | None = None,
+    compute_price_func_name: str | None = None
 ) -> ServerHandle:
     worker_classes = {
         ServerType.DATASITE: Datasite,
@@ -204,6 +209,11 @@ def deploy_to_python(
         "background_tasks": background_tasks,
         "debug": debug,
         "migrate": migrate,
+        "payment_required": payment_required,
+        "server_payment_handle": server_payment_handle,
+        "payment_api": payment_api,
+        "compute_price_module_path": compute_price_module_path,
+        "compute_price_func_name": compute_price_func_name
     }
 
     if port:
@@ -300,6 +310,11 @@ def launch(
         background_tasks: bool = False,
         debug: bool = False,
         migrate: bool = False,
+        payment_required: bool = False,
+        server_payment_handle: str | None = None,
+        payment_api: str | None = None,
+        compute_price_module_path: str | None = None,
+        compute_price_func_name: str | None = None
     ) -> ServerHandle:
         if dev_mode is True:
             thread_workers = True
@@ -338,6 +353,11 @@ def launch(
                 background_tasks=background_tasks,
                 debug=debug,
                 migrate=migrate,
+                payment_required=payment_required,
+                server_payment_handle=server_payment_handle,
+                payment_api=payment_api,
+                compute_price_module_path=compute_price_module_path,
+                compute_price_func_name=compute_price_func_name
             )
             display(
                 SyftInfo(

packages/syft/src/syft/protocol/protocol_version.json

@@ -117,23 +117,23 @@
         }
       },
       "User": {
-        "3": {
-          "version": 3,
-          "hash": "cdd25c47e32ac79e177bd26b8d3231847f4722247543954fc967686159ac17a2",
+        "4": {
+          "version": 4,
+          "hash": "524e8bc1ee0e7c1cb0a40ca458a3ab9d67ec2de6f251284c3615b9ad2fce2911",
           "action": "add"
         }
       },
       "UserUpdate": {
-        "3": {
-          "version": 3,
-          "hash": "01c3a470b326229bc892bea96b425c059edfde3bca57b0af83491a4fecbd0402",
+        "4": {
+          "version": 4,
+          "hash": "db54c3edaf8dc683851391c75331becee033956a95cb7f5263d8168352b11338",
           "action": "add"
         }
       },
       "UserCreate": {
-        "3": {
-          "version": 3,
-          "hash": "cb2d9872cba80c960c748b232547556fa69901c108d84ab46f177cb47f0dd3fc",
+        "4": {
+          "version": 4,
+          "hash": "974d7c99097da1c1714b7a2a9ccee6e7961a715bec1231884a38a3d44a58eef3",
           "action": "add"
         }
       },
@@ -145,9 +145,9 @@
         }
       },
       "UserView": {
-        "3": {
-          "version": 3,
-          "hash": "a07bebf2362391b915fb2554e61c98e35cb4f945ba8c3c3299dd059dc6f0e137",
+        "4": {
+          "version": 4,
+          "hash": "47e4befb08f1dac419e91573549b9e29012ae992af618ffd321176319cf1fbe8",
           "action": "add"
         }
       },
@@ -777,6 +777,27 @@
           "action": "add"
         }
       },
+      "SyftObjectMigrationState": {
+        "2": {
+          "version": 2,
+          "hash": "51f416fbce4ff408903b96aafa924fc3adde33b19172122b06acd04a1651c7ab",
+          "action": "add"
+        }
+      },
+      "StoreMetadata": {
+        "1": {
+          "version": 1,
+          "hash": "8de9a22a2765ef976bc161cb0704347d30350c085da8c8ffa876065cfca3e5fd",
+          "action": "add"
+        }
+      },
+      "MigrationData": {
+        "1": {
+          "version": 1,
+          "hash": "ae07a6345762b8ebe9d2a100776e2405fd17516c9d224913a3358c96480ba889",
+          "action": "add"
+        }
+      },
       "OnDiskBlobDeposit": {
         "2": {
           "version": 2,
@@ -1093,10 +1114,17 @@
           "action": "add"
         }
       },
-      "SyftObjectMigrationState": {
+      "ZMQClientConfig": {
+        "4": {
+          "version": 4,
+          "hash": "f54c62694a0b058ebf41d48b783b70423894a0609f50bc3d347b7b135d142674",
+          "action": "add"
+        }
+      },
+      "SQLiteStoreConfig": {
         "2": {
           "version": 2,
-          "hash": "51f416fbce4ff408903b96aafa924fc3adde33b19172122b06acd04a1651c7ab",
+          "hash": "b6d391bd7884886daef88b08734cf41b4d278b474c22cfc3574b9e7ec140c9e3",
           "action": "add"
         }
       },
@@ -1156,40 +1184,12 @@
           "action": "add"
         }
       },
-      "ZMQClientConfig": {
-        "4": {
-          "version": 4,
-          "hash": "f54c62694a0b058ebf41d48b783b70423894a0609f50bc3d347b7b135d142674",
-          "action": "add"
-        }
-      },
-      "SQLiteStoreConfig": {
-        "2": {
-          "version": 2,
-          "hash": "b6d391bd7884886daef88b08734cf41b4d278b474c22cfc3574b9e7ec140c9e3",
-          "action": "add"
-        }
-      },
       "Plan": {
         "2": {
           "version": 2,
           "hash": "eec308b8fd9437fc74a65f0cf70dff586130b35866d1e1512de49d45e7896981",
           "action": "add"
         }
-      },
-      "StoreMetadata": {
-        "1": {
-          "version": 1,
-          "hash": "8de9a22a2765ef976bc161cb0704347d30350c085da8c8ffa876065cfca3e5fd",
-          "action": "add"
-        }
-      },
-      "MigrationData": {
-        "1": {
-          "version": 1,
-          "hash": "ae07a6345762b8ebe9d2a100776e2405fd17516c9d224913a3358c96480ba889",
-          "action": "add"
-        }
       }
     }
   }

packages/syft/src/syft/server/run.py

@@ -20,13 +20,6 @@ def run() -> ServerHandle | None:
     parser.add_argument(
         "--name", help="server name", type=str, default="syft-server", dest="name"
     )
-    parser.add_argument(
-        "--server-type",
-        help="server type",
-        type=str,
-        default="python",
-        dest="server_type",
-    )
     parser.add_argument(
         "--host",
         help="host for binding",
@@ -73,6 +66,41 @@ def run() -> ServerHandle | None:
         default="True",
         dest="tail",
     )
+    parser.add_argument(
+        "--payment-required",
+        help="payment required",
+        type=str,
+        default="False",
+        dest="payment_required",
+    )
+    parser.add_argument(
+        "--server-payment-handle",
+        help="server payment handle",
+        type=str,
+        default="",
+        dest="server_payment_handle",
+    )
+    parser.add_argument(
+        "--payment-api",
+        help="payment api",
+        type=str,
+        default="",
+        dest="payment_api",
+    )
+    parser.add_argument(
+        "--compute-price-module-path",
+        help="compute price module path",
+        type=str,
+        default="",
+        dest="compute_price_module_path",
+    )
+    parser.add_argument(
+        "--compute-price-func-name",
+        help="compute price function name",
+        type=str,
+        default="",
+        dest="compute_price_func_name",
+    )
 
     args = parser.parse_args()
     if args.command != "launch":
@@ -82,18 +110,26 @@ def run() -> ServerHandle | None:
     args.reset = str_to_bool(args.reset)
     args.local_db = str_to_bool(args.local_db)
     args.tail = str_to_bool(args.tail)
-    args.cmd = str_to_bool(args.cmd)
+    args.payment_required = str_to_bool(args.payment_required)
+    args.server_payment_handle = str(args.server_payment_handle)
+    args.payment_api = str(args.payment_api)
+    args.compute_price_module_path = str(args.compute_price_module_path)
+    args.compute_price_func_name = str(args.compute_price_func_name)
 
     server = Orchestra.launch(
         name=args.name,
-        server_type=args.server_type,
         host=args.host,
         port=args.port,
         dev_mode=args.dev_mode,
         reset=args.reset,
         local_db=args.local_db,
         processes=args.processes,
         tail=args.tail,
+        payment_required=args.payment_required,
+        server_payment_handle=args.server_payment_handle,
+        payment_api=args.payment_api,
+        compute_price_module_path=args.compute_price_module_path,
+        compute_price_func_name=args.compute_price_func_name,
     )
     if not args.tail:
         return server