futurehouse-client 0.3.18.dev186__py3-none-any.whl → 0.3.19.dev111__py3-none-any.whl

futurehouse_client/__init__.py

@@ -1,12 +1,11 @@
 from .clients.job_client import JobClient, JobNames
-from .clients.rest_client import PQATaskResponse, TaskResponse, TaskResponseVerbose
 from .clients.rest_client import RestClient as FutureHouseClient
+from .clients.rest_client import TaskResponse, TaskResponseVerbose
 
 __all__ = [
     "FutureHouseClient",
     "JobClient",
     "JobNames",
-    "PQATaskResponse",
     "TaskResponse",
     "TaskResponseVerbose",
 ]

futurehouse_client/clients/__init__.py

@@ -1,12 +1,11 @@
 from .job_client import JobClient, JobNames
-from .rest_client import PQATaskResponse, TaskResponse, TaskResponseVerbose
 from .rest_client import RestClient as FutureHouseClient
+from .rest_client import TaskResponse, TaskResponseVerbose
 
 __all__ = [
     "FutureHouseClient",
     "JobClient",
     "JobNames",
-    "PQATaskResponse",
     "TaskResponse",
     "TaskResponseVerbose",
 ]

futurehouse_client/clients/job_client.py

@@ -8,7 +8,13 @@ from aviary.env import Frame
 from pydantic import BaseModel
 from tenacity import before_sleep_log, retry, stop_after_attempt, wait_exponential
 
-from futurehouse_client.models.app import Stage
+from futurehouse_client.models.app import (
+    FinchTaskResponse,
+    PhoenixTaskResponse,
+    PQATaskResponse,
+    Stage,
+    TaskResponse,
+)
 from futurehouse_client.models.rest import (
     FinalEnvironmentRequest,
     StoreAgentStatePostRequest,
@@ -31,6 +37,19 @@ class JobNames(StrEnum):
     DUMMY = "job-futurehouse-dummy-env"
     PHOENIX = "job-futurehouse-phoenix"
     FINCH = "job-futurehouse-data-analysis-crow-high"
+    CHIMP = "job-futurehouse-chimp"
+
+    @classmethod
+    def _get_response_mapping(cls) -> dict[str, type[TaskResponse]]:
+        return {
+            cls.CROW: PQATaskResponse,
+            cls.FALCON: PQATaskResponse,
+            cls.OWL: PQATaskResponse,
+            cls.CHIMP: PQATaskResponse,
+            cls.PHOENIX: PhoenixTaskResponse,
+            cls.FINCH: FinchTaskResponse,
+            cls.DUMMY: TaskResponse,
+        }
 
     @classmethod
     def from_stage(cls, job_name: str, stage: Stage | None = None) -> str:
@@ -52,6 +71,13 @@ class JobNames(StrEnum):
                 f"Invalid job name: {job_name}. \nOptions are: {', '.join([name.name for name in cls])}"
             ) from e
 
+    @staticmethod
+    def get_response_object_from_job(job_name: str) -> type[TaskResponse]:
+        return JobNames._get_response_mapping()[job_name]
+
+    def get_response_object(self) -> type[TaskResponse]:
+        return self._get_response_mapping()[self.name]
+
 
 class JobClient:
     REQUEST_TIMEOUT: ClassVar[float] = 30.0  # sec

futurehouse_client/clients/rest_client.py

@@ -13,9 +13,10 @@ import tempfile
 import time
 import uuid
 from collections.abc import Collection
+from concurrent.futures import ThreadPoolExecutor, as_completed
 from pathlib import Path
 from types import ModuleType
-from typing import Any, ClassVar, assert_never, cast
+from typing import Any, ClassVar, cast
 from uuid import UUID
 
 import cloudpickle
@@ -31,6 +32,7 @@ from httpx import (
     ReadError,
     ReadTimeout,
     RemoteProtocolError,
+    codes,
 )
 from ldp.agent import AgentConfig
 from requests.exceptions import RequestException, Timeout
@@ -45,21 +47,15 @@ from tqdm.asyncio import tqdm
 
 from futurehouse_client.clients import JobNames
 from futurehouse_client.models.app import (
-    APIKeyPayload,
     AuthType,
     JobDeploymentConfig,
-    PQATaskResponse,
     Stage,
     TaskRequest,
     TaskResponse,
     TaskResponseVerbose,
 )
 from futurehouse_client.models.rest import ExecutionStatus
