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 fe589528ee4 Show loop iterations and outcomes throughout the UI
(#74354)
add ed572aff106 Bound LLM operator usage_limits across task retries
(#74396)
No new revisions were added by this update.
Summary of changes:
providers/common/ai/docs/changelog.rst | 11 +
providers/common/ai/docs/operators/llm.rst | 53 +++--
.../providers/common/ai/mixins/usage_budget.py | 152 +++++++++++++
.../airflow/providers/common/ai/operators/agent.py | 65 ++----
.../airflow/providers/common/ai/operators/llm.py | 49 +++-
.../providers/common/ai/operators/llm_branch.py | 8 +-
.../common/ai/operators/llm_file_analysis.py | 5 +-
.../common/ai/operators/llm_schema_compare.py | 20 +-
.../providers/common/ai/operators/llm_sql.py | 8 +-
.../providers/common/ai/utils/usage_budget.py | 15 +-
.../common/ai/tests/unit/common/ai/conftest.py | 48 ++++
.../unit/common/ai/mixins/test_usage_budget.py | 190 ++++++++++++++++
.../tests/unit/common/ai/operators/test_agent.py | 244 ++++++++++----------
.../ai/tests/unit/common/ai/operators/test_llm.py | 246 ++++++++++++++++++++-
.../unit/common/ai/operators/test_llm_branch.py | 121 +++++++++-
.../common/ai/operators/test_llm_file_analysis.py | 88 +++++++-
.../common/ai/operators/test_llm_schema_compare.py | 129 ++++++++++-
.../tests/unit/common/ai/operators/test_llm_sql.py | 121 +++++++++-
18 files changed, 1335 insertions(+), 238 deletions(-)
create mode 100644
providers/common/ai/src/airflow/providers/common/ai/mixins/usage_budget.py
create mode 100644
providers/common/ai/tests/unit/common/ai/mixins/test_usage_budget.py
