Lee-W commented on code in PR #69551: URL: https://github.com/apache/airflow/pull/69551#discussion_r3550598459
########## providers/common/ai/docs/index.rst: ########## @@ -19,6 +19,46 @@ ``apache-airflow-providers-common-ai`` ################################################## +When to use this provider +-------------------------- + +``common.ai`` is the vendor-neutral way to put LLM and agent steps in a Dag. It is built on +`pydantic-ai <https://ai.pydantic.dev/>`__, so the model vendor (OpenAI, Anthropic, Google, +Bedrock, …) is picked by the connection ``llm_conn_id`` points at — switching providers later +is a connection change, not a Dag rewrite. The AI step is orchestrated by Airflow: the model +calls, the agent loop, and any tools all run in the Airflow worker, where they get retries, +logging, and observability like any other task. + +Use it when a Dag needs: + +* **Generation, classification, summarization, or structured extraction** — + :doc:`LLMOperator and @task.llm <operators/llm>`, with Pydantic-typed output pushed to XCom. +* **Branching on a model's decision** — :doc:`LLMBranchOperator <operators/llm_branch>`. +* **Agents with tools** — :doc:`AgentOperator <operators/agent>` runs a multi-turn agent loop Review Comment: <img width="1944" height="366" alt="image" src="https://github.com/user-attachments/assets/0987ccd2-b5db-49c4-9c2b-6f8111d45ad1"; /> added both in https://github.com/apache/airflow/pull/69649/changes -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