-from futurehouse_client.utils.auth import (
-    AUTH_ERRORS_TO_RETRY_ON,
-    AuthError,
-    refresh_token_on_auth_error,
-)
+from futurehouse_client.utils.auth import RefreshingJWT
 from futurehouse_client.utils.general import gather_with_concurrency
 from futurehouse_client.utils.module_utils import (
     OrganizationSelector,
@@ -128,8 +124,6 @@ retry_if_connection_error = retry_if_exception_type((
     FileUploadError,
 ))
 
-# 5 minute default for JWTs
-JWT_TOKEN_CACHE_EXPIRY: int = 300  # seconds
 DEFAULT_AGENT_TIMEOUT: int = 2400  # seconds
 
 
@@ -140,6 +134,9 @@ class RestClient:
     MAX_RETRY_WAIT: ClassVar[int] = 10
     DEFAULT_POLLING_TIME: ClassVar[int] = 5  # seconds
     CHUNK_SIZE: ClassVar[int] = 16 * 1024 * 1024  # 16MB chunks
+    ASSEMBLY_POLLING_INTERVAL: ClassVar[int] = 10  # seconds
+    MAX_ASSEMBLY_WAIT_TIME: ClassVar[int] = 1800  # 30 minutes
+    MAX_CONCURRENT_CHUNKS: ClassVar[int] = 12  # Maximum concurrent chunk uploads
 
     def __init__(
         self,
@@ -163,69 +160,87 @@ class RestClient:
         self.api_key = api_key
         self._clients: dict[str, Client | AsyncClient] = {}
         self.headers = headers or {}
-        self.auth_jwt = self._run_auth(jwt=jwt)
+        self.jwt = jwt
         self.organizations: list[str] = self._filter_orgs(organization)
 
     @property
     def client(self) -> Client:
-        """Lazily initialized and cached HTTP client with authentication."""
-        return cast(Client, self.get_client("application/json", with_auth=True))
+        """Authenticated HTTP client for regular API calls."""
+        return cast(Client, self.get_client("application/json", authenticated=True))
 
     @property
     def async_client(self) -> AsyncClient:
-        """Lazily initialized and cached HTTP client with authentication."""
+        """Authenticated async HTTP client for regular API calls."""
         return cast(
             AsyncClient,
-            self.get_client("application/json", with_auth=True, with_async=True),
+            self.get_client("application/json", authenticated=True, async_client=True),
         )
 
     @property
-    def auth_client(self) -> Client:
-        """Lazily initialized and cached HTTP client without authentication."""
-        return cast(Client, self.get_client("application/json", with_auth=False))
+    def unauthenticated_client(self) -> Client:
+        """Unauthenticated HTTP client for auth operations."""
+        return cast(Client, self.get_client("application/json", authenticated=False))
 
     @property
     def multipart_client(self) -> Client:
-        """Lazily initialized and cached HTTP client for multipart uploads."""
-        return cast(Client, self.get_client(None, with_auth=True))
+        """Authenticated HTTP client for multipart uploads."""
+        return cast(Client, self.get_client(None, authenticated=True))
 
     def get_client(
         self,
         content_type: str | None = "application/json",
-        with_auth: bool = True,
-        with_async: bool = False,
+        authenticated: bool = True,
+        async_client: bool = False,
     ) -> Client | AsyncClient:
         """Return a cached HTTP client or create one if needed.
 
         Args:
             content_type: The desired content type header. Use None for multipart uploads.
-            with_auth: Whether the client should include an Authorization header.
-            with_async: Whether to use an async client.
+            authenticated: Whether the client should include authentication.
+            async_client: Whether to use an async client.
 
         Returns:
             An HTTP client configured with the appropriate headers.
         """
-        # Create a composite key based on content type and auth flag.
-        key = f"{content_type or 'multipart'}_{with_auth}_{with_async}"
+        # Create a composite key based on content type and auth flag
+        key = f"{content_type or 'multipart'}_{authenticated}_{async_client}"
+
         if key not in self._clients:
             headers = copy.deepcopy(self.headers)
-            if with_auth:
-                headers["Authorization"] = f"Bearer {self.auth_jwt}"
+            auth = None
+
+            if authenticated:
+                auth = RefreshingJWT(
+                    # authenticated=False will always return a synchronous client
+                    auth_client=cast(
+                        Client, self.get_client("application/json", authenticated=False)
+                    ),
+                    auth_type=self.auth_type,
+                    api_key=self.api_key,
+                    jwt=self.jwt,
+                )
+
             if content_type:
                 headers["Content-Type"] = content_type
+
+            headers["x-client"] = "sdk"
+
             self._clients[key] = (
                 AsyncClient(
                     base_url=self.base_url,
                     headers=headers,
                     timeout=self.REQUEST_TIMEOUT,
+                    auth=auth,
                 )
-                if with_async
+                if async_client
                 else Client(
                     base_url=self.base_url,
                     headers=headers,
                     timeout=self.REQUEST_TIMEOUT,
+                    auth=auth,
                 )
             )
+
         return self._clients[key]
 
     def close(self):
@@ -255,32 +270,6 @@ class RestClient:
             raise ValueError(f"Organization '{organization}' not found.")
         return filtered_orgs
 
-    def _run_auth(self, jwt: str | None = None) -> str:
-        auth_payload: APIKeyPayload | None
-        if self.auth_type == AuthType.API_KEY:
-            auth_payload = APIKeyPayload(api_key=self.api_key)
-        elif self.auth_type == AuthType.JWT:
-            auth_payload = None
-        else:
-            assert_never(self.auth_type)
-        try:
-            # Use the unauthenticated client for login
-            if auth_payload:
-                response = self.auth_client.post(
-                    "/auth/login", json=auth_payload.model_dump()
-                )
-                response.raise_for_status()
-                token_data = response.json()
-            elif jwt:
-                token_data = {"access_token": jwt, "expires_in": JWT_TOKEN_CACHE_EXPIRY}
-            else:
-                raise ValueError("JWT token required for JWT authentication.")
-
-            return token_data["access_token"]
-        except Exception as e:
-            raise RestClientError(f"Error authenticating: {e!s}") from e
-
-    @refresh_token_on_auth_error()
     def _check_job(self, name: str, organization: str) -> dict[str, Any]:
         try:
             response = self.client.get(
@@ -288,25 +277,113 @@ 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
-            raise
         except Exception as e:
             raise JobFetchError(f"Error checking job: {e!s}") from e
 
-    @refresh_token_on_auth_error()
     def _fetch_my_orgs(self) -> list[str]:
         response = self.client.get(f"/v0.1/organizations?filter={True}")
         response.raise_for_status()
         orgs = response.json()
         return [org["name"] for org in orgs]
 
+    def _check_assembly_status(
+        self, job_name: str, upload_id: str, file_name: str
+    ) -> dict[str, Any]:
+        """Check the assembly status of an uploaded file.
+
+        Args:
+            job_name: The name of the futurehouse job
+            upload_id: The upload ID
+            file_name: The name of the file
+
+        Returns:
+            Dict containing status information
+
+        Raises:
+            RestClientError: If there's an error checking status
+        """
+        try:
+            url = f"/v0.1/crows/{job_name}/assembly-status/{upload_id}/{file_name}"
+            response = self.client.get(url)
+            response.raise_for_status()
+            return response.json()
+        except Exception as e:
+            raise RestClientError(f"Error checking assembly status: {e}") from e
+
+    def _wait_for_all_assemblies_completion(
+        self,
+        job_name: str,
+        upload_id: str,
+        file_names: list[str],
+        timeout: int = MAX_ASSEMBLY_WAIT_TIME,
+    ) -> bool:
+        """Wait for all file assemblies to complete.
+
+        Args:
+            job_name: The name of the futurehouse job
+            upload_id: The upload ID
+            file_names: List of file names to wait for
+            timeout: Maximum time to wait in seconds
+
+        Returns:
+            True if all assemblies succeeded, False if any failed or timed out
+
+        Raises:
+            RestClientError: If any assembly fails
+        """
+        if not file_names:
+            return True
+
+        start_time = time.time()
+        logger.info(f"Waiting for assembly of {len(file_names)} file(s) to complete...")
+
+        completed_files: set[str] = set()
+
+        while (time.time() - start_time) < timeout and len(completed_files) < len(
+            file_names
+        ):
+            for file_name in file_names:
+                if file_name in completed_files:
+                    continue
+
+                try:
+                    status_data = self._check_assembly_status(
+                        job_name, upload_id, file_name
+                    )
+                    status = status_data.get("status")
+
+                    if status == ExecutionStatus.SUCCESS.value:
+                        logger.info(f"Assembly completed for {file_name}")
+                        completed_files.add(file_name)
+                    elif status == ExecutionStatus.FAIL.value:
+                        error_msg = status_data.get("error", "Unknown assembly error")
+                        raise RestClientError(
+                            f"Assembly failed for {file_name}: {error_msg}"
+                        )
+                    elif status == ExecutionStatus.IN_PROGRESS.value:
+                        logger.debug(f"Assembly in progress for {file_name}...")
+
+                except RestClientError:
+                    raise  # Re-raise assembly errors
+                except Exception as e:
+                    logger.warning(
+                        f"Error checking assembly status for {file_name}: {e}"
+                    )
+
+            # Don't sleep if all files are complete
+            if len(completed_files) < len(file_names):
+                time.sleep(self.ASSEMBLY_POLLING_INTERVAL)
+
+        if len(completed_files) < len(file_names):
+            remaining_files = set(file_names) - completed_files
+            logger.warning(
+                f"Assembly timeout for files: {remaining_files} after {timeout} seconds"
+            )
+            return False
+
+        logger.info(f"All {len(file_names)} file assemblies completed successfully")
+        return True
+
     @staticmethod
     def _validate_module_path(path: Path) -> None:
         """Validates that the given path exists and is a directory.
@@ -358,7 +435,6 @@ 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),
@@ -394,25 +470,12 @@ class RestClient:
 
             if verbose:
                 return verbose_response
-            if any(
-                JobNames.from_string(job_name) in verbose_response.job_name
-                for job_name in ["crow", "falcon", "owl", "dummy"]
-            ):
-                return PQATaskResponse(**data)
-            return TaskResponse(**data)
-        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
+            return JobNames.get_response_object_from_job(verbose_response.job_name)(
+                **data
+            )
         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),
@@ -450,25 +513,12 @@ class RestClient:
 
             if verbose:
                 return verbose_response
-            if any(
-                JobNames.from_string(job_name) in verbose_response.job_name
-                for job_name in ["crow", "falcon", "owl", "dummy"]
-            ):
-                return PQATaskResponse(**data)
-            return TaskResponse(**data)
-        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
+            return JobNames.get_response_object_from_job(verbose_response.job_name)(
+                **data
+            )
         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),
@@ -492,20 +542,10 @@ class RestClient:
             response.raise_for_status()
             trajectory_id = response.json()["trajectory_id"]
             self.trajectory_id = trajectory_id
-        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:
             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),
@@ -529,15 +569,6 @@ class RestClient:
             response.raise_for_status()
             trajectory_id = response.json()["trajectory_id"]
             self.trajectory_id = trajectory_id
-        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:
             raise TaskFetchError(f"Error creating task: {e!s}") from e
         return trajectory_id
@@ -683,7 +714,6 @@ 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),
@@ -695,19 +725,11 @@ class RestClient:
             build_id = build_id or self.build_id
             response = self.client.get(f"/v0.1/builds/{build_id}")
             response.raise_for_status()
-        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:
+            raise JobFetchError(f"Error getting build status: {e!s}") from e
         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),
@@ -887,13 +909,6 @@ class RestClient:
                 build_context = response.json()
                 self.build_id = build_context["build_id"]
             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
                 error_detail = response.json()
                 error_message = error_detail.get("detail", str(e))
                 raise JobCreationError(
@@ -903,6 +918,8 @@ class RestClient:
             raise JobCreationError(f"Error generating docker image: {e!s}") from e
         return build_context
 
+    # TODO: we should have have an async upload_file, check_assembly_status,
+    # wait_for_assembly_completion, upload_directory, upload_single_file
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -913,6 +930,8 @@ class RestClient:
         job_name: str,
         file_path: str | os.PathLike,
         upload_id: str | None = None,
+        wait_for_assembly: bool = True,
+        assembly_timeout: int = MAX_ASSEMBLY_WAIT_TIME,
     ) -> str:
         """Upload a file or directory to a futurehouse job bucket.
 
@@ -920,29 +939,47 @@ class RestClient:
             job_name: The name of the futurehouse job to upload to.
             file_path: The local path to the file or directory to upload.
             upload_id: Optional folder name to use for the upload. If not provided, a random UUID will be used.
+            wait_for_assembly: After file chunking, wait for the assembly to be processed.
+            assembly_timeout: Maximum time to wait for assembly in seconds.
 
         Returns:
             The upload ID used for the upload.
 
         Raises:
             FileUploadError: If there's an error uploading the file.
+            RestClientError: If assembly fails or times out.
         """
         file_path = Path(file_path)
         if not file_path.exists():
             raise FileNotFoundError(f"File or directory not found: {file_path}")
 
         upload_id = upload_id or str(uuid.uuid4())
+        uploaded_files: list[str] = []
 
         if file_path.is_dir():
             # Process directory recursively
-            self._upload_directory(job_name, file_path, upload_id)
+            uploaded_files = self._upload_directory(job_name, file_path, upload_id)
         else:
             # Process single file
             self._upload_single_file(job_name, file_path, upload_id)
+            uploaded_files = [file_path.name]
+
+        # Wait for all assemblies if requested and we have files
+        if wait_for_assembly and uploaded_files:
+            success = self._wait_for_all_assemblies_completion(
+                job_name, upload_id, uploaded_files, assembly_timeout
+            )
+            if not success:
+                raise RestClientError(
+                    f"Assembly failed or timed out for one or more files: {uploaded_files}"
+                )
+
         logger.info(f"Successfully uploaded {file_path} to {upload_id}")
         return upload_id
 
-    def _upload_directory(self, job_name: str, dir_path: Path, upload_id: str) -> None:
+    def _upload_directory(
+        self, job_name: str, dir_path: Path, upload_id: str
+    ) -> list[str]:
         """Upload all files in a directory recursively.
 
         Args:
@@ -950,12 +987,17 @@ class RestClient:
             dir_path: The path to the directory to upload.
             upload_id: The upload ID to use.
 
+        Returns:
+            List of uploaded file names.
+
         Raises:
             FileUploadError: If there's an error uploading any file.
         """
         # Skip common directories that shouldn't be uploaded
         if any(ignore in dir_path.parts for ignore in FILE_UPLOAD_IGNORE_PARTS):
-            return
+            return []
+
+        uploaded_files: list[str] = []
 
         try:
             # Upload all files in the directory recursively
@@ -965,24 +1007,27 @@ class RestClient:
                 ):
                     # Use path relative to the original directory as file name
                     rel_path = path.relative_to(dir_path)
+                    file_name = str(rel_path)
                     self._upload_single_file(
                         job_name,
                         path,
                         upload_id,
-                        file_name=str(rel_path),
+                        file_name=file_name,
                     )
+                    uploaded_files.append(file_name)
         except Exception as e:
             raise FileUploadError(f"Error uploading directory {dir_path}: {e}") from e
 
-    @refresh_token_on_auth_error()
+        return uploaded_files
+
     def _upload_single_file(
         self,
         job_name: str,
         file_path: Path,
         upload_id: str,
         file_name: str | None = None,
-    ) -> None:
-        """Upload a single file in chunks.
+    ) -> str | None:
+        """Upload a single file in chunks using parallel uploads.
 
         Args:
             job_name: The key of the crow to upload to.
@@ -990,6 +1035,9 @@ class RestClient:
             upload_id: The upload ID to use.
             file_name: Optional name to use for the file. If not provided, the file's name will be used.
 
+        Returns:
+            The status URL if this was the last chunk, None otherwise.
+
         Raises:
             FileUploadError: If there's an error uploading the file.
         """
@@ -999,16 +1047,190 @@ class RestClient:
         # Skip empty files
         if file_size == 0:
             logger.warning(f"Skipping upload of empty file: {file_path}")
-            return
+            return None
 
         total_chunks = (file_size + self.CHUNK_SIZE - 1) // self.CHUNK_SIZE
 
         logger.info(f"Uploading {file_path} as {file_name} ({total_chunks} chunks)")
 
+        status_url = None
+
         try:
-            with open(file_path, "rb") as f:
-                for chunk_index in range(total_chunks):
-                    # Read the chunk from the file
+            # Upload all chunks except the last one in parallel
+            if total_chunks > 1:
+                self._upload_chunks_parallel(
+                    job_name,
+                    file_path,
+                    file_name,
+                    upload_id,
+                    total_chunks - 1,
+                    total_chunks,
+                )
+
+            # Upload the last chunk separately (handles assembly)
+            status_url = self._upload_final_chunk(
+                job_name,
+                file_path,
+                file_name,
+                upload_id,
+                total_chunks - 1,
+                total_chunks,
+            )
+
+            logger.info(f"Successfully uploaded {file_name}")
+        except Exception as e:
+            logger.exception(f"Error uploading file {file_path}")
+            raise FileUploadError(f"Error uploading file {file_path}: {e}") from e
+        return status_url
+
+    def _upload_chunks_parallel(
+        self,
+        job_name: str,
+        file_path: Path,
+        file_name: str,
+        upload_id: str,
+        num_regular_chunks: int,
+        total_chunks: int,
+    ) -> None:
+        """Upload chunks in parallel batches.
+
+        Args:
+            job_name: The key of the crow to upload to.
+            file_path: The path to the file to upload.
+            file_name: The name to use for the file.
+            upload_id: The upload ID to use.
+            num_regular_chunks: Number of regular chunks (excluding final chunk).
+            total_chunks: Total number of chunks.
+
+        Raises:
+            FileUploadError: If there's an error uploading any chunk.
+        """
+        if num_regular_chunks <= 0:
+            return
+
+        # Process chunks in batches
+        for batch_start in range(0, num_regular_chunks, self.MAX_CONCURRENT_CHUNKS):
+            batch_end = min(
+                batch_start + self.MAX_CONCURRENT_CHUNKS, num_regular_chunks
+            )
+
+            # Upload chunks in this batch concurrently
+            with ThreadPoolExecutor(max_workers=self.MAX_CONCURRENT_CHUNKS) as executor:
+                futures = {
+                    executor.submit(
+                        self._upload_single_chunk,
+                        job_name,
+                        file_path,
+                        file_name,
+                        upload_id,
+                        chunk_index,
+                        total_chunks,
+                    ): chunk_index
+                    for chunk_index in range(batch_start, batch_end)
+                }
+
+                for future in as_completed(futures):
+                    chunk_index = futures[future]
+                    try:
+                        future.result()
+                        logger.debug(
+                            f"Uploaded chunk {chunk_index + 1}/{total_chunks} of {file_name}"
+                        )
+                    except Exception as e:
+                        logger.error(f"Error uploading chunk {chunk_index}: {e}")
+                        raise FileUploadError(
+                            f"Error uploading chunk {chunk_index} of {file_name}: {e}"
+                        ) from e
+
+    def _upload_single_chunk(
+        self,
+        job_name: str,
+        file_path: Path,
+        file_name: str,
+        upload_id: str,
+        chunk_index: int,
+        total_chunks: int,
+    ) -> None:
+        """Upload a single chunk.
+
+        Args:
+            job_name: The key of the crow to upload to.
+            file_path: The path to the file to upload.
+            file_name: The name to use for the file.
+            upload_id: The upload ID to use.
+            chunk_index: The index of this chunk.
+            total_chunks: Total number of chunks.
+
+        Raises:
+            Exception: If there's an error uploading the chunk.
+        """
+        with open(file_path, "rb") as f:
+            # Read the chunk from the file
+            f.seek(chunk_index * self.CHUNK_SIZE)
+            chunk_data = f.read(self.CHUNK_SIZE)
+
+            # Prepare and send the chunk
+            with tempfile.NamedTemporaryFile() as temp_file:
+                temp_file.write(chunk_data)
+                temp_file.flush()
+
+                # Create form data
+                with open(temp_file.name, "rb") as chunk_file_obj:
+                    files = {
+                        "chunk": (
+                            file_name,
+                            chunk_file_obj,
+                            "application/octet-stream",
+                        )
+                    }
+                    data = {
+                        "file_name": file_name,
+                        "chunk_index": chunk_index,
+                        "total_chunks": total_chunks,
+                        "upload_id": upload_id,
+                    }
+
+                    # Send the chunk
+                    response = self.multipart_client.post(
+                        f"/v0.1/crows/{job_name}/upload-chunk",
+                        files=files,
+                        data=data,
+                    )
+                    response.raise_for_status()
+
+    def _upload_final_chunk(
+        self,
+        job_name: str,
+        file_path: Path,
+        file_name: str,
+        upload_id: str,
+        chunk_index: int,
+        total_chunks: int,
+    ) -> str | None:
+        """Upload the final chunk with retry logic for missing chunks.
+
+        Args:
+            job_name: The key of the crow to upload to.
+            file_path: The path to the file to upload.
+            file_name: The name to use for the file.
+            upload_id: The upload ID to use.
+            chunk_index: The index of the final chunk.
+            total_chunks: Total number of chunks.
+
+        Returns:
+            The status URL from the response.
+
+        Raises:
+            FileUploadError: If there's an error uploading the final chunk.
+        """
+        retries = 0
+        max_retries = 3
+        retry_delay = 2.0  # seconds
+
+        while retries < max_retries:
+            try:
+                with open(file_path, "rb") as f:
+                    # Read the final chunk from the file
                     f.seek(chunk_index * self.CHUNK_SIZE)
                     chunk_data = f.read(self.CHUNK_SIZE)
 
@@ -1033,35 +1255,47 @@ class RestClient:
                                 "upload_id": upload_id,
                             }
 
-                            # Send the chunk
+                            # Send the final chunk
                             response = self.multipart_client.post(
                                 f"/v0.1/crows/{job_name}/upload-chunk",
                                 files=files,
                                 data=data,
                             )
+
+                            # Handle missing chunks (status 409)
+                            if response.status_code == codes.CONFLICT:
+                                retries += 1
+                                if retries < max_retries:
+                                    logger.warning(
+                                        f"Missing chunks detected for {file_name}, retrying in {retry_delay}s... (attempt {retries}/{max_retries})"
+                                    )
+                                    time.sleep(retry_delay)
+                                    continue
+
                             response.raise_for_status()
+                            response_data = response.json()
+                            status_url = response_data.get("status_url")
 
-                        # Call progress callback if provided
+                            logger.debug(
+                                f"Uploaded final chunk {chunk_index + 1}/{total_chunks} of {file_name}"
+                            )
+                            return status_url
 
-                        logger.debug(
-                            f"Uploaded chunk {chunk_index + 1}/{total_chunks} of {file_name}"
-                        )
+            except Exception as e:
+                if retries >= max_retries - 1:
+                    raise FileUploadError(
+                        f"Error uploading final chunk of {file_name}: {e}"
+                    ) from e
+                retries += 1
+                logger.warning(
+                    f"Error uploading final chunk of {file_name}, retrying in {retry_delay}s... (attempt {retries}/{max_retries}): {e}"
+                )
+                time.sleep(retry_delay)
 
-            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
+        raise FileUploadError(
+            f"Failed to upload final chunk of {file_name} after {max_retries} retries"
+        )
 
-    @refresh_token_on_auth_error()
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -1098,13 +1332,6 @@ 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}"
             )
