dbos 0.26.0a25__tar.gz → 0.26.1__tar.gz

{dbos-0.26.0a25 → dbos-0.26.1}/PKG-INFO

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

{dbos-0.26.0a25 → dbos-0.26.1}/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
 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",
     "WorkflowStatus",
     "WorkflowStatusString",

{dbos-0.26.0a25 → dbos-0.26.1}/pyproject.toml

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

{dbos-0.26.0a25 → dbos-0.26.1}/tests/test_async.py

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

{dbos-0.26.0a25 → dbos-0.26.1}/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.26.0a25 → dbos-0.26.1}/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