Added reset method (#16)

kilroybot · Oct 17, 2022 · b23a43e · b23a43e
1 parent ca787a6
commit b23a43e
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/kilroy_module_client_py_sdk/src/kilroy_module_client_py_sdk/client.py b/kilroy_module_client_py_sdk/src/kilroy_module_client_py_sdk/client.py
@@ -49,6 +49,8 @@
     WatchMetricsResponse,
     WatchStatusRequest,
     WatchStatusResponse,
+    ResetRequest,
+    ResetResponse,
 )
 
 from kilroy_module_client_py_sdk.metrics import MetricConfig, MetricData
@@ -300,6 +302,23 @@ async def watch_metrics(
         ):
             yield response
 
+    async def reset(
+        self,
+        reset_request: "ResetRequest",
+        *,
+        timeout: Optional[float] = None,
+        deadline: Optional["Deadline"] = None,
+        metadata: Optional["MetadataLike"] = None
+    ) -> "ResetResponse":
+        return await self._unary_unary(
+            "/kilroy.module.v1alpha.ModuleService/Reset",
+            reset_request,
+            ResetResponse,
+            timeout=timeout,
+            deadline=deadline,
+            metadata=metadata,
+        )
+
 
 class ModuleService:
     def __init__(self, *args, **kwargs) -> None:
@@ -431,3 +450,6 @@ async def watch_metrics(
                 dataset_id=response.dataset_id,
                 data=json.loads(response.data),
             )
+
+    async def reset(self, *args, **kwargs) -> None:
+        await self._stub.reset(ResetRequest(), *args, **kwargs)