@@ -1117,7 +1344,6 @@ 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),
@@ -1165,13 +1391,6 @@ 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/models/app.py

@@ -1,3 +1,4 @@
+import copy
 import json
 import os
 import re
@@ -675,7 +676,8 @@ class TaskResponse(BaseModel):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_fields(cls, data: Mapping[str, Any]) -> Mapping[str, Any]:
+    def validate_fields(cls, original_data: Mapping[str, Any]) -> Mapping[str, Any]:
+        data = copy.deepcopy(original_data)  # Avoid mutating the original data
         # Extract fields from environment frame state
         if not isinstance(data, dict):
             return data
@@ -690,7 +692,72 @@ class TaskResponse(BaseModel):
         return data
 
 
+class PhoenixTaskResponse(TaskResponse):
+    """
+    Response scheme for tasks executed with Phoenix.
+
+    Additional fields:
+        answer: Final answer from Phoenix
+    """
+
+    model_config = ConfigDict(extra="ignore")
+    answer: str | None = None
+
+    @model_validator(mode="before")
+    @classmethod
+    def validate_phoenix_fields(
+        cls, original_data: Mapping[str, Any]
+    ) -> Mapping[str, Any]:
+        data = copy.deepcopy(original_data)
+        if not isinstance(data, dict):
+            return data
+        if not (env_frame := data.get("environment_frame", {})):
+            return data
+        state = env_frame.get("state", {}).get("state", {})
+        data["answer"] = state.get("answer")
+        return data
+
+
+class FinchTaskResponse(TaskResponse):
+    """
+    Response scheme for tasks executed with Finch.
+
+    Additional fields:
+        answer: Final answer from Finch
+        notebook: a dictionary with `cells` and `metadata` regarding the notebook content
+    """
+
+    model_config = ConfigDict(extra="ignore")
+    answer: str | None = None
+    notebook: dict[str, Any] | None = None
+
+    @model_validator(mode="before")
+    @classmethod
+    def validate_finch_fields(
+        cls, original_data: Mapping[str, Any]
+    ) -> Mapping[str, Any]:
+        data = copy.deepcopy(original_data)
+        if not isinstance(data, dict):
+            return data
+        if not (env_frame := data.get("environment_frame", {})):
+            return data
+        state = env_frame.get("state", {}).get("state", {})
+        data["answer"] = state.get("answer")
+        data["notebook"] = state.get("nb_state")
+        return data
+
+
 class PQATaskResponse(TaskResponse):
