airflow.providers.common.ai.durable.step_counter

Shared step counter for durable execution caching.

Classes

DurableStepCounter

Monotonically increasing counter shared between CachingModel and CachingToolset.

Module Contents

class airflow.providers.common.ai.durable.step_counter.DurableStepCounter[source]

Monotonically increasing counter shared between CachingModel and CachingToolset.

Each model call and tool call increments the counter. The step index is used as the cache key, ensuring deterministic replay on retry.

replayed_model: int = 0[source]
replayed_tool: int = 0[source]
cached_model: int = 0[source]
cached_tool: int = 0[source]
next_step()[source]

Return the current step and advance the counter.

property total_steps: int[source]

Total number of steps executed so far.

Was this entry helpful?