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(): diff --git a/requirements.txt b/requirements.txt index b0e1633..6b5e5cd 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,4 @@ cryptography==44.0.1 prometheus-client>=0.20.0 scipy>=1.6.0 scikit-learn>=1.5.1 -platform-api-python-client==4.6.0 +platform-api-python-client==4.8.4