+    """
+    Response scheme for tasks executed with PQA.
+
+    Additional fields:
+        answer: Final answer from PQA
+        formatted_answer: Formatted answer from PQA
+        answer_reasoning: Reasoning used to generate the final answer, if available
+        has_successful_answer: Whether the answer is successful
+    """
+
     model_config = ConfigDict(extra="ignore")
 
     answer: str | None = None
@@ -702,7 +769,8 @@ class PQATaskResponse(TaskResponse):
 
     @model_validator(mode="before")
     @classmethod
-    def validate_pqa_fields(cls, data: Mapping[str, Any]) -> Mapping[str, Any]:
+    def validate_pqa_fields(cls, original_data: Mapping[str, Any]) -> Mapping[str, Any]:
+        data = copy.deepcopy(original_data)  # Avoid mutating the original data
         if not isinstance(data, dict):
             return data
         if not (env_frame := data.get("environment_frame", {})):

futurehouse_client/utils/auth.py

@@ -1,107 +1,92 @@
-import asyncio
 import logging
-from collections.abc import Callable, Coroutine
-from functools import wraps
-from typing import Any, Final, Optional, ParamSpec, TypeVar, overload
+from collections.abc import Collection, Generator
+from typing import ClassVar, Final
 
 import httpx
-from httpx import HTTPStatusError
 
