futurehouse-client 0.3.18.dev184__tar.gz → 0.3.18.dev186__tar.gz

{futurehouse_client-0.3.18.dev184 → futurehouse_client-0.3.18.dev186}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev184
+Version: 0.3.18.dev186
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent

{futurehouse_client-0.3.18.dev184 → futurehouse_client-0.3.18.dev186}/futurehouse_client/clients/rest_client.py

@@ -358,12 +358,12 @@ class RestClient:
         if not files:
             raise TaskFetchError(f"No files found in {path}")
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    @refresh_token_on_auth_error()
     def get_task(
         self, task_id: str | None = None, history: bool = False, verbose: bool = False
     ) -> "TaskResponse":
@@ -385,6 +385,7 @@ class RestClient:
                 ),
                 self.client.stream("GET", url, params={"history": history}) as response,
             ):
+                response.raise_for_status()
                 json_data = "".join(response.iter_text(chunk_size=1024))
                 data = json.loads(json_data)
                 if "id" not in data:
@@ -411,12 +412,12 @@ class RestClient:
         except Exception as e:
             raise TaskFetchError(f"Error getting task: {e!s}") from e
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    @refresh_token_on_auth_error()
     async def aget_task(
         self, task_id: str | None = None, history: bool = False, verbose: bool = False
     ) -> "TaskResponse":
@@ -504,12 +505,12 @@ class RestClient:
             raise TaskFetchError(f"Error creating task: {e!s}") from e
         return trajectory_id
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    @refresh_token_on_auth_error()
     async def acreate_task(self, task_data: TaskRequest | dict[str, Any]):
         """Create a new futurehouse task."""
         if isinstance(task_data, dict):
@@ -682,12 +683,12 @@ class RestClient:
             for task_id in trajectory_ids
         ]
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    @refresh_token_on_auth_error()
     def get_build_status(self, build_id: UUID | None = None) -> dict[str, Any]:
         """Get the status of a build."""
         try:
@@ -706,12 +707,12 @@ class RestClient:
         return response.json()
 
     # TODO: Refactor later so we don't have to ignore PLR0915
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
         retry=retry_if_connection_error,
     )
-    @refresh_token_on_auth_error()
     def create_job(self, config: JobDeploymentConfig) -> dict[str, Any]:  # noqa: PLR0915
         """Creates a futurehouse job deployment from the environment and environment files.
 
@@ -973,6 +974,7 @@ class RestClient:
         except Exception as e:
             raise FileUploadError(f"Error uploading directory {dir_path}: {e}") from e
 
+    @refresh_token_on_auth_error()
     def _upload_single_file(
         self,
         job_name: str,
@@ -1046,11 +1048,20 @@ class RestClient:
                         )
 
             logger.info(f"Successfully uploaded {file_name}")
-
+        except HTTPStatusError as e:
+            if e.response.status_code in AUTH_ERRORS_TO_RETRY_ON:
+                raise AuthError(
+                    e.response.status_code,
+                    f"Authentication failed: {e}",
+                    request=e.request,
+                    response=e.response,
+                ) from e
+            raise
         except Exception as e:
             logger.exception(f"Error uploading file {file_path}")
             raise FileUploadError(f"Error uploading file {file_path}: {e}") from e
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -1087,6 +1098,13 @@ class RestClient:
             response.raise_for_status()
             return response.json()
         except HTTPStatusError as e:
+            if e.response.status_code in AUTH_ERRORS_TO_RETRY_ON:
+                raise AuthError(
+                    e.response.status_code,
+                    f"Authentication failed: {e}",
+                    request=e.request,
+                    response=e.response,
+                ) from e
             logger.exception(
                 f"Error listing files for job {job_name}, trajectory {trajectory_id}, upload_id {upload_id}: {e.response.text}"
             )
@@ -1099,6 +1117,7 @@ class RestClient:
             )
             raise RestClientError(f"Error listing files: {e!s}") from e
 
+    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -1146,6 +1165,13 @@ class RestClient:
 
             logger.info(f"File {file_path} downloaded to {destination_path}")
         except HTTPStatusError as e:
+            if e.response.status_code in AUTH_ERRORS_TO_RETRY_ON:
+                raise AuthError(
+                    e.response.status_code,
+                    f"Authentication failed: {e}",
+                    request=e.request,
+                    response=e.response,
+                ) from e
             logger.exception(
                 f"Error downloading file {file_path} for job {job_name}, trajectory_id {trajectory_id}: {e.response.text}"
             )

{futurehouse_client-0.3.18.dev184 → futurehouse_client-0.3.18.dev186}/futurehouse_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev184
+Version: 0.3.18.dev186
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 Classifier: Operating System :: OS Independent