This is an automated email from the ASF dual-hosted git repository.
kaxil pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 5aeb726b1f9 Sync RuntimeTaskInstanceProtocol with RuntimeTaskInstance
(#67216)
5aeb726b1f9 is described below
commit 5aeb726b1f9146e81c3549a6d34073b7b09fd895
Author: Kaxil Naik <[email protected]>
AuthorDate: Wed May 20 09:31:01 2026 +0100
Sync RuntimeTaskInstanceProtocol with RuntimeTaskInstance (#67216)
---
task-sdk/src/airflow/sdk/types.py | 19 +++++++++++++++++--
1 file changed, 17 insertions(+), 2 deletions(-)
diff --git a/task-sdk/src/airflow/sdk/types.py
b/task-sdk/src/airflow/sdk/types.py
index 3a00fea6d96..711dbe71ca4 100644
--- a/task-sdk/src/airflow/sdk/types.py
+++ b/task-sdk/src/airflow/sdk/types.py
@@ -30,6 +30,7 @@ __all__ = ["TaskInstance", "TaskInstanceKey"]
if TYPE_CHECKING:
from collections.abc import Iterator
+ import jinja2
from pydantic import AwareDatetime, JsonValue
from airflow.models.taskinstance import TaskInstance as
SchedulerTaskInstance
@@ -141,10 +142,18 @@ class RuntimeTaskInstanceProtocol(Protocol):
start_date: AwareDatetime
end_date: AwareDatetime | None = None
state: TaskInstanceState | None = None
+ is_mapped: bool | None = None
+ rendered_map_index: str | None = None
+
+ @property
+ def log_url(self) -> str: ...
+
+ @property
+ def mark_success_url(self) -> str: ...
def xcom_pull(
self,
- task_ids: str | list[str] | None = None,
+ task_ids: str | Iterable[str] | None = None,
dag_id: str | None = None,
key: str = BaseXCom.XCOM_RETURN_KEY,
include_prior_dates: bool = False,
@@ -158,7 +167,13 @@ class RuntimeTaskInstanceProtocol(Protocol):
def get_template_context(self) -> Context: ...
- def get_first_reschedule_date(self, first_try_number) -> AwareDatetime |
None: ...
+ def render_templates(
+ self,
+ context: Context | None = None,
+ jinja_env: jinja2.Environment | None = None,
+ ) -> BaseOperator: ...
+
+ def get_first_reschedule_date(self, context: Context) -> AwareDatetime |
None: ...
def get_previous_dagrun(self, state: str | None = None) -> DagRunProtocol
| None: ...
