Skip to content

Commit 96385e7

Browse files
authored
add optional permissions parameter to store secrets utils (#45)
* add optional permissions parameter to store secrets utils * reformat docstring
1 parent b11b128 commit 96385e7

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

examples/common/utils.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,10 @@
44
import time
55
from typing import Any, Callable, Dict, List
66

7-
import numpy as np
8-
import py_nillion_client as nillion
9-
107
import nada_numpy as na
118
import nada_numpy.client as na_client
9+
import numpy as np
10+
import py_nillion_client as nillion
1211

1312

1413
def async_timer(file_path: os.PathLike) -> Callable:
@@ -92,6 +91,7 @@ async def store_secret_array(
9291
secret_array: np.ndarray,
9392
name: str,
9493
nada_type: Any,
94+
permissions: nillion.Permissions = None,
9595
):
9696
"""
9797
Asynchronous function to store secret arrays on the nillion client.
@@ -105,6 +105,8 @@ async def store_secret_array(
105105
secret_array (np.ndarray): Secret array.
106106
name (str): Secrets name.
107107
nada_type (Any): Nada type.
108+
permissions (nillion.Permissions): Optional Permissions.
109+
108110
109111
Returns:
110112
str: Store ID.
@@ -118,6 +120,7 @@ async def store_secret_array(
118120
party_id,
119121
party_name,
120122
secrets,
123+
permissions,
121124
)
122125
return store_id
123126

@@ -131,6 +134,8 @@ async def store_secret_value(
131134
secret_value: Any,
132135
name: str,
133136
nada_type: Any,
137+
permissions: nillion.Permissions = None,
138+
134139
):
135140
"""
136141
Asynchronous function to store secret values on the nillion client.
@@ -144,6 +149,7 @@ async def store_secret_value(
144149
secret_value (Any): Secret single value.
145150
name (str): Secrets name.
146151
nada_type (Any): Nada type.
152+
permissions (nillion.Permissions): Optional Permissions.
147153
148154
Returns:
149155
str: Store ID.
@@ -163,6 +169,7 @@ async def store_secret_value(
163169
party_id,
164170
party_name,
165171
secrets,
172+
permissions,
166173
)
167174
return store_id
168175

@@ -174,6 +181,7 @@ async def store_secrets(
174181
party_id: str,
175182
party_name: str,
176183
secrets: nillion.Secrets,
184+
permissions: nillion.Permissions = None
177185
):
178186
"""
179187
Asynchronous function to store secret values on the nillion client.
@@ -185,13 +193,14 @@ async def store_secrets(
185193
party_id (str): Party ID.
186194
party_name (str): Party name.
187195
secrets (nillion.Secrets): Secrets.
196+
permissions (nillion.Permissions): Optional Permissions.
188197
189198
Returns:
190199
str: Store ID.
191200
"""
192201
secret_bindings = nillion.ProgramBindings(program_id)
193202
secret_bindings.add_input_party(party_name, party_id)
194-
store_id = await client.store_secrets(cluster_id, secret_bindings, secrets, None)
203+
store_id = await client.store_secrets(cluster_id, secret_bindings, secrets, permissions)
195204
return store_id
196205

197206

@@ -233,4 +242,4 @@ async def compute(
233242
if verbose:
234243
print(f"✅ Compute complete for compute_id {compute_event.uuid}")
235244
print(f"🖥️ The result is {compute_event.result.value}")
236-
return compute_event.result.value
245+
return compute_event.result.value

0 commit comments

Comments
 (0)