Apache Airflow를 위한 Common AI Provider 출시: LLM 및 AI 에이전트 지원

@task.llm(
    llm_conn_id="my_openai_conn",
    system_prompt="Extract named entities.",
    output_type=Entities,
)
def extract(text: str):
    return f"Extract entities from: {text}"

@task.agent(
    llm_conn_id="my_openai_conn",
    system_prompt="You are a SQL analyst. Use tools to answer questions with data.",
    toolsets=[
        SQLToolset(
            db_conn_id="postgres_default",
            allowed_tables=["customers", "orders"],
            max_rows=20,
        )
    ],
)
def analyze(question: str):
    return f"Answer this question about our data: {question}"