code: def upgrade():
      """Apply Make XCom primary key columns non-nullable"""
      conn = op.get_bind()
      with op.batch_alter_table("xcom") as bop:
          bop.alter_column(
              "key", type_=StringID(length=512), nullable=False
          )
          bop.alter_column(
              "execution_date", type_=TIMESTAMP, nullable=False
          )
          if conn.dialect.name == "mssql":
              bop.create_primary_key(
                  constraint_name="pk_xcom",
                  columns=[
                      "dag_id",
                      "task_id",
                      "key",
                      "execution_date",
                  ],
              )
  
imports: from __future__ import annotations
  from alembic import op
  from airflow.migrations.db_types import TIMESTAMP, StringID
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types