# 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/skills/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-state-store](https://omgskills.com/skills/astronomer/agents/skills/airflow-state-store/index.md) — Persists task and asset state across retries and DAG runs using Airflow 3.3's AIP-103 key/value stores...
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 456 stars
- [authoring-go-sdk-tasks](https://omgskills.com/skills/astronomer/agents/skills/authoring-go-sdk-tasks/index.md) — Writes Airflow task logic in Go using the Airflow Go SDK.
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 456 stars
- [authoring-java-sdk-tasks](https://omgskills.com/skills/astronomer/agents/skills/authoring-java-sdk-tasks/index.md) — Writes Airflow task logic in Java, Kotlin, or any JVM language using the Airflow Java SDK.
   - [@astronomer](https://omgskills.com/library/astronomer/index.md) · 456 stars