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 typing import Optional
from datetime import datetime
from langflow.services.database.models.base import SQLModelSerializable, SQLModel
from langflow.services.database.models.base import SQLModelSerializable
class ApiKeyBase(SQLModelSerializable):
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)
last_used_at: Optional[datetime] = Field()
last_used_at: Optional[datetime] = Field(default=None)
class ApiKey(ApiKeyBase, table=True):
id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True)
class ApiKeyCreate(SQLModel):
name: str = Field()
class ApiKeyCreate(ApiKeyBase):
pass
class ApiKeyRead(SQLModel):
id: UUID = Field(default_factory=uuid4, primary_key=True, unique=True)
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()
class ApiKeyRead(ApiKeyBase):
id: UUID