astronomer / astro-cli / 6e9dd8d1-3b8b-47cc-a01f-b67fb42c13d6

40%

main: 45%

Build Type

Pull #2100

circleci

Committed by jlaneve

Commit Message

Add `astro agent` command with Otto binary management

Adds a new top-level CLI command that downloads and launches Otto,
Astronomer's AI coding agent for Airflow. Introduces a shared
`pkg/agent/` package so other tools can reuse the same binary lifecycle.

Commands:

    astro agent                        # interactive TUI
    astro agent "fix my DAG"           # one-shot with a prompt
    astro agent -c                     # continue last session
    astro agent update                 # force-update Otto
    astro agent version                # show installed version

Otto auto-updates on launch by default (cached-latest > installed →
sync download + swap). Opt out with `astro config set -g
agent.auto_update false`.

Surrounding code fixes required to make the agent reliable:

- `config/config.go`: `saveConfig` wraps viper writes in a `gofrs/flock`
  so concurrent astro processes can't corrupt `~/.astro/config.yaml`.
  `gofrs/flock` was already a transitive dep.
- `config/context.go`: `SetContextKey` / `SetExpiresIn` work around
  viper#1106 — `viper.Set` on a nested path leaves the override layer
  holding a partial tree, so later `UnmarshalKey` silently zeros every
  untouched sibling.
- `airflow/proxy/`: `DetectAirflow` prefers the proxy hostname URL
  over the direct container port, which rotates on every
  `astro dev restart`.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Pull Request Pull Request #2100: Add `astro agent` command with Otto binary management

Coverage Stats

250 of 562 new or added lines in 11 files covered. (44.48%)

2 existing lines in 2 files now uncovered.

25340 of 64074 relevant lines covered (39.55%)

9.56 hits per line