Author dags with the Astro IDE

This feature is in Public Preview.

The Astro IDE supports building, updating, understanding, and customizing Airflow dags with or without the Astronomer AI assistant.

Author dag code with or without AI

The Astro IDE allows you to write or modify dag code directly. Or, you can use AI-powered features to generate or refactor code, including:

Use context-aware AI tools to suggest, generate, or improve dag code, operator logic, and project structure.
Ask the Astro IDE questions about the project to better understand existing dag code structure, logic, and dependencies.
View the Astro IDE’s reasoning, streaming live in the AI panel as it proposes changes.
Review all proposed code changes before choosing to accept or reject them.
Manually edit Python scripts, dag files, and supporting resources in the built-in editor.

When you author dags in the Astro IDE, this consumes some of your Workspace AI credits. See the Public Preview FAQs for more details.

You can define custom instructions or rules that guide the AI for all code generation tasks within your Airflow project.

Open the Project Settings from the More Actions menu in the toolbar.

Enter your linting rules, conventions, or any special instructions for the Astro IDE in the Custom Instructions field.

Click Update Project to apply your custom rules. These instructions will be used for all AI-generated code in this project.