futurehouse-client 0.5.3__py3-none-any.whl → 0.5.4__py3-none-any.whl

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/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 = None

{futurehouse_client-0.5.3.dist-info → futurehouse_client-0.5.4.dist-info}/METADATA

@@ -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.dist-info → futurehouse_client-0.5.4.dist-info}/RECORD

@@ -1,10 +1,10 @@
 futurehouse_client/__init__.py,sha256=q5cpcuPkhTaueXsySsgWpH0F-2EsRxcdJfP91ze6khU,991
 futurehouse_client/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-futurehouse_client/version.py,sha256=EWl7XaGZUG57Di8WiRltpKAkwy1CShJuJ-i6_rAPr-w,704
+futurehouse_client/version.py,sha256=OrfVZdCDQ-QC6dUnxdROooJjwvLfeDMedTBstpAdSBU,704
 futurehouse_client/clients/__init__.py,sha256=-HXNj-XJ3LRO5XM6MZ709iPs29YpApss0Q2YYg1qMZw,280
 futurehouse_client/clients/data_storage_methods.py,sha256=3YurBAbEduZ8BdoHGdR9nro8a20pQPHAcbI-FRpVApY,121341
 futurehouse_client/clients/job_client.py,sha256=b5gpzulZpxpv9R337r3UKItnMdtd6CGlI1sV3_VQJso,13985
-futurehouse_client/clients/rest_client.py,sha256=LI8-F4cUcv8gnWhI06NB0k6iMnfDZVKv9Uh0FfXiDwo,110915
+futurehouse_client/clients/rest_client.py,sha256=en9kYwqx19DJdYuwVu4ChTbU25PWh6WtyGlpt0CioC4,111224
 futurehouse_client/models/__init__.py,sha256=N1MwDUYonsMN9NdaShsYcJspyL7H756MYj7VWFeD3fk,978
 futurehouse_client/models/app.py,sha256=2bFtQY1wl9oZM5Gyx1NrybKC9-fKzk20YRu20h2LiHY,32581
 futurehouse_client/models/client.py,sha256=554vp7Cr-17BTeRZtN5DhCRQesRRtr31ZPkHXUrhyCE,3835
@@ -17,8 +17,8 @@ futurehouse_client/utils/general.py,sha256=PIkGLCSA3kUvc6mwR-prEB7YnMdKILOIm6cPo
 futurehouse_client/utils/module_utils.py,sha256=aFyd-X-pDARXz9GWpn8SSViUVYdSbuy9vSkrzcVIaGI,4955
 futurehouse_client/utils/monitoring.py,sha256=UjRlufe67kI3VxRHOd5fLtJmlCbVA2Wqwpd4uZhXkQM,8728
 futurehouse_client/utils/world_model_tools.py,sha256=v2krZGrco0ur2a_pcRMtnQL05SxlIoBXuJ5R1JkQNws,2921
-futurehouse_client-0.5.3.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
-futurehouse_client-0.5.3.dist-info/METADATA,sha256=fvr2V59GYUzFP4lAaWmIntFOwb-S1pjz_3pwzuYgAQc,27054
-futurehouse_client-0.5.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-futurehouse_client-0.5.3.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
-futurehouse_client-0.5.3.dist-info/RECORD,,
+futurehouse_client-0.5.4.dist-info/licenses/LICENSE,sha256=oQ9ZHjUi-_6GfP3gs14FlPb0OlGwE1QCCKFGnJ4LD2I,11341
+futurehouse_client-0.5.4.dist-info/METADATA,sha256=C309S78nXoXmoqfG08gHZUrilQH-oc31MVLCeHqLz_M,27054
+futurehouse_client-0.5.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+futurehouse_client-0.5.4.dist-info/top_level.txt,sha256=TRuLUCt_qBnggdFHCX4O_BoCu1j2X43lKfIZC-ElwWY,19
+futurehouse_client-0.5.4.dist-info/RECORD,,