code: def upgrade():
      # ### commands auto generated by Alembic - please adjust! ###
      op.create_table(
          "query_context",
          sa.Column("id", sa.Integer(), nullable=False),
          sa.Column("company_id", sa.Integer(), nullable=True),
          sa.Column("query", sa.String(), nullable=False),
          sa.Column("context_name", sa.String(), nullable=False),
          sa.Column("values", sa.JSON(), nullable=True),
          sa.Column("updated_at", sa.DateTime(), nullable=True),
          sa.Column("created_at", sa.DateTime(), nullable=True),
          sa.PrimaryKeyConstraint("id"),
      )
  
imports: from alembic import op
  import sqlalchemy
  import mindsdb.interfaces.storage.db
label_desc: Use image node_id 1 for calling a global function with example usage: upgrade() without return types