This is an automated email from the ASF dual-hosted git repository.
Lee-W pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
from a8a62573f54 TS SDK: embed one source region per native Dag file
(#73723)
add bfa2fe02fc1 Account for AgentOperator spend on failed runs and across
retries (#73706)
No new revisions were added by this update.
Summary of changes:
providers/common/ai/docs/changelog.rst | 14 +
providers/common/ai/docs/observability.rst | 23 +-
providers/common/ai/docs/operators/agent.rst | 58 +-
providers/common/ai/docs/operators/llm.rst | 30 +-
.../airflow/providers/common/ai/durable/base.py | 10 +
.../providers/common/ai/durable/caching_model.py | 108 +-
.../providers/common/ai/durable/caching_toolset.py | 14 +-
.../providers/common/ai/durable/replay_usage.py | 240 +++++
.../airflow/providers/common/ai/operators/agent.py | 359 +++++--
.../airflow/providers/common/ai/utils/logging.py | 42 +-
.../providers/common/ai/utils/usage_budget.py | 214 ++++
.../tests/unit/common/ai/decorators/test_agent.py | 6 +-
.../unit/common/ai/durable/test_caching_model.py | 211 +++-
.../unit/common/ai/durable/test_replay_cost.py | 171 ++-
.../unit/common/ai/durable/test_replay_usage.py | 114 ++
.../tests/unit/common/ai/operators/test_agent.py | 1128 +++++++++++++++++++-
.../ai/operators/test_agent_tool_approval.py | 45 +-
.../ai/tests/unit/common/ai/utils/test_logging.py | 77 ++
.../unit/common/ai/utils/test_usage_budget.py | 242 +++++
19 files changed, 2958 insertions(+), 148 deletions(-)
create mode 100644
providers/common/ai/src/airflow/providers/common/ai/durable/replay_usage.py
create mode 100644
providers/common/ai/src/airflow/providers/common/ai/utils/usage_budget.py
create mode 100644
providers/common/ai/tests/unit/common/ai/durable/test_replay_usage.py
create mode 100644
providers/common/ai/tests/unit/common/ai/utils/test_usage_budget.py
