futurehouse-client 0.3.20.dev411__tar.gz → 0.4.1__tar.gz

{futurehouse_client-0.3.20.dev411/futurehouse_client.egg-info → futurehouse_client-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev411
+Version: 0.4.1
 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.3.20.dev411 → futurehouse_client-0.4.1}/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,
@@ -157,15 +160,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 +239,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 +251,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 +282,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,
                 )
             )
@@ -1630,16 +1621,17 @@ class RestClient(DataStorageMethods):
             A list of world model names.
         """
         try:
+            # Use the consolidated endpoint with search parameters
             response = self.client.get(
-                "/v0.1/world-models/search/",
+                "/v0.1/world-models",
                 params={
-                    "query": query,
+                    "q": query,
                     "size": size,
-                    "total_search_size": total_search_size,
                     "search_all_versions": search_all_versions,
                 },
             )
             response.raise_for_status()
+            # The new endpoint returns a list of models directly
             return response.json()
         except HTTPStatusError as e:
             raise WorldModelFetchError(

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1}/futurehouse_client/models/app.py

@@ -27,10 +27,7 @@ if TYPE_CHECKING:
 MAX_CROW_JOB_RUN_TIMEOUT = 60 * 60 * 24  # 24 hours in sec
 MIN_CROW_JOB_RUN_TIMEOUT = 0  # sec
 
-
-class PythonVersion(StrEnum):
-    V3_11 = "3.11"
-    V3_12 = "3.12"
+DEFAULT_PYTHON_VERSION_USED_FOR_JOB_BUILDS = "3.13"
 
 
 class AuthType(StrEnum):
@@ -420,9 +417,9 @@ class JobDeploymentConfig(BaseModel):
         description="The configuration for the cloud run container.",
     )
 
-    python_version: PythonVersion = Field(
-        default=PythonVersion.V3_12,
-        description="The python version your docker image should build with.",
+    python_version: str = Field(
+        default=DEFAULT_PYTHON_VERSION_USED_FOR_JOB_BUILDS,
+        description="The python version your docker image should build with (e.g., '3.11', '3.12', '3.13').",
     )
 
     agent: Agent | AgentConfig | str = Field(

futurehouse_client-0.4.1/futurehouse_client/utils/general.py

@@ -0,0 +1,29 @@
+import asyncio
+from collections.abc import Awaitable, Iterable
+from typing import TypeVar
+
+from tqdm.asyncio import tqdm
+
+T = TypeVar("T")
+
+
+async def gather_with_concurrency(
+    n: int | asyncio.Semaphore, coros: Iterable[Awaitable[T]], progress: bool = False
+) -> list[T]:
+    """
+    Run asyncio.gather with a concurrency limit.
+
+    SEE: https://stackoverflow.com/a/61478547/2392535
+    """
+    semaphore = asyncio.Semaphore(n) if isinstance(n, int) else n
+
+    async def sem_coro(coro: Awaitable[T]) -> T:
+        async with semaphore:
+            return await coro
+
+    if progress:
+        return await tqdm.gather(
+            *(sem_coro(c) for c in coros), desc="Gathering", ncols=0
+        )
+
+    return await asyncio.gather(*(sem_coro(c) for c in coros))

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1}/futurehouse_client/utils/world_model_tools.py

@@ -46,16 +46,17 @@ class WorldModelTools:
         return WorldModelTools._get_client().create_world_model(world_model)
 
     @staticmethod
-    def search_world_models(query: str) -> list[str]:
+    def search_world_models(query: str, size: int = 10) -> list[str]:
         """Search for world models using a text query.
 
         Args:
             query: The search query string to match against world model content.
