dbos 0.24.0a15__tar.gz → 0.25.0a1__tar.gz

{dbos-0.24.0a15 → dbos-0.25.0a1}/PKG-INFO

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

{dbos-0.24.0a15 → dbos-0.25.0a1}/dbos/_dbos.py

@@ -981,6 +981,15 @@ class DBOS:
         ), "workflow_id is only available within a DBOS operation."
         return ctx.workflow_id
 
+    @classproperty
+    def step_id(cls) -> int:
+        """Return the step ID for the current context. This is a unique identifier of the current step within the workflow."""
+        ctx = assert_current_dbos_context()
+        assert (
+            ctx.is_within_workflow()
+        ), "step_id is only available within a DBOS workflow."
+        return ctx.function_id
+
     @classproperty
     def parent_workflow_id(cls) -> str:
         """

{dbos-0.24.0a15 → dbos-0.25.0a1}/pyproject.toml

@@ -27,7 +27,7 @@ dependencies = [
 ]
 requires-python = ">=3.9"
 readme = "README.md"
-version = "0.24.0a15"
+version = "0.25.0a1"
 
 [project.license]
 text = "MIT"

{dbos-0.24.0a15 → dbos-0.25.0a1}/tests/test_dbos.py

@@ -38,6 +38,7 @@ def test_simple_workflow(dbos: DBOS) -> None:
 
     @DBOS.transaction(isolation_level="REPEATABLE READ")
     def test_transaction(var2: str) -> str:
+        assert DBOS.step_id == 1
         rows = DBOS.sql_session.execute(sa.text("SELECT 1")).fetchall()
         nonlocal txn_counter
         txn_counter += 1
@@ -46,6 +47,7 @@ def test_simple_workflow(dbos: DBOS) -> None:
 
     @DBOS.step()
     def test_step(var: str) -> str:
+        assert DBOS.step_id == 2
         nonlocal step_counter
         step_counter += 1
         DBOS.logger.info("I'm test_step")