diff --git a/centml/sdk/api.py b/centml/sdk/api.py index e1e11d3..ea07c20 100644 --- a/centml/sdk/api.py +++ b/centml/sdk/api.py @@ -8,6 +8,7 @@ CreateComputeDeploymentRequest, CreateCServeV3DeploymentRequest, ApiException, + InviteUserRequest, Metric, ) @@ -140,6 +141,10 @@ def get_credits(self): def initialize_user(self): return self._api.setup_stripe_customer_payments_setup_post() + def invite_user(self, email: str): + request = InviteUserRequest(email=email) + return self._api.invite_user_organizations_invite_post(request) + @contextmanager def get_centml_client():