+            size: The number of results to return (default: 10).
 
         Returns:
             list[str]: A list of world model IDs that match the search query.
         """
-        return WorldModelTools._get_client().search_world_models(query, size=1)
+        return WorldModelTools._get_client().search_world_models(query, size=size)
 
 
 create_world_model_tool = Tool.from_function(WorldModelTools.create_world_model)

futurehouse_client-0.4.1/futurehouse_client/version.py

@@ -0,0 +1,21 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+else:
+    VERSION_TUPLE = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+
+__version__ = version = '0.4.1'
+__version_tuple__ = version_tuple = (0, 4, 1)

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1/futurehouse_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev411
+Version: 0.4.1
 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.3.20.dev411 → futurehouse_client-0.4.1}/futurehouse_client.egg-info/SOURCES.txt

@@ -12,13 +12,11 @@ futurehouse_client.egg-info/dependency_links.txt
 futurehouse_client.egg-info/requires.txt
 futurehouse_client.egg-info/top_level.txt
 futurehouse_client/clients/__init__.py
-futurehouse_client/clients/data_storage_methods.py
 futurehouse_client/clients/job_client.py
 futurehouse_client/clients/rest_client.py
 futurehouse_client/models/__init__.py
 futurehouse_client/models/app.py
 futurehouse_client/models/client.py
-futurehouse_client/models/data_storage_methods.py
 futurehouse_client/models/rest.py
 futurehouse_client/utils/__init__.py
 futurehouse_client/utils/auth.py

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1}/futurehouse_client.egg-info/requires.txt

@@ -1,20 +1,16 @@
-aiofiles
 cloudpickle
 fhaviary
-google-resumable-media[aiohttp]
 httpx
 ldp>=0.22.0
 litellm
 pydantic
 python-dotenv
-requests
 tenacity
 tqdm>=4.62
 
 [dev]
 black
 futurehouse-client[monitoring,typing]
-ipykernel
 jupyter
 jupyterlab
 mypy
@@ -36,6 +32,5 @@ setuptools_scm
 newrelic>=8.8.0
 
 [typing]
-types-PyYAML
 types-requests
 types-tqdm

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1}/pyproject.toml

@@ -17,16 +17,13 @@ classifiers = [
     "Programming Language :: Python",
 ]
 dependencies = [
-    "aiofiles",
     "cloudpickle",
     "fhaviary",
-    "google-resumable-media[aiohttp]",
     "httpx",
     "ldp>=0.22.0",
     "litellm",
     "pydantic",
     "python-dotenv",
-    "requests",
     "tenacity",
     "tqdm>=4.62",  # For tqdm.asyncio stdlib consistency
 ]
@@ -41,7 +38,6 @@ requires-python = ">=3.11,<3.14"
 dev = [
     "black",
     "futurehouse-client[monitoring,typing]",
-    "ipykernel",
     "jupyter",
     "jupyterlab",
     "mypy",
@@ -63,7 +59,6 @@ monitoring = [
     "newrelic>=8.8.0",
 ]
 typing = [
-    "types-PyYAML",
     "types-requests",
     "types-tqdm",
 ]

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1}/tests/test_rest.py

@@ -978,6 +978,7 @@ class TestUserAgentRequestOperations:
         payload = UserAgentRequestPostPayload(
             trajectory_id=running_trajectory_id,
             request={"question": "Do you approve?"},
+            notify_user={"email": False, "sms": False},  # avoid sending notifications
         )
         request_id = admin_client.create_user_agent_request(payload)
         assert isinstance(request_id, UUID)
@@ -1045,6 +1046,7 @@ class TestAsyncUserAgentRequestOperations:
                 query="Why would I follow up on this query?",
             ).model_dump(mode="json"),
             expires_in_seconds=10,
+            notify_user={"email": False, "sms": False},  # avoid sending notifications
         )
 
         request_id = await admin_client.acreate_user_agent_request(payload)
@@ -1097,6 +1099,7 @@ class TestAsyncUserAgentRequestOperations:
                 name=JobNames.from_string("dummy"),
                 query="Why would I follow up on this query?",
             ).model_dump(mode="json"),
+            notify_user={"email": False, "sms": False},  # avoid sending notifications
         )
 
         request_id = await admin_client.acreate_user_agent_request(payload)