# migrating-ai-sdk-to-common-ai

> Claude/Codex skill by [@astronomer](https://omgskills.com/library/astronomer/index.md) · 454 stars · Updated 2026-10-14

Migrates Airflow projects from airflow-ai-sdk to apache-airflow-providers-common-ai 0.4.0+. Use when replacing airflow-ai-sdk with the official Airflow AI provider - migrating LLM decorators (@task.llm, @task.agent, @task.llm\_branch, @task.embed), switching from model strings/objects to connection-based LLM configuration, updating imports from airflow\_ai\_sdk to the new provider, or upgrading an existing common-ai 0.1.x setup to 0.4.x (multimodal prompts, toolsets, embedding operators); also when common-ai provider, AIP-99, a pydanticai connection or migrating away from airflow-ai-sdk come up.

Tags: agentic-workflow, agents, ai, ai-agents, airflow, apache-airflow, claude, cursor, dag, data-engineering, data-pipelines, dbt, llm, mcp, orchestrator, skills, workflow-automation, workflow-management, workflow-orchestration, workflows

## Install

```sh
git clone https://github.com/astronomer/agents /tmp/agents && ln -s /tmp/agents/skills/migrating-ai-sdk-to-common-ai ~/.claude/skills/migrating-ai-sdk-to-common-ai
```

## From README

Migrate airflow-ai-sdk to apache-airflow-providers-common-ai This skill migrates Airflow projects from airflow-ai-sdk to apache-airflow-providers-common-ai (target 0.4.0+), the official Airflow AI provider built on PydanticAI. It also covers upgrading projects already on common-ai 0.1.x, since several capabilities (multimodal prompts, toolsets, embedding operators, structured-output XCom behavior) changed between 0.1.0 and 0.4.0. CRITICAL: The new provider requires Airflow 3.0+ and (for 0.4.0) pydantic-ai-slim \>= 1.71.0. The API surface has changed: LLM configuration moves from code (model strings/objects) to Airflow connections (pydanticai type). There is no @task.embed in the new provider; embeddings move to the LlamaIndex integration or a plain @task (see Step 3).

## Links

- HTML page: https://omgskills.com/library/astronomer/agents/migrating-ai-sdk-to-common-ai/
- GitHub: https://github.com/astronomer/agents
- Author: [@astronomer](https://omgskills.com/library/astronomer/index.md)

## More from this repo

- [airflow-adapter](https://omgskills.com/library/astronomer/agents/astro-airflow-mcp/dot-claude/skills/airflow-adapter/index.md) — Airflow adapter pattern for v2/v3 API compatibility.
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 459 stars
- [airflow-plugins](https://omgskills.com/library/astronomer/agents/skills/airflow-plugins/index.md) — Builds Airflow 3.1+ plugins that embed FastAPI apps, custom UI pages, React components, middleware, macros, and...
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 459 stars
- [analyzing-data](https://omgskills.com/library/astronomer/agents/skills/analyzing-data/index.md) — Queries the data warehouse with SQL and answers business questions about data.
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 459 stars