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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev411
+Version: 0.4.1.dev95
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1.dev95}/futurehouse_client/clients/data_storage_methods.py

@@ -979,6 +979,7 @@ class DataStorageMethods:
         file_path: Path,
         description: str | None,
         path: str | None = None,
+        dataset_id: UUID | None = None,
     ) -> DataStorageResponse:
         """Asynchronously upload a single file."""
         file_size, text_payload = self._prepare_single_file_upload(
@@ -987,6 +988,7 @@ class DataStorageMethods:
 
         if text_payload:
             logger.debug("Sending file as text content")
+            text_payload.dataset_id = dataset_id
             return await self._acreate_data_storage_entry(text_payload)
 
         logger.debug(
@@ -997,6 +999,7 @@ class DataStorageMethods:
             description=description,
             path=path,
             is_collection=False,
+            dataset_id=dataset_id,
         )
 
         data_storage_response = await self._acreate_data_storage_entry(payload)
@@ -1482,6 +1485,7 @@ class DataStorageMethods:
         content: str,
         description: str | None = None,
         path: str | None = None,
+        dataset_id: UUID | None = None,
     ) -> DataStorageResponse:
         """Asynchronously store content as a string in the data storage system.
 
@@ -1490,6 +1494,7 @@ class DataStorageMethods:
             content: Content to store as a string
             description: Optional description of the data storage entry
             path: Optional path for the data storage entry
+            dataset_id: Optional dataset ID to add entry to, or None to create new dataset
 
         Returns:
             DataStorageResponse containing the created data storage entry and storage locations
@@ -1503,6 +1508,7 @@ class DataStorageMethods:
                 content=content,
                 description=description,
                 path=path,
+                dataset_id=dataset_id,
             )
             return await self._acreate_data_storage_entry(payload)
         except HTTPStatusError as e:
@@ -1599,6 +1605,7 @@ class DataStorageMethods:
         manifest_filename: str | None = None,
         ignore_patterns: list[str] | None = None,
         ignore_filename: str = ".gitignore",
+        dataset_id: UUID | None = None,
     ) -> DataStorageResponse:
         """Asynchronously store file or directory content in the data storage system.
 
@@ -1612,6 +1619,7 @@ class DataStorageMethods:
             manifest_filename: Optional manifest file for hierarchical uploads.
             ignore_patterns: List of patterns to ignore when zipping.
             ignore_filename: Name of ignore file to read (default: .gitignore).
+            dataset_id: Optional dataset ID to add entry to, or None to create new dataset.
 
         Returns:
             The `DataStorageResponse` for the created entry. For hierarchical uploads,
@@ -1637,6 +1645,7 @@ class DataStorageMethods:
                     manifest_filename=manifest_filename,
                     ignore_patterns=ignore_patterns,
                     ignore_filename=ignore_filename,
+                    dataset_id=dataset_id,
                 )
                 if not responses:
                     raise DataStorageCreationError(
@@ -1644,7 +1653,7 @@ class DataStorageMethods:
                     )
                 return responses[0]
             return await self._aupload_data_single_file(
-                name, file_path, description, path
+                name, file_path, description, path, dataset_id
             )
 
         except HTTPStatusError as e:

futurehouse_client-0.4.1.dev95/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.dev95'
+__version_tuple__ = version_tuple = (0, 4, 1, 'dev95')

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1.dev95/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.dev95
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License

{futurehouse_client-0.3.20.dev411 → futurehouse_client-0.4.1.dev95}/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)

futurehouse_client-0.3.20.dev411/futurehouse_client/version.py

@@ -1,34 +0,0 @@
-# file generated by setuptools-scm
-# don't change, don't track in version control
-
-__all__ = [
-    "__version__",
-    "__version_tuple__",
-    "version",
-    "version_tuple",
-    "__commit_id__",
-    "commit_id",
-]
-
-TYPE_CHECKING = False
-if TYPE_CHECKING:
-    from typing import Tuple
-    from typing import Union
-
-    VERSION_TUPLE = Tuple[Union[int, str], ...]
-    COMMIT_ID = Union[str, None]
-else:
-    VERSION_TUPLE = object
-    COMMIT_ID = object
-
-version: str
-__version__: str
-__version_tuple__: VERSION_TUPLE
-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')
-
-__commit_id__ = commit_id = 'gbf8ac9d70'