-logger = logging.getLogger(__name__)
-
-T = TypeVar("T")
-P = ParamSpec("P")
-
-AUTH_ERRORS_TO_RETRY_ON: Final[set[int]] = {
-    httpx.codes.UNAUTHORIZED,
-    httpx.codes.FORBIDDEN,
-}
-
-
-class AuthError(Exception):
-    """Raised when authentication fails with 401/403 status."""
-
-    def __init__(self, status_code: int, message: str, request=None, response=None):
-        self.status_code = status_code
-        self.request = request
-        self.response = response
-        super().__init__(message)
-
-
-def is_auth_error(e: Exception) -> bool:
-    if isinstance(e, AuthError):
-        return True
-    if isinstance(e, HTTPStatusError):
-        return e.response.status_code in AUTH_ERRORS_TO_RETRY_ON
-    return False
-
-
-def get_status_code(e: Exception) -> Optional[int]:
-    if isinstance(e, AuthError):
-        return e.status_code
-    if isinstance(e, HTTPStatusError):
-        return e.response.status_code
-    return None
+from futurehouse_client.models.app import APIKeyPayload, AuthType
 
+logger = logging.getLogger(__name__)
 
-@overload
-def refresh_token_on_auth_error(
-    func: Callable[P, Coroutine[Any, Any, T]],
-) -> Callable[P, Coroutine[Any, Any, T]]: ...
-
-
-@overload
-def refresh_token_on_auth_error(
-    func: None = None, *, max_retries: int = ...
-) -> Callable[[Callable[P, T]], Callable[P, T]]: ...
-
-
-def refresh_token_on_auth_error(func=None, max_retries=1):
-    """Decorator that refreshes JWT token on 401/403 auth errors."""
-
-    def decorator(fn):
-        @wraps(fn)
-        def sync_wrapper(self, *args, **kwargs):
-            retries = 0
-            while True:
-                try:
-                    return fn(self, *args, **kwargs)
-                except Exception as e:
-                    if is_auth_error(e) and retries < max_retries:
-                        retries += 1
-                        status = get_status_code(e) or "Unknown"
-                        logger.info(
-                            f"Received auth error {status}, "
-                            f"refreshing token and retrying (attempt {retries}/{max_retries})..."
-                        )
-                        self.auth_jwt = self._run_auth()
-                        self._clients = {}
-                        continue
-                    raise
-
-        @wraps(fn)
-        async def async_wrapper(self, *args, **kwargs):
-            retries = 0
-            while True:
-                try:
-                    return await fn(self, *args, **kwargs)
-                except Exception as e:
-                    if is_auth_error(e) and retries < max_retries:
-                        retries += 1
-                        status = get_status_code(e) or "Unknown"
-                        logger.info(
-                            f"Received auth error {status}, "
-                            f"refreshing token and retrying (attempt {retries}/{max_retries})..."
-                        )
-                        self.auth_jwt = self._run_auth()
-                        self._clients = {}
-                        continue
-                    raise
-
-        if asyncio.iscoroutinefunction(fn):
-            return async_wrapper
-        return sync_wrapper
-
-    if callable(func):
-        return decorator(func)
-    return decorator
+INVALID_REFRESH_TYPE_MSG: Final[str] = (
+    "API key auth is required to refresh auth tokens."
+)
+JWT_TOKEN_CACHE_EXPIRY: int = 300  # seconds
+
+
+def _run_auth(
+    client: httpx.Client,
+    auth_type: AuthType = AuthType.API_KEY,
+    api_key: str | None = None,
+    jwt: str | None = None,
+) -> str:
+    auth_payload: APIKeyPayload | None
+    if auth_type == AuthType.API_KEY:
+        auth_payload = APIKeyPayload(api_key=api_key)
+    elif auth_type == AuthType.JWT:
+        auth_payload = None
+    try:
+        if auth_payload:
+            response = client.post("/auth/login", json=auth_payload.model_dump())
+            response.raise_for_status()
+            token_data = response.json()
+        elif jwt:
+            token_data = {"access_token": jwt, "expires_in": JWT_TOKEN_CACHE_EXPIRY}
+        else:
+            raise ValueError("JWT token required for JWT authentication.")
+
+        return token_data["access_token"]
+    except Exception as e:
+        raise Exception("Failed to authenticate") from e  # noqa: TRY002
+
+
+class RefreshingJWT(httpx.Auth):
+    """Automatically (re-)inject a JWT and transparently retry exactly once when we hit a 401/403."""
+
+    RETRY_STATUSES: ClassVar[Collection[httpx.codes]] = {
+        httpx.codes.UNAUTHORIZED,
+        httpx.codes.FORBIDDEN,
+    }
+
+    def __init__(
+        self,
+        auth_client: httpx.Client,
+        auth_type: AuthType = AuthType.API_KEY,
+        api_key: str | None = None,
+        jwt: str | None = None,
+    ):
+        self.auth_type = auth_type
+        self.auth_client = auth_client
+        self.api_key = api_key
+        self._jwt = _run_auth(
+            client=auth_client,
+            jwt=jwt,
+            auth_type=auth_type,
+            api_key=api_key,
+        )
+
+    def refresh_token(self) -> None:
+        if self.auth_type == AuthType.JWT:
+            logger.error(INVALID_REFRESH_TYPE_MSG)
+            raise ValueError(INVALID_REFRESH_TYPE_MSG)
+        self._jwt = _run_auth(
+            client=self.auth_client,
+            auth_type=self.auth_type,
+            api_key=self.api_key,
+        )
+
+    def auth_flow(
+        self, request: httpx.Request
+    ) -> Generator[httpx.Request, httpx.Response, None]:
+        request.headers["Authorization"] = f"Bearer {self._jwt}"
+        response = yield request
+
+        # If it failed, refresh once and replay the request
+        if response.status_code in self.RETRY_STATUSES:
+            logger.info(
+                "Received %s, refreshing token and retrying …",
+                response.status_code,
+            )
+            self.refresh_token()
+            request.headers["Authorization"] = f"Bearer {self._jwt}"
+            yield request  # second (and final) attempt, again or use a while loop

