[josix]

TypeScript task handlers had no access to the per-task-instance state store that Python tasks reach through context["task_state_store"], so a retried TypeScript task could not reconnect to an external job an earlier attempt had submitted.

The supervisor and the generated wire types already carry the four state-store messages. This change exposes them through the TypeScript client, covers them in the example bundle and the ts_sdk e2e mode, and marks task-state-store as supported in the compatibility matrix.

getClient().taskStateStore returns a store scoped to the running task instance, with the same four operations as the Python accessor and the Java and Go stores:

import { getClient, NEVER_EXPIRE } from "apache-airflow-ts-sdk";

export async function runSparkJob() {
  const store = getClient().taskStateStore;

  let jobId = await store.get<string>("job_id");
  if (jobId == null) {
    jobId = await submitSparkJob();
    await store.set("job_id", jobId, { retentionMs: NEVER_EXPIRE });
  }

  const result = await waitForSparkJob(jobId);
  await store.delete("job_id");
  return result;
}

• get<T>(key) returns the value, or null when the key is missing.
• set(key, value, { retentionMs? }) stores a value. Omitting retentionMs follows the deployment's [state_store] default_retention_days, and NEVER_EXPIRE stores the key with no expiry.
• delete(key) removes one key, and resolves when the key does not exist.
• clear() removes every key for the current task instance.

Related: #73420 (Go SDK), #73464 (Java SDK).

---

Was generative AI tooling used to co-author this PR?

• Yes (Claude Code Fable 5.1)

---

• Read the Pull Request Guidelines for more information. Note: commit author/co-author name and email in commits become permanently public when merged.
• For fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
• When adding dependency, check compliance with the ASF 3rd Party License Policy.
• For significant user-facing changes create newsfragment: {pr_number}.significant.rst, in airflow-core/newsfragments. You can add this file in a follow-up commit after the PR is created so you know the PR number.

[jason810496]

Nice, thanks! LGTM overall.

Let's also add a TODO before the serialization to respect the state_store/max_value_storage_bytes that will be passed from the coordinator side.

[jason810496]

I just fixed the review comments myself.