airflow.providers.databricks.operators.agent¶

Operators for invoking Databricks agents.

Classes¶

DatabricksAgentInvokeOperator

Invoke a DurableAgentServer on Databricks Apps and optionally wait for its result.

Module Contents¶

class airflow.providers.databricks.operators.agent.DatabricksAgentInvokeOperator(*, app_url, input, databricks_conn_id='databricks_default', session_id=None, invocation_id=None, wait_for_termination=True, polling_period_seconds=10, timeout=3600, deferrable=conf.getboolean('operators', 'default_deferrable', fallback=False), **kwargs)[source]¶

Bases: airflow.providers.common.compat.sdk.BaseOperator

Invoke a DurableAgentServer on Databricks Apps and optionally wait for its result.

Parameters:

app_url (str) – HTTPS base URL of the deployed app. (templated)
input (Any) – JSON-serializable agent input. (templated)
databricks_conn_id (str) – Databricks connection using service principal OAuth or OAuth token federation. Defaults to databricks_default. (templated)
session_id (str | None) – Conversation ID. CLI template agents require this field. Defaults to None. (templated)
invocation_id (str | None) – Idempotency UUID. Defaults to a stable UUID for this task instance, app URL, input and session, shared across retries and clears with unchanged input and session. A stored failure requires a new UUID to run again. (templated)
wait_for_termination (bool) – Wait for completion or interruption. If false, return the submission response. Defaults to True.
polling_period_seconds (float) – Seconds between status checks. Defaults to 10.
timeout (float) – Maximum seconds to wait after submission. Timing out does not cancel the remote invocation. Defaults to 3600.
deferrable (bool) – Release the worker while waiting. Defaults to [operators] default_deferrable, with a fallback of False.