fix(api): correctly pass headers to client (#85)

* fix(api): correctly pass headers to client

* fix cassette

* fix _all_ cassette records

* ensure backwards compatibility through argument order

Author: Lukas Pfannschmidt

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/PyCQA/autoflake
-    rev: v1.4
+    rev: v2.2.1
     hooks:
       - id: autoflake
         name: autoflake (python)

superai/apis/data.py

@@ -10,7 +10,7 @@ class DataApiMixin(ABC):
     _resource = "data"
 
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     @property

superai/apis/data_program.py

@@ -9,7 +9,7 @@
 
 class DataProgramApiMixin(WorkflowApiMixin):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     @logdecorator.log_on_start(

superai/apis/ground_truth.py

@@ -4,7 +4,7 @@
 
 class GroundTruthApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     def create_ground_truth(

superai/apis/jobs.py

@@ -16,7 +16,7 @@
 
 class JobsApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     def create_jobs(

superai/apis/operations.py

@@ -7,7 +7,7 @@
 
 class OperationsApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     def get_operation_download_info(

superai/apis/project.py

@@ -9,7 +9,7 @@
 
 class ProjectApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     @logdecorator.log_on_start(
@@ -72,6 +72,7 @@ def get_project(self, uuid, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -136,6 +137,7 @@ def get_selected_workflow(self, uuid, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -232,6 +234,7 @@ def list_projects(self, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -306,6 +309,7 @@ def update_project(self, uuid, body, **kwargs):
             method="PATCH",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -378,5 +382,6 @@ def create_project(self, body, org=None, **kwargs):
             method="POST",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )

superai/apis/super_task.py

@@ -9,7 +9,7 @@
 
 class SuperTaskApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     @logdecorator.log_on_start(
@@ -83,6 +83,7 @@ def get_supertask(self, task_template_name, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -184,6 +185,7 @@ def list_supertasks(self, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -269,6 +271,7 @@ def put_supertask(self, task_template_name, body, **kwargs):
             method="PUT",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
             required_auth_token=True,
         )

superai/apis/tasks.py

@@ -15,7 +15,7 @@
 
 class TasksApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     def download_tasks(

superai/apis/workflow.py

@@ -8,7 +8,7 @@
 
 class WorkflowApiMixin(ABC):
     @abstractmethod
-    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False):
+    def request(self, uri, method, body_params=None, query_params=None, required_api_key=False, header_params=None):
         pass
 
     @logdecorator.log_on_start(
@@ -72,6 +72,7 @@ def get_workflow(self, workflow_name, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -173,6 +174,7 @@ def list_workflows(self, **kwargs):
             method="GET",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
         )
 
@@ -255,6 +257,7 @@ def put_workflow(self, workflow_name, body, **kwargs):
             method="PUT",
             query_params=query_params,
             body_params=body_params,
+            header_params=header_params,
             required_api_key=True,
             required_auth_token=True,
         )