futurehouse-client 0.3.20.dev411__py3-none-any.whl → 0.4.0__py3-none-any.whl

futurehouse_client/__init__.py

@@ -8,11 +8,6 @@ from .models.app import (
     TaskResponse,
     TaskResponseVerbose,
 )
-from .utils.world_model_tools import (
-    create_world_model_tool,
-    make_world_model_tools,
-    search_world_model_tool,
-)
 
 __all__ = [
     "FinchTaskResponse",
@@ -24,7 +19,4 @@ __all__ = [
     "TaskRequest",
     "TaskResponse",
     "TaskResponseVerbose",
-    "create_world_model_tool",
-    "make_world_model_tools",
-    "search_world_model_tool",
 ]

futurehouse_client/clients/rest_client.py

@@ -26,14 +26,21 @@ from httpx import (
     AsyncClient,
     Client,
     CloseError,
+    ConnectError,
+    ConnectTimeout,
     HTTPStatusError,
+    NetworkError,
+    ReadError,
+    ReadTimeout,
     RemoteProtocolError,
     codes,
 )
 from ldp.agent import AgentConfig
+from requests.exceptions import RequestException, Timeout
 from tenacity import (
     before_sleep_log,
     retry,
+    retry_if_exception_type,
     stop_after_attempt,
     wait_exponential,
 )
@@ -41,7 +48,6 @@ from tqdm import tqdm as sync_tqdm
 from tqdm.asyncio import tqdm
 
 from futurehouse_client.clients import JobNames
-from futurehouse_client.clients.data_storage_methods import DataStorageMethods
 from futurehouse_client.models.app import (
     AuthType,
     JobDeploymentConfig,
@@ -62,10 +68,7 @@ from futurehouse_client.models.rest import (
     WorldModelResponse,
 )
 from futurehouse_client.utils.auth import RefreshingJWT
-from futurehouse_client.utils.general import (
-    create_retry_if_connection_error,
-    gather_with_concurrency,
-)
+from futurehouse_client.utils.general import gather_with_concurrency
 from futurehouse_client.utils.module_utils import (
     OrganizationSelector,
     fetch_environment_function_docstring,
@@ -133,10 +136,6 @@ class WorldModelCreationError(RestClientError):
     """Raised when there's an error creating a world model."""
 
 
-class WorldModelDeletionError(RestClientError):
-    """Raised when there's an error deleting a world model."""
-
-
 class ProjectError(RestClientError):
     """Raised when there's an error with trajectory group operations."""
 
@@ -157,15 +156,28 @@ class FileUploadError(RestClientError):
     """Raised when there's an error uploading a file."""
 
 
-retry_if_connection_error = create_retry_if_connection_error(FileUploadError)
+retry_if_connection_error = retry_if_exception_type((
+    # From requests
+    Timeout,
+    ConnectionError,
+    RequestException,
+    # From httpx
+    ConnectError,
+    ConnectTimeout,
+    ReadTimeout,
+    ReadError,
+    NetworkError,
+    RemoteProtocolError,
+    CloseError,
+    FileUploadError,
+))
 
 DEFAULT_AGENT_TIMEOUT: int = 2400  # seconds
 
 
 # pylint: disable=too-many-public-methods
-class RestClient(DataStorageMethods):
-    REQUEST_TIMEOUT: ClassVar[float] = 30.0  # sec - for general API calls
-    FILE_UPLOAD_TIMEOUT: ClassVar[float] = 600.0  # 10 minutes - for file uploads
+class RestClient:
+    REQUEST_TIMEOUT: ClassVar[float] = 30.0  # sec
     MAX_RETRY_ATTEMPTS: ClassVar[int] = 3
     RETRY_MULTIPLIER: ClassVar[int] = 1
     MAX_RETRY_WAIT: ClassVar[int] = 10
@@ -223,35 +235,11 @@ class RestClient(DataStorageMethods):
         """Authenticated HTTP client for multipart uploads."""
         return cast(Client, self.get_client(None, authenticated=True))
 
-    @property
-    def file_upload_client(self) -> Client:
-        """Authenticated HTTP client with extended timeout for file uploads."""
-        return cast(
-            Client,
-            self.get_client(
-                "application/json", authenticated=True, timeout=self.FILE_UPLOAD_TIMEOUT
-            ),
-        )
-
-    @property
-    def async_file_upload_client(self) -> AsyncClient:
-        """Authenticated async HTTP client with extended timeout for file uploads."""
-        return cast(
-            AsyncClient,
-            self.get_client(
-                "application/json",
-                authenticated=True,
-                async_client=True,
-                timeout=self.FILE_UPLOAD_TIMEOUT,
-            ),
-        )
-
     def get_client(
         self,
         content_type: str | None = "application/json",
         authenticated: bool = True,
         async_client: bool = False,
-        timeout: float | None = None,
     ) -> Client | AsyncClient:
         """Return a cached HTTP client or create one if needed.
 
@@ -259,13 +247,12 @@ class RestClient(DataStorageMethods):
             content_type: The desired content type header. Use None for multipart uploads.
             authenticated: Whether the client should include authentication.
             async_client: Whether to use an async client.
-            timeout: Custom timeout in seconds. Uses REQUEST_TIMEOUT if not provided.
 
         Returns:
             An HTTP client configured with the appropriate headers.
         """
-        client_timeout = timeout or self.REQUEST_TIMEOUT
-        key = f"{content_type or 'multipart'}_{authenticated}_{async_client}_{client_timeout}"
+        # 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)
@@ -291,14 +278,14 @@ class RestClient(DataStorageMethods):
                 AsyncClient(
                     base_url=self.base_url,
                     headers=headers,
-                    timeout=client_timeout,
+                    timeout=self.REQUEST_TIMEOUT,
                     auth=auth,
                 )
                 if async_client
                 else Client(
                     base_url=self.base_url,
                     headers=headers,
-                    timeout=client_timeout,
+                    timeout=self.REQUEST_TIMEOUT,
                     auth=auth,
                 )
             )
@@ -1606,48 +1593,6 @@ class RestClient(DataStorageMethods):
         except Exception as e:
             raise WorldModelFetchError(f"An unexpected error occurred: {e!r}.") from e
 
-    @retry(
-        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
-        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
-        retry=retry_if_connection_error,
-    )
-    def search_world_models(
-        self,
-        query: str,
-        size: int = 10,
-        total_search_size: int = 50,
-        search_all_versions: bool = False,
-    ) -> list[str]:
-        """Search for world models.
-
-        Args:
-            query: The search query.
-            size: The number of results to return.
-            total_search_size: The number of results to search for.
-            search_all_versions: Whether to search all versions of the world model or just the latest one.
-
-        Returns:
-            A list of world model names.
-        """
-        try:
-            response = self.client.get(
-                "/v0.1/world-models/search/",
-                params={
-                    "query": query,
-                    "size": size,
-                    "total_search_size": total_search_size,
-                    "search_all_versions": search_all_versions,
-                },
-            )
-            response.raise_for_status()
-            return response.json()
-        except HTTPStatusError as e:
-            raise WorldModelFetchError(
-                f"Error searching world models: {e.response.status_code} - {e.response.text}"
-            ) from e
-        except Exception as e:
-            raise WorldModelFetchError(f"An unexpected error occurred: {e!r}.") from e
-
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
@@ -1723,32 +1668,6 @@ class RestClient(DataStorageMethods):
                 f"An unexpected error occurred during world model creation: {e!r}."
             ) from e
 
-    @retry(
-        stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
-        wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),
-        retry=retry_if_connection_error,
-    )
-    async def delete_world_model(self, world_model_id: UUID) -> None:
-        """Delete a world model snapshot by its ID.
-
-        Args:
-            world_model_id: The unique ID of the world model snapshot to delete.
-
-        Raises:
-            WorldModelDeletionError: If the API call fails.
-        """
-        try:
-            response = await self.async_client.delete(
-                f"/v0.1/world-models/{world_model_id}"
-            )
-            response.raise_for_status()
-        except HTTPStatusError as e:
-            raise WorldModelDeletionError(
-                f"Error deleting world model: {e.response.status_code} - {e.response.text}"
-            ) from e
-        except Exception as e:
-            raise WorldModelDeletionError(f"An unexpected error occurred: {e}") from e
-
     @retry(
         stop=stop_after_attempt(MAX_RETRY_ATTEMPTS),
         wait=wait_exponential(multiplier=RETRY_MULTIPLIER, max=MAX_RETRY_WAIT),

futurehouse_client/utils/general.py

@@ -2,45 +2,11 @@ import asyncio
 from collections.abc import Awaitable, Iterable
 from typing import TypeVar
 
-from httpx import (
-    CloseError,
-    ConnectError,
-    ConnectTimeout,
-    NetworkError,
-    ReadError,
-    ReadTimeout,
-    RemoteProtocolError,
-)
-from requests.exceptions import RequestException, Timeout
-from tenacity import retry_if_exception_type
 from tqdm.asyncio import tqdm
 
 T = TypeVar("T")
 
 
-_BASE_CONNECTION_ERRORS = (
-    # From requests
-    Timeout,
-    ConnectionError,
-    RequestException,
-    # From httpx
-    ConnectError,
-    ConnectTimeout,
-    ReadTimeout,
-    ReadError,
-    NetworkError,
-    RemoteProtocolError,
-    CloseError,
-)
-
-retry_if_connection_error = retry_if_exception_type(_BASE_CONNECTION_ERRORS)
-
-
-def create_retry_if_connection_error(*additional_exceptions):
-    """Create a retry condition with base connection errors plus additional exceptions."""
-    return retry_if_exception_type(_BASE_CONNECTION_ERRORS + additional_exceptions)
-
-
 async def gather_with_concurrency(
     n: int | asyncio.Semaphore, coros: Iterable[Awaitable[T]], progress: bool = False
 ) -> list[T]:

futurehouse_client/version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.20.dev411'
-__version_tuple__ = version_tuple = (0, 3, 20, 'dev411')
+__version__ = version = '0.4.0'
+__version_tuple__ = version_tuple = (0, 4, 0)
 
 __commit_id__ = commit_id = None

{futurehouse_client-0.3.20.dev411.dist-info → futurehouse_client-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev411
+Version: 0.4.0
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License
@@ -213,22 +213,18 @@ Classifier: Programming Language :: Python
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: aiofiles
 Requires-Dist: cloudpickle
 Requires-Dist: fhaviary
-Requires-Dist: google-resumable-media[aiohttp]
 Requires-Dist: httpx
 Requires-Dist: ldp>=0.22.0
 Requires-Dist: litellm
 Requires-Dist: pydantic
 Requires-Dist: python-dotenv
-Requires-Dist: requests
 Requires-Dist: tenacity
 Requires-Dist: tqdm>=4.62
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: futurehouse-client[monitoring,typing]; extra == "dev"
-Requires-Dist: ipykernel; extra == "dev"
 Requires-Dist: jupyter; extra == "dev"
 Requires-Dist: jupyterlab; extra == "dev"
 Requires-Dist: mypy; extra == "dev"
@@ -248,7 +244,6 @@ Requires-Dist: setuptools_scm; extra == "dev"
 Provides-Extra: monitoring
 Requires-Dist: newrelic>=8.8.0; extra == "monitoring"
 Provides-Extra: typing
-Requires-Dist: types-PyYAML; extra == "typing"
 Requires-Dist: types-requests; extra == "typing"
 Requires-Dist: types-tqdm; extra == "typing"
 Dynamic: license-file

futurehouse_client-0.4.0.dist-info/RECORD

@@ -0,0 +1,20 @@
+futurehouse_client/__init__.py,sha256=BztM_ntbgmIEjzvnBWcvPhvLjM8xGDFCK0Upf3-nIn8,488
+futurehouse_client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+futurehouse_client/version.py,sha256=2_0GUP7yBCXRus-qiJKxQD62z172WSs1sQ6DVpPsbmM,704
+futurehouse_client/clients/__init__.py,sha256=-HXNj-XJ3LRO5XM6MZ709iPs29YpApss0Q2YYg1qMZw,280
+futurehouse_client/clients/job_client.py,sha256=D51_qTxya6g5Wfg_ZfJdP031TV_YDJeXkGMiYAJ1qRc,11962
+futurehouse_client/clients/rest_client.py,sha256=iGBMTdtSVRA2eaqdLVsHrQoF8ToKOnFfbIVhfibjzFs,97022
+futurehouse_client/models/__init__.py,sha256=kQ4R7VEuRxO0IQEW_sk9CndBL7zzl8rUKI24ddyYLM0,647
+futurehouse_client/models/app.py,sha256=TGoAeENNPc5mSBkMHjh-Z8VIlnaUNcoWUJLxUhRIkEE,31868
+futurehouse_client/models/client.py,sha256=n4HD0KStKLm6Ek9nL9ylP-bkK10yzAaD1uIDF83Qp_A,1828
+futurehouse_client/models/rest.py,sha256=ybelLsyTsKYud7DYUCF0sFF6u81bl8WmS_wWAnbX-0M,3382
+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.4.0.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
+futurehouse_client-0.4.0.dist-info/METADATA,sha256=cDJvUHmNJHAFw6kELrjYIOHyIeSeYZuaDwHJskg8wjs,26797
+futurehouse_client-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+futurehouse_client-0.4.0.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.4.0.dist-info/RECORD,,