Author dags with the Astro IDE

Private Preview
This feature is in Private 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:

  • Manually edit Python scripts, dag files, and supporting resources in the built-in editor.
  • Use 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.

Add custom project rules for code generation

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

1

Open Project Settings

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

2

Add custom instructions

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

3

Save your changes

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