labtasker 0.2.9__tar.gz → 0.2.10__tar.gz

{labtasker-0.2.9 → labtasker-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: labtasker
-Version: 0.2.9
+Version: 0.2.10
 Summary: A task queue system for lab experiments
 Author-email: Your Name <your.email@example.com>
 License: Apache License 2.0
@@ -32,7 +32,7 @@ Requires-Dist: pyyaml<7.0.0,>=6.0.0
 Requires-Dist: tomlkit<0.14.0,>=0.13.2
 Requires-Dist: importlib-metadata<9.0.0,>=8.5.0
 Requires-Dist: packaging<26.0,>=24.2
-Requires-Dist: sse-starlette<3.0.0,>=2.1.3
+Requires-Dist: sse-starlette<4.0.0,>=2.1.3
 Requires-Dist: httpx-sse<0.5.0,>=0.4.0
 Requires-Dist: stamina<26.0.0,>=25.1.0
 Requires-Dist: noneprompt<0.2.0,>=0.1.9

{labtasker-0.2.9 → labtasker-0.2.10}/labtasker/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.2.9"
+__version__ = "0.2.10"
 
 from labtasker.client.client_api import *
 from labtasker.client.core.config import get_client_config

{labtasker-0.2.9 → labtasker-0.2.10}/labtasker/client/client_api.py

@@ -42,27 +42,31 @@ __all__ = [
     "connect_events",
     "EventListener",
     # http api (you should be careful with these unless you know what you are doing)
+    "get_httpx_client",
     "close_httpx_client",
     "health_check",
-    "submit_task",
-    "delete_worker",
     "create_queue",
-    "create_worker",
+    "get_queue",
     "delete_queue",
-    "delete_task",
-    "delete_worker",
+    "submit_task",
     "fetch_task",
-    "get_queue",
-    "health_check",
-    "ls_tasks",
-    "ls_workers",
-    "refresh_task_heartbeat",
     "report_task_status",
+    "refresh_task_heartbeat",
+    "create_worker",
+    "ls_workers",
+    "report_worker_status",
+    "ls_tasks",
+    "update_tasks",
+    "delete_task",
+    "update_queue",
+    "delete_worker",
     # utilities
     "run_with_pty",
     "run_with_subprocess",
 ]
 
+assert len(set(__all__)) == len(__all__), "Duplicated symbols in __all__"
+
 
 def loop(
     required_fields: Optional[List[str]] = None,

{labtasker-0.2.9 → labtasker-0.2.10}/labtasker/client/core/job_runner.py

@@ -12,6 +12,7 @@ import labtasker
 from labtasker.api_models import TaskUpdateRequest
 from labtasker.client.core.api import (
     create_worker,
+    delete_worker,
     fetch_task,
     get_queue,
     report_task_status,
@@ -155,9 +156,12 @@ def loop_run(
         raise e
 
     # Create worker if not exists
+    auto_create_worker = False
+
     if current_worker_id() is None:
-        new_worker_id = worker_id or create_worker(**(create_worker_kwargs or {}))
-        set_current_worker_id(new_worker_id)
+        auto_create_worker = worker_id is None
+        worker_id = worker_id or create_worker(**(create_worker_kwargs or {}))
+        set_current_worker_id(worker_id)
 
     if cmd is None:
         cmd = sys.argv
@@ -270,12 +274,9 @@ def loop_run(
                                 resp = update_tasks(
                                     task_updates=[
                                         TaskUpdateRequest(
-                                            **{
-                                                "task_id": current_task_id(),
-                                                "status": "pending",  # running -> pending
-                                                "retries": 0,
-                                                "worker_id": None,
-                                            }
+                                            task_id=current_task_id(),  # noqa
+                                            status="pending",  # running -> pending
+                                            retries=0,
                                         )
                                     ]
                                 )
@@ -336,6 +337,10 @@ def loop_run(
                                 finish(status="success")
                             end_heartbeat()
                 except _LabtaskerLoopExit:
+                    # clean up the worker
+                    if auto_create_worker:  # worker is managed automatically
+                        delete_worker(worker_id=current_worker_id())
+
                     logger.info("Exiting task loop.")
                     break
                 except WorkerSuspended:

{labtasker-0.2.9 → labtasker-0.2.10}/labtasker/server/database.py

@@ -614,6 +614,7 @@ class DBService:
                         "last_modified": now,
                         "worker_id": worker_id,
                         "cmd": cmd,
+                        "summary": {},  # clear previous summary before fetched
                     }
                 }
 

{labtasker-0.2.9 → labtasker-0.2.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "labtasker"
-version = "0.2.9"
+version = "0.2.10"
 description = "A task queue system for lab experiments"
 authors = [{ name = "Your Name", email = "your.email@example.com" }]
 license = { text = "Apache License 2.0" }
@@ -36,7 +36,7 @@ dependencies = [
     "tomlkit (>=0.13.2,<0.14.0)",
     "importlib-metadata (>=8.5.0,<9.0.0)",
     "packaging (>=24.2,<26.0)",
-    "sse-starlette (>=2.1.3,<3.0.0)",
+    "sse-starlette (>=2.1.3,<4.0.0)",
     "httpx-sse (>=0.4.0,<0.5.0)",
     "stamina (>=25.1.0,<26.0.0)",
     "noneprompt (>=0.1.9,<0.2.0)",