2025_10_23_1736-7737c7eba9c4_server_proxy_improvement.py 1007 B

123456789101112131415161718192021222324252627282930313233343536
  1. """Server Proxy Improvement
  2. Revision ID: 7737c7eba9c4
  3. Revises: eca16ce6dedd
  4. Create Date: 2025-10-23 17:36:07.871032
  5. """
  6. from typing import Sequence, Union
  7. from alembic import op
  8. import sqlalchemy as sa
  9. import sqlmodel
  10. import gpustack
  11. # revision identifiers, used by Alembic.
  12. revision: str = '7737c7eba9c4'
  13. down_revision: Union[str, None] = 'eca16ce6dedd'
  14. branch_labels: Union[str, Sequence[str], None] = None
  15. depends_on: Union[str, Sequence[str], None] = None
  16. def upgrade() -> None:
  17. with op.batch_alter_table("clusters", schema=None) as batch_op:
  18. batch_op.add_column(
  19. sa.Column('gateway_endpoint', sa.String(length=255), nullable=True)
  20. )
  21. batch_op.add_column(
  22. sa.Column('reported_gateway_endpoint', sa.String(length=255), nullable=True)
  23. )
  24. def downgrade() -> None:
  25. with op.batch_alter_table("clusters", schema=None) as batch_op:
  26. batch_op.drop_column('gateway_endpoint')
  27. batch_op.drop_column('reported_gateway_endpoint')