diff --git a/src/backend/langflow/alembic/versions/595c9c2a2ad4_changes_for_the_store.py b/src/backend/langflow/alembic/versions/595c9c2a2ad4_changes_for_the_store.py new file mode 100644 index 000000000..6ff6079f6 --- /dev/null +++ b/src/backend/langflow/alembic/versions/595c9c2a2ad4_changes_for_the_store.py @@ -0,0 +1,41 @@ +"""Changes for the store + +Revision ID: 595c9c2a2ad4 +Revises: eb5866d51fd2 +Create Date: 2023-10-18 16:27:37.781613 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa +import sqlmodel + + +# revision identifiers, used by Alembic. +revision: str = '595c9c2a2ad4' +down_revision: Union[str, None] = 'eb5866d51fd2' +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! ### + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.add_column(sa.Column('is_component', sa.Boolean(), nullable=False)) + + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.add_column(sa.Column('store_api_key', sqlmodel.sql.sqltypes.AutoString(), nullable=True)) + + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + with op.batch_alter_table('user', schema=None) as batch_op: + batch_op.drop_column('store_api_key') + + with op.batch_alter_table('flow', schema=None) as batch_op: + batch_op.drop_column('is_component') + + # ### end Alembic commands ### diff --git a/src/backend/langflow/services/database/models/flow/flow.py b/src/backend/langflow/services/database/models/flow/flow.py index e578f37c4..83d18b4eb 100644 --- a/src/backend/langflow/services/database/models/flow/flow.py +++ b/src/backend/langflow/services/database/models/flow/flow.py @@ -15,6 +15,7 @@ class FlowBase(SQLModelSerializable): name: str = Field(index=True) description: Optional[str] = Field(index=True) data: Optional[Dict] = Field(default=None, nullable=True) + is_component: bool = Field(default=False) @validator("data") def validate_json(v): diff --git a/src/backend/langflow/services/database/models/user/user.py b/src/backend/langflow/services/database/models/user/user.py index b1f514531..1f769c8ec 100644 --- a/src/backend/langflow/services/database/models/user/user.py +++ b/src/backend/langflow/services/database/models/user/user.py @@ -25,6 +25,7 @@ class User(SQLModelSerializable, table=True): back_populates="user", sa_relationship_kwargs={"cascade": "delete"}, ) + store_api_key: str = Field(default=None, nullable=True) flows: list["Flow"] = Relationship(back_populates="user")