add run for demo

Author: alexluck-sift

python/lib/sift_client/_internal/low_level_wrappers/runs.py

@@ -42,12 +42,10 @@ def __init__(self, grpc_client: GrpcClient):
         super().__init__(grpc_client)
 
     async def get_run(
-        self, 
+        self,
         run_id: str,
         *,
-        use_cache: bool = True,
         force_refresh: bool = False,
-        ttl: int | None = None,
     ) -> Run:
         """Get a run by run_id.
 
@@ -63,12 +61,11 @@ async def get_run(
         """
         request = GetRunRequest(run_id=run_id)
         stub = self._grpc_client.get_stub(RunServiceStub)
-        response = await self._call_with_cache(
+        response = await self.call_with_cache(
             stub.GetRun,
             request,
-            use_cache=use_cache,
+            use_cache=self._grpc_client.has_cache,
             force_refresh=force_refresh,
-            ttl=ttl,
         )
         grpc_run = cast("GetRunResponse", response).run
         return Run._from_proto(grpc_run)
@@ -80,9 +77,7 @@ async def list_runs(
         page_token: str | None = None,
         query_filter: str | None = None,
         order_by: str | None = None,
-        use_cache: bool = True,
         force_refresh: bool = False,
-        ttl: int | None = None,
     ) -> tuple[list[Run], str]:
         """List runs with optional filtering and pagination.
 
@@ -107,12 +102,11 @@ async def list_runs(
 
         request = ListRunsRequest(**request_kwargs)
         stub = self._grpc_client.get_stub(RunServiceStub)
-        response = await self._call_with_cache(
+        response = await self.call_with_cache(
             stub.ListRuns,
             request,
-            use_cache=use_cache,
+            use_cache=self._grpc_client.has_cache,
             force_refresh=force_refresh,
-            ttl=ttl,
         )
         response = cast("ListRunsResponse", response)
 
@@ -125,9 +119,7 @@ async def list_all_runs(
         query_filter: str | None = None,
         order_by: str | None = None,
         max_results: int | None = None,
-        use_cache: bool = True,
         force_refresh: bool = False,
-        ttl: int | None = None,
     ) -> list[Run]:
         """List all runs with optional filtering.
 
@@ -144,64 +136,27 @@ async def list_all_runs(
                 **k,
                 query_filter=query_filter,
                 order_by=order_by,
-                use_cache=use_cache,
                 force_refresh=force_refresh,
-                ttl=ttl,
             ),
             kwargs={},
             order_by=order_by,
             max_results=max_results,
         )
 
-    async def create_run(
-        self, 
-        *, 
-        create: RunCreate,
-        use_cache: bool = False,  # Default to False for write operations
-        force_refresh: bool = False,
-        ttl: int | None = None,
-    ) -> Run:
+    async def create_run(self, *, create: RunCreate) -> Run:
         request_proto = create.to_proto()
-        stub = self._grpc_client.get_stub(RunServiceStub)
-        response = await self._call_with_cache(
-            stub.CreateRun,
-            request_proto,
-            use_cache=use_cache,
-            force_refresh=force_refresh,
-            ttl=ttl,
-        )
+        response = await self._grpc_client.get_stub(RunServiceStub).CreateRun(request_proto)
         grpc_run = cast("CreateRunResponse", response).run
         return Run._from_proto(grpc_run)
 
-    async def update_run(
-        self, 
-        update: RunUpdate,
-        *,
-        use_cache: bool = False,  # Default to False for write operations
-        force_refresh: bool = False,
-        ttl: int | None = None,
-    ) -> Run:
+    async def update_run(self, update: RunUpdate) -> Run:
         grpc_run, update_mask = update.to_proto_with_mask()
         request = UpdateRunRequest(run=grpc_run, update_mask=update_mask)
-        stub = self._grpc_client.get_stub(RunServiceStub)
-        response = await self._call_with_cache(
-            stub.UpdateRun,
-            request,
-            use_cache=use_cache,
-            force_refresh=force_refresh,
-            ttl=ttl,
-        )
+        response = await self._grpc_client.get_stub(RunServiceStub).UpdateRun(request)
         updated_grpc_run = cast("UpdateRunResponse", response).run
         return Run._from_proto(updated_grpc_run)
 
-    async def stop_run(
-        self, 
-        run_id: str,
-        *,
-        use_cache: bool = False,  # Default to False for write operations
-        force_refresh: bool = False,
-        ttl: int | None = None,
-    ) -> None:
+    async def stop_run(self, run_id: str) -> None:
         """Stop a run by setting its stop time to the current time.
 
         Args:
@@ -214,23 +169,10 @@ async def stop_run(
             raise ValueError("run_id must be provided")
 
         request = StopRunRequest(run_id=run_id)
-        stub = self._grpc_client.get_stub(RunServiceStub)
-        await self._call_with_cache(
-            stub.StopRun,
-            request,
-            use_cache=use_cache,
-            force_refresh=force_refresh,
-            ttl=ttl,
-        )
+        await self._grpc_client.get_stub(RunServiceStub).StopRun(request)
 
     async def create_automatic_run_association_for_assets(
-        self, 
-        run_id: str, 
-        asset_names: list[str],
-        *,
-        use_cache: bool = False,  # Default to False for write operations
-        force_refresh: bool = False,
-        ttl: int | None = None,
+            self, run_id: str, asset_names: list[str]
     ) -> None:
         """Associate assets with a run for automatic data ingestion.
 
@@ -247,14 +189,8 @@ async def create_automatic_run_association_for_assets(
             raise ValueError("asset_names must be provided")
 
         request = CreateAutomaticRunAssociationForAssetsRequest(
-            run_id=run_id, 
-            asset_names=asset_names
+            run_id=run_id, asset_names=asset_names
         )
-        stub = self._grpc_client.get_stub(RunServiceStub)
-        await self._call_with_cache(
-            stub.CreateAutomaticRunAssociationForAssets,
-            request,
-            use_cache=use_cache,
-            force_refresh=force_refresh,
-            ttl=ttl,
+        await self._grpc_client.get_stub(RunServiceStub).CreateAutomaticRunAssociationForAssets(
+            request
         )