futurehouse-client 0.5.3__tar.gz → 0.5.4__tar.gz

{futurehouse_client-0.5.3/src/futurehouse_client.egg-info → futurehouse_client-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.5.3
+Version: 0.5.4
 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.5.3 → futurehouse_client-0.5.4}/src/futurehouse_client/clients/rest_client.py

@@ -793,7 +793,11 @@ class RestClient(DataStorageMethods):
                     task_id not in completed_tasks
                     and ExecutionStatus(task.status).is_terminal_state()
                 ):
-                    completed_tasks[task_id] = task
+                    # on completion fetches the full state and messages of the task
+                    completed_tasks[task_id] = await self.aget_task(
+                        task_id=task_id,
+                        lite=False,
+                    )
                     if progress_bar:
                         progress.update(1)
 
@@ -870,7 +874,10 @@ class RestClient(DataStorageMethods):
                 if not ExecutionStatus(task.status).is_terminal_state():
                     all_done = False
                 elif task_id not in completed_tasks:
-                    completed_tasks[task_id] = task
+                    completed_tasks[task_id] = self.get_task(
+                        task_id=task_id,
+                        lite=False,
+                    )
                     if progress_bar:
                         progress.update(1)
 

{futurehouse_client-0.5.3 → futurehouse_client-0.5.4}/src/futurehouse_client/version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.5.3'
-__version_tuple__ = version_tuple = (0, 5, 3)
+__version__ = version = '0.5.4'
+__version_tuple__ = version_tuple = (0, 5, 4)
 
-__commit_id__ = commit_id = 'gaaaa80cba'
+__commit_id__ = commit_id = 'gcb60aa15d'

{futurehouse_client-0.5.3 → futurehouse_client-0.5.4/src/futurehouse_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.5.3
+Version: 0.5.4
 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.5.3 → futurehouse_client-0.5.4}/tests/test_rest.py

@@ -44,6 +44,7 @@ from futurehouse_client.models.app import (
     PQATaskResponse,
     Stage,
     TaskRequest,
+    TaskResponse,
     TaskResponseVerbose,
 )
 from futurehouse_client.models.job_event import (
@@ -244,6 +245,25 @@ def test_run_until_done_futurehouse_dummy_env_crow(
     assert all(task.status == "success" for task in results)
 
 
+@pytest.mark.timeout(300)
+@pytest.mark.flaky(reruns=3)
+def test_run_until_done_returns_task_response(
+    admin_client: RestClient, task_req: TaskRequest
+):
+    """Test that run_tasks_until_done returns TaskResponse instead of LiteTaskResponse."""
+    tasks_to_do = [task_req, task_req]
+
+    results = admin_client.run_tasks_until_done(tasks_to_do)
+
+    assert len(results) == len(tasks_to_do), "Should return 2 tasks."
+    assert all(task.status == "success" for task in results)
+
+    for task in results:
+        assert isinstance(task, TaskResponse), (
+            f"Expected TaskResponse, got {type(task)}"
+        )
+
+
 @pytest.mark.timeout(300)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio
@@ -258,6 +278,26 @@ async def test_arun_until_done_futurehouse_dummy_env_crow(
     assert all(task.status == "success" for task in results)
 
 
+@pytest.mark.timeout(300)
+@pytest.mark.flaky(reruns=3)
+@pytest.mark.asyncio
+async def test_arun_until_done_returns_task_response(
+    admin_client: RestClient, task_req: TaskRequest
+):
+    """Test that arun_tasks_until_done returns TaskResponse instead of LiteTaskResponse."""
+    tasks_to_do = [task_req, task_req]
+
+    results = await admin_client.arun_tasks_until_done(tasks_to_do)
+
+    assert len(results) == len(tasks_to_do), "Should return 2 tasks."
+    assert all(task.status == "success" for task in results)
+
+    for task in results:
+        assert isinstance(task, TaskResponse), (
+            f"Expected TaskResponse, got {type(task)}"
+        )
+
+
 @pytest.mark.timeout(300)
 @pytest.mark.flaky(reruns=3)
 @pytest.mark.asyncio