This is an automated email from the ASF dual-hosted git repository.
guan404ming pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
from 402ab11a4b0 Avoid loading timetables for dags in Execution API
endpoints (#75765)
add 27c7624430a Support multimodal prompts with require_approval in LLM
operators (#74388)
No new revisions were added by this update.
Summary of changes:
providers/common/ai/docs/operators/llm.rst | 6 +-
providers/common/ai/docs/operators/llm_branch.rst | 4 -
.../ai/docs/operators/llm_schema_compare.rst | 4 -
.../airflow/providers/common/ai/decorators/llm.py | 11 +--
.../providers/common/ai/decorators/llm_branch.py | 11 +--
.../common/ai/decorators/llm_schema_compare.py | 11 +--
.../providers/common/ai/decorators/llm_sql.py | 11 +--
.../airflow/providers/common/ai/mixins/approval.py | 20 ++---
.../airflow/providers/common/ai/operators/llm.py | 6 +-
.../providers/common/ai/operators/llm_branch.py | 3 +-
.../common/ai/operators/llm_schema_compare.py | 3 +-
.../airflow/providers/common/ai/utils/prompt.py | 65 ++++++++++++++
.../ai/tests/unit/common/ai/decorators/test_llm.py | 93 +++++++++++++++----
.../unit/common/ai/decorators/test_llm_branch.py | 22 -----
.../ai/decorators/test_llm_schema_compare.py | 24 -----
.../unit/common/ai/decorators/test_llm_sql.py | 20 -----
.../tests/unit/common/ai/mixins/test_approval.py | 26 ++++--
.../ai/tests/unit/common/ai/operators/test_llm.py | 100 +++++++++++++--------
.../unit/common/ai/operators/test_llm_branch.py | 16 ++--
.../common/ai/operators/test_llm_schema_compare.py | 36 ++++++--
.../tests/unit/common/ai/operators/test_llm_sql.py | 26 ------
.../ai/tests/unit/common/ai/utils/test_prompt.py | 92 +++++++++++++++++++
22 files changed, 376 insertions(+), 234 deletions(-)
create mode 100644
providers/common/ai/src/airflow/providers/common/ai/utils/prompt.py
create mode 100644
providers/common/ai/tests/unit/common/ai/utils/test_prompt.py
