Hello, Astronomer just open sourced decorators that allow Airflow to orchestrate LLM agent calls, which is a game changer. Any possibility of Metaflow doing the same? https://github.com/astronomer/airflow-ai-sdk

what features of it you find appealing? It seems like a thin wrapped over Pydantic Ai which you can easily call from Metaflow today. Calling a Python library in a Metaflow step is straightforward. Changing flow execution based on LLM output is potentially more interesting but based on a quick look at their

branch

example, they just route tickets to different dag branches by splitting data, which is easy to do in Metaflow today. Admittedly I took only a quick look so maybe I am missing something 🤔