diff --git a/datamaker/client/mixins/integration.py b/datamaker/client/mixins/integration.py index 5e40c89..80571a0 100644 --- a/datamaker/client/mixins/integration.py +++ b/datamaker/client/mixins/integration.py @@ -8,5 +8,9 @@ def create_logs(self, data): return self._post(path, payload=data) def create_task(self, data): - path = 'agent_tasks/' + path = 'tasks/' return self._post(path, payload=data) + + def update_task(self, pk, data): + path = f'tasks/{pk}/' + return self._patch(path, payload=data)