{futurehouse_client-0.3.18.dev186.dist-info → futurehouse_client-0.3.19.dev111.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.18.dev186
+Version: 0.3.19.dev111
 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.19.dev111.dist-info/RECORD

@@ -0,0 +1,17 @@
+futurehouse_client/__init__.py,sha256=OzGDkVm5UTUzd4n8yOmRjMF73YrK0FaIQX5gS3Dk8Zo,304
+futurehouse_client/clients/__init__.py,sha256=-HXNj-XJ3LRO5XM6MZ709iPs29YpApss0Q2YYg1qMZw,280
+futurehouse_client/clients/job_client.py,sha256=JgB5IUAyCmnhGRsYc3bgKldA-lkM1JLwHRwwUeOCdus,11944
+futurehouse_client/clients/rest_client.py,sha256=_XgkzA9OhUKjL9vpkU6ixh2lUW9StgqfGgLk2qHjGgI,55518
+futurehouse_client/models/__init__.py,sha256=5x-f9AoM1hGzJBEHcHAXSt7tPeImST5oZLuMdwp0mXc,554
+futurehouse_client/models/app.py,sha256=VCtg0ygd-TSrR6DtfljTBt9jnl1eBNal8UXHFdkDg88,28587
+futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
+futurehouse_client/models/rest.py,sha256=lgwkMIXz0af-49BYSkKeS7SRqvN3motqnAikDN4YGTc,789
+futurehouse_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+futurehouse_client/utils/auth.py,sha256=tgWELjKfg8eWme_qdcRmc8TjQN9DVZuHHaVXZNHLchk,2960
+futurehouse_client/utils/general.py,sha256=A_rtTiYW30ELGEZlWCIArO7q1nEmqi8hUlmBRYkMQ_c,767
+futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
+futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
+futurehouse_client-0.3.19.dev111.dist-info/METADATA,sha256=N4Msi8W4mMBXFs_-Pl8Ii12RcLRm2eBl9NiIFCy5--E,12767
+futurehouse_client-0.3.19.dev111.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+futurehouse_client-0.3.19.dev111.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.3.19.dev111.dist-info/RECORD,,

{futurehouse_client-0.3.18.dev186.dist-info → futurehouse_client-0.3.19.dev111.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

futurehouse_client-0.3.18.dev186.dist-info/RECORD

@@ -1,17 +0,0 @@
-futurehouse_client/__init__.py,sha256=ddxO7JE97c6bt7LjNglZZ2Ql8bYCGI9laSFeh9MP6VU,344
-futurehouse_client/clients/__init__.py,sha256=tFWqwIAY5PvwfOVsCje4imjTpf6xXNRMh_UHIKVI1_0,320
-futurehouse_client/clients/job_client.py,sha256=uNkqQbeZw7wbA0qDWcIOwOykrosza-jev58paJZ_mbA,11150
-futurehouse_client/clients/rest_client.py,sha256=W9ASP1ZKYS7UL5J9b-Km77YXEiDQ9hCf4X_9PqaZZZc,47914
-futurehouse_client/models/__init__.py,sha256=5x-f9AoM1hGzJBEHcHAXSt7tPeImST5oZLuMdwp0mXc,554
-futurehouse_client/models/app.py,sha256=w_1e4F0IiC-BKeOLqYkABYo4U-Nka1S-F64S_eHB2KM,26421
-futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
-futurehouse_client/models/rest.py,sha256=lgwkMIXz0af-49BYSkKeS7SRqvN3motqnAikDN4YGTc,789
-futurehouse_client/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-futurehouse_client/utils/auth.py,sha256=Lq9mjSGc7iuRP6fmLICCS6KjzLHN6-tJUuhYp0XXrkE,3342
-futurehouse_client/utils/general.py,sha256=A_rtTiYW30ELGEZlWCIArO7q1nEmqi8hUlmBRYkMQ_c,767
-futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
-futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
-futurehouse_client-0.3.18.dev186.dist-info/METADATA,sha256=PvjehEQZu2ihl7kG1uDvWJVUxyYbV7J-VmAe42Ml3zo,12767
-futurehouse_client-0.3.18.dev186.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-futurehouse_client-0.3.18.dev186.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
-futurehouse_client-0.3.18.dev186.dist-info/RECORD,,