Skip to content

Commit ae744d9

Browse files
authored
Add get_user_vault (#94)
Add create_inference example with private endpoint
1 parent e0cb86e commit ae744d9

File tree

2 files changed

+41
-0
lines changed

2 files changed

+41
-0
lines changed

centml/sdk/api.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,11 @@ def get_cluster_id(self, hardware_instance_id):
7878

7979
return filtered_hw[0].cluster_id
8080

81+
def get_user_vault(self, type):
82+
items = self._api.get_all_user_vault_items_endpoint_user_vault_get(type).results
83+
84+
return {i.key: i.value for i in items}
85+
8186

8287
@contextmanager
8388
def get_centml_client():

examples/sdk/create_inference.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
import centml
2+
from centml.sdk.api import get_centml_client
3+
from centml.sdk import DeploymentType, CreateInferenceDeploymentRequest, UserVaultType
4+
5+
6+
def main():
7+
with get_centml_client() as cclient:
8+
certs = cclient.get_user_vault(UserVaultType.CERTIFICATES)
9+
10+
request = CreateInferenceDeploymentRequest(
11+
name="nginx",
12+
cluster_id=1000,
13+
hardware_instance_id=1000,
14+
image_url="nginxinc/nginx-unprivileged",
15+
port=8080,
16+
min_scale=1,
17+
max_scale=1,
18+
endpoint_certificate_authority=certs["my_cert"],
19+
)
20+
response = cclient.create_inference(request)
21+
print("Create deployment response: ", response)
22+
23+
### Get deployment details
24+
deployment = cclient.get_inference(response.id)
25+
print("Deployment details: ", deployment)
26+
27+
'''
28+
### Pause the deployment
29+
cclient.pause(deployment.id)
30+
31+
### Delete the deployment
32+
cclient.delete(deployment.id)
33+
'''
34+
35+
if __name__ == "__main__":
36+
main()

0 commit comments

Comments
 (0)