jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 chore(alembic): add ApiKey table and remove FlowStyle and Component tables

Browse source 
🔄 refactor(alembic): downgrade migration to recreate FlowStyle and Component tables and remove User and ApiKey tables
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-23 21:43:14 -03:00
commit 63ca408506
2 changed files with 91 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

84
src/backend/langflow/alembic/versions/5512e39b4012_add_apikey_table.py Normal file
View file

@ -0,0 +1,84 @@
"""Add ApiKey table
Revision ID: 5512e39b4012
Revises: 0a534bdfd84b
Create Date: 2023-08-23 21:05:51.042203
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
import sqlmodel
# revision identifiers, used by Alembic.
revision: str = '5512e39b4012'
down_revision: Union[str, None] = '0a534bdfd84b'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_table('apikey',
sa.Column('api_key', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column('create_at', sa.DateTime(), nullable=False),
sa.Column('last_used_at', sa.DateTime(), nullable=True),
sa.Column('id', sqlmodel.sql.sqltypes.GUID(), nullable=False),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('id')
)
op.create_index(op.f('ix_apikey_api_key'), 'apikey', ['api_key'], unique=True)
op.create_table('user',
sa.Column('id', sqlmodel.sql.sqltypes.GUID(), nullable=False),
sa.Column('username', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column('password', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
sa.Column('is_active', sa.Boolean(), nullable=False),
sa.Column('is_superuser', sa.Boolean(), nullable=False),
sa.Column('create_at', sa.DateTime(), nullable=False),
sa.Column('updated_at', sa.DateTime(), nullable=False),
sa.Column('last_login_at', sa.DateTime(), nullable=True),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('id')
)
op.create_index(op.f('ix_user_username'), 'user', ['username'], unique=True)
op.drop_table('flowstyle')
op.drop_index('ix_component_frontend_node_id', table_name='component')
op.drop_index('ix_component_name', table_name='component')
op.drop_table('component')
# ### end Alembic commands ###
def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_table('component',
sa.Column('id', sa.CHAR(length=32), nullable=False),
sa.Column('frontend_node_id', sa.CHAR(length=32), nullable=False),
sa.Column('name', sa.VARCHAR(), nullable=False),
sa.Column('description', sa.VARCHAR(), nullable=True),
sa.Column('python_code', sa.VARCHAR(), nullable=True),
sa.Column('return_type', sa.VARCHAR(), nullable=True),
sa.Column('is_disabled', sa.BOOLEAN(), nullable=False),
sa.Column('is_read_only', sa.BOOLEAN(), nullable=False),
sa.Column('create_at', sa.DATETIME(), nullable=False),
sa.Column('update_at', sa.DATETIME(), nullable=False),
sa.PrimaryKeyConstraint('id')
)
op.create_index('ix_component_name', 'component', ['name'], unique=False)
op.create_index('ix_component_frontend_node_id', 'component', ['frontend_node_id'], unique=False)
op.create_table('flowstyle',
sa.Column('color', sa.VARCHAR(), nullable=False),
sa.Column('emoji', sa.VARCHAR(), nullable=False),
sa.Column('flow_id', sa.CHAR(length=32), nullable=True),
sa.Column('id', sa.CHAR(length=32), nullable=False),
sa.ForeignKeyConstraint(['flow_id'], ['flow.id'], ),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('id')
)
op.drop_index(op.f('ix_user_username'), table_name='user')
op.drop_table('user')
op.drop_index(op.f('ix_apikey_api_key'), table_name='apikey')
op.drop_table('apikey')
# ### end Alembic commands ###

18
src/backend/langflow/services/database/models/api_key.py
View file

@ -2,27 +2,23 @@ from sqlmodel import Field
from uuid import UUID, uuid4 from uuid import UUID, uuid4
from typing import Optional from typing import Optional
from datetime import datetime from datetime import datetime
from langflow.services.database.models.base import SQLModelSerializable, SQLModel from langflow.services.database.models.base import SQLModelSerializable
class ApiKeyBase(SQLModelSerializable): class ApiKeyBase(SQLModelSerializable):
api_key: str = Field(index=True, unique=True) api_key: str = Field(index=True, unique=True)
name: str = Field() name: Optional[str] = Field(index=True)
create_at: datetime = Field(default_factory=datetime.utcnow) create_at: datetime = Field(default_factory=datetime.utcnow)
last_used_at: Optional[datetime] = Field() last_used_at: Optional[datetime] = Field(default=None)
class ApiKey(ApiKeyBase, table=True): class ApiKey(ApiKeyBase, table=True):
id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True) id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True)
class ApiKeyCreate(SQLModel): class ApiKeyCreate(ApiKeyBase):
name: str = Field() pass
class ApiKeyRead(SQLModel): class ApiKeyRead(ApiKeyBase):
id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True) id: UUID
api_key: str = Field(index=True, unique=True)
name: str = Field()
create_at: datetime = Field(default_factory=datetime.utcnow)
last_used_at: Optional[datetime] = Field()