From ad23bcb7b3f8f346ea7a9b96be46fecd0f75de24 Mon Sep 17 00:00:00 2001 From: Ivan Malison Date: Mon, 5 Aug 2024 14:58:11 -0600 Subject: [PATCH] Forward all arguments in workflow run list --- hatchet_sdk/clients/rest_client.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/hatchet_sdk/clients/rest_client.py b/hatchet_sdk/clients/rest_client.py index a18896ce..c09e7994 100644 --- a/hatchet_sdk/clients/rest_client.py +++ b/hatchet_sdk/clients/rest_client.py @@ -1,3 +1,4 @@ +import functools from typing import Any, Dict, List from pydantic import StrictInt, StrictStr @@ -8,6 +9,7 @@ ) from hatchet_sdk.clients.rest.models.event_order_by_field import EventOrderByField from hatchet_sdk.clients.rest.models.replay_event_request import ReplayEventRequest +from hatchet_sdk.clients.rest.models.workflow_run_list import WorkflowRunList from hatchet_sdk.clients.rest.models.workflow_run_status import WorkflowRunStatus from hatchet_sdk.clients.rest.models.workflow_runs_cancel_request import ( WorkflowRunsCancelRequest, @@ -56,22 +58,10 @@ def workflow_version_get(self, workflow_id: str, version: str | None = None): version=version, ) - def workflow_run_list( - self, - workflow_id: str | None = None, - offset: int | None = None, - limit: int | None = None, - event_id: str | None = None, - additional_metadata: List[StrictStr] | None = None, - ): - return self.workflow_api.workflow_run_list( - tenant=self.tenant_id, - offset=offset, - limit=limit, - workflow_id=workflow_id, - event_id=event_id, - additional_metadata=additional_metadata, - ) + @functools.wraps(WorkflowApi.workflow_run_list, assigned=["__doc__", "__annotations__"]) + def workflow_run_list(self, **kwargs) -> WorkflowRunList: + kwargs.setdefault("tenant", self.tenant_id) + return self.workflow_api.workflow_run_list(**kwargs) def workflow_run_get(self, workflow_run_id: str): return self.workflow_api.workflow_run_get(