dbos 0.27.0a4__tar.gz → 0.27.0a7__tar.gz

{dbos-0.27.0a4 → dbos-0.27.0a7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbos
-Version: 0.27.0a4
+Version: 0.27.0a7
 Summary: Ultra-lightweight durable execution in Python
 Author-Email: "DBOS, Inc." <contact@dbos.dev>
 License: MIT

{dbos-0.27.0a4 → dbos-0.27.0a7}/dbos/__init__.py

@@ -1,6 +1,11 @@
 from . import _error as error
 from ._client import DBOSClient, EnqueueOptions
-from ._context import DBOSContextEnsure, DBOSContextSetAuth, SetWorkflowID
+from ._context import (
+    DBOSContextEnsure,
+    DBOSContextSetAuth,
+    SetWorkflowID,
+    SetWorkflowTimeout,
+)
 from ._dbos import DBOS, DBOSConfiguredInstance, WorkflowHandle, WorkflowHandleAsync
 from ._dbos_config import ConfigFile, DBOSConfig, get_dbos_database_url, load_config
 from ._kafka_message import KafkaMessage
@@ -19,6 +24,7 @@ __all__ = [
     "GetWorkflowsInput",
     "KafkaMessage",
     "SetWorkflowID",
+    "SetWorkflowTimeout",
     "WorkflowHandle",
     "WorkflowHandleAsync",
     "WorkflowStatus",

{dbos-0.27.0a4 → dbos-0.27.0a7}/dbos/_admin_server.py

@@ -209,7 +209,7 @@ class AdminRequestHandler(BaseHTTPRequestHandler):
         self._end_headers()
 
     def _handle_steps(self, workflow_id: str) -> None:
-        steps = self.dbos._sys_db.get_workflow_steps(workflow_id)
+        steps = self.dbos.list_workflow_steps(workflow_id)
 
         updated_steps = [
             {

{dbos-0.27.0a4 → dbos-0.27.0a7}/dbos/cli/cli.py

@@ -372,7 +372,7 @@ def list(
         typer.Option(
             "--sort-desc",
             "-d",
-            help="Sort the results in descending order",
+            help="Sort the results in descending order (older first)",
         ),
     ] = False,
     offset: Annotated[
@@ -442,7 +442,7 @@ def steps(
     help="Cancel a workflow so it is no longer automatically retried or restarted"
 )
 def cancel(
-    uuid: Annotated[str, typer.Argument()],
+    workflow_id: Annotated[str, typer.Argument()],
     db_url: Annotated[
         typing.Optional[str],
         typer.Option(
@@ -452,12 +452,12 @@ def cancel(
         ),
     ] = None,
 ) -> None:
-    start_client(db_url=db_url).cancel_workflow(workflow_id=uuid)
+    start_client(db_url=db_url).cancel_workflow(workflow_id=workflow_id)
 
 
 @workflow.command(help="Resume a workflow that has been cancelled")
 def resume(
-    uuid: Annotated[str, typer.Argument()],
+    workflow_id: Annotated[str, typer.Argument()],
     db_url: Annotated[
         typing.Optional[str],
         typer.Option(
@@ -467,12 +467,12 @@ def resume(
         ),
     ] = None,
 ) -> None:
-    start_client(db_url=db_url).resume_workflow(workflow_id=uuid)
+    start_client(db_url=db_url).resume_workflow(workflow_id=workflow_id)
 
 
 @workflow.command(help="Restart a workflow from the beginning with a new id")
 def restart(
-    uuid: Annotated[str, typer.Argument()],
+    workflow_id: Annotated[str, typer.Argument()],
     db_url: Annotated[
         typing.Optional[str],
         typer.Option(
@@ -484,7 +484,7 @@ def restart(
 ) -> None:
     status = (
         start_client(db_url=db_url)
-        .fork_workflow(workflow_id=uuid, start_step=1)
+        .fork_workflow(workflow_id=workflow_id, start_step=1)
         .get_status()
     )
     print(jsonpickle.encode(status, unpicklable=False))
@@ -494,13 +494,13 @@ def restart(
     help="fork a workflow from the beginning with a new id and from a step"
 )
 def fork(
-    uuid: Annotated[str, typer.Argument()],
+    workflow_id: Annotated[str, typer.Argument()],
     step: Annotated[
         int,
         typer.Option(
             "--step",
             "-s",
-            help="Restart from this step (default: first step)",
+            help="Restart from this step",
         ),
     ] = 1,
     db_url: Annotated[
@@ -514,7 +514,7 @@ def fork(
 ) -> None:
     status = (
         start_client(db_url=db_url)
-        .fork_workflow(workflow_id=uuid, start_step=step)
+        .fork_workflow(workflow_id=workflow_id, start_step=step)
         .get_status()
     )
     print(jsonpickle.encode(status, unpicklable=False))
@@ -579,7 +579,7 @@ def list_queue(
         typer.Option(
             "--sort-desc",
             "-d",
-            help="Sort the results in descending order",
+            help="Sort the results in descending order (older first)",
         ),
     ] = False,
     offset: Annotated[

{dbos-0.27.0a4 → dbos-0.27.0a7}/pyproject.toml

@@ -28,7 +28,7 @@ dependencies = [
 ]
 requires-python = ">=3.9"
 readme = "README.md"
-version = "0.27.0a4"
+version = "0.27.0a7"
 
 [project.license]
 text = "MIT"

{dbos-0.27.0a4 → dbos-0.27.0a7}/tests/test_async.py

@@ -7,7 +7,7 @@ import pytest
 import sqlalchemy as sa
 
 # Public API
-from dbos import DBOS, Queue, SetWorkflowID, WorkflowHandleAsync
+from dbos import DBOS, Queue, SetWorkflowID, SetWorkflowTimeout, WorkflowHandleAsync
 from dbos._context import SetWorkflowTimeout, assert_current_dbos_context
 from dbos._dbos import WorkflowHandle
 from dbos._dbos_config import ConfigFile

{dbos-0.27.0a4 → dbos-0.27.0a7}/tests/test_dbos.py

@@ -12,14 +12,17 @@ import pytest
 import sqlalchemy as sa
 
 # Public API
-from dbos import DBOS, ConfigFile, SetWorkflowID, WorkflowHandle, WorkflowStatusString
-
-# Private API because this is a test
-from dbos._context import (
+from dbos import (
+    DBOS,
+    ConfigFile,
+    SetWorkflowID,
     SetWorkflowTimeout,
-    assert_current_dbos_context,
-    get_local_dbos_context,
+    WorkflowHandle,
+    WorkflowStatusString,
 )
+
+# Private API because this is a test
+from dbos._context import assert_current_dbos_context, get_local_dbos_context
 from dbos._error import (
     DBOSConflictingRegistrationError,
     DBOSMaxStepRetriesExceeded,

{dbos-0.27.0a4 → dbos-0.27.0a7}/tests/test_queue.py

@@ -16,9 +16,10 @@ from dbos import (
     DBOSConfiguredInstance,
     Queue,
     SetWorkflowID,
+    SetWorkflowTimeout,
     WorkflowHandle,
 )
-from dbos._context import SetWorkflowTimeout, assert_current_dbos_context
+from dbos._context import assert_current_dbos_context
 from dbos._schemas.system_database import SystemSchema
 from dbos._sys_db import WorkflowStatusString
 from dbos._utils import GlobalParams

{dbos-0.27.0a4 → dbos-0.27.0a7}/tests/test_spans.py

@@ -46,7 +46,7 @@ def test_spans(dbos: DBOS) -> None:
         assert span.context is not None
 
     assert spans[0].name == test_step.__name__
-    assert spans[1].name == 'a new span'
+    assert spans[1].name == "a new span"
     assert spans[2].name == test_workflow.__name__
     assert spans[3].name == test_step.__name__
 
@@ -91,7 +91,7 @@ async def test_spans_async(dbos: DBOS) -> None:
         assert span.context is not None
 
     assert spans[0].name == test_step.__name__
-    assert spans[1].name == 'a new span'
+    assert spans[1].name == "a new span"
     assert spans[2].name == test_workflow.__name__
     assert spans[3].name == test_step.__name__