dbos 0.27.0a6__tar.gz → 0.27.0a7__tar.gz

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

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

{dbos-0.27.0a6 → 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.0a6 → dbos-0.27.0a7}/pyproject.toml

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

{dbos-0.27.0a6 → 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.0a6 → 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.0a6 → 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.0a6 → 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__