Merge pull request #30 from yuvabe-ai-labs/feat/chatbot

alembic/versions/5b5a3c7a6255_update_cascade_deletion.py

@@ -0,0 +1,99 @@
+"""update: cascade deletion
+
+Revision ID: 5b5a3c7a6255
+Revises: 52828bff621c
+Create Date: 2025-11-28 11:27:38.960269
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision: str = '5b5a3c7a6255'
+down_revision: Union[str, Sequence[str], None] = '52828bff621c'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(op.f('assets_user_id_fkey'), 'assets', type_='foreignkey')
+    op.create_foreign_key(None, 'assets', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.drop_constraint(op.f('comments_user_id_fkey'), 'comments', type_='foreignkey')
+    op.drop_constraint(op.f('comments_post_id_fkey'), 'comments', type_='foreignkey')
+    op.create_foreign_key(None, 'comments', 'posts', ['post_id'], ['id'], ondelete='CASCADE')
+    op.create_foreign_key(None, 'comments', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.drop_constraint(op.f('emotion_logs_user_id_fkey'), 'emotion_logs', type_='foreignkey')
+    op.create_foreign_key(None, 'emotion_logs', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.drop_constraint(op.f('knowledge_chunk_kb_id_fkey'), 'knowledge_chunk', type_='foreignkey')
+    op.create_foreign_key(None, 'knowledge_chunk', 'knowledge_base', ['kb_id'], ['id'], ondelete='CASCADE')
+    op.alter_column('leave', 'mentor_id',
+               existing_type=sa.UUID(),
+               nullable=True)
+    op.alter_column('leave', 'lead_id',
+               existing_type=sa.UUID(),
+               nullable=True)
+    op.drop_constraint(op.f('leave_lead_id_fkey'), 'leave', type_='foreignkey')
+    op.drop_constraint(op.f('leave_user_id_fkey'), 'leave', type_='foreignkey')
+    op.drop_constraint(op.f('leave_mentor_id_fkey'), 'leave', type_='foreignkey')
+    op.create_foreign_key(None, 'leave', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.create_foreign_key(None, 'leave', 'users', ['mentor_id'], ['id'], ondelete='SET NULL')
+    op.create_foreign_key(None, 'leave', 'users', ['lead_id'], ['id'], ondelete='SET NULL')
+    op.drop_constraint(op.f('likes_user_id_fkey'), 'likes', type_='foreignkey')
+    op.drop_constraint(op.f('likes_post_id_fkey'), 'likes', type_='foreignkey')
+    op.create_foreign_key(None, 'likes', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.create_foreign_key(None, 'likes', 'posts', ['post_id'], ['id'], ondelete='CASCADE')
+    op.drop_constraint(op.f('user_devices_user_id_fkey'), 'user_devices', type_='foreignkey')
+    op.create_foreign_key(None, 'user_devices', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.drop_constraint(op.f('user_teams_role_team_id_fkey'), 'user_teams_role', type_='foreignkey')
+    op.drop_constraint(op.f('user_teams_role_role_id_fkey'), 'user_teams_role', type_='foreignkey')
+    op.drop_constraint(op.f('user_teams_role_user_id_fkey'), 'user_teams_role', type_='foreignkey')
+    op.create_foreign_key(None, 'user_teams_role', 'teams', ['team_id'], ['id'], ondelete='CASCADE')
+    op.create_foreign_key(None, 'user_teams_role', 'users', ['user_id'], ['id'], ondelete='CASCADE')
+    op.create_foreign_key(None, 'user_teams_role', 'roles', ['role_id'], ['id'], ondelete='CASCADE')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, 'user_teams_role', type_='foreignkey')
+    op.drop_constraint(None, 'user_teams_role', type_='foreignkey')
+    op.drop_constraint(None, 'user_teams_role', type_='foreignkey')
+    op.create_foreign_key(op.f('user_teams_role_user_id_fkey'), 'user_teams_role', 'users', ['user_id'], ['id'])
+    op.create_foreign_key(op.f('user_teams_role_role_id_fkey'), 'user_teams_role', 'roles', ['role_id'], ['id'])
+    op.create_foreign_key(op.f('user_teams_role_team_id_fkey'), 'user_teams_role', 'teams', ['team_id'], ['id'])
+    op.drop_constraint(None, 'user_devices', type_='foreignkey')
+    op.create_foreign_key(op.f('user_devices_user_id_fkey'), 'user_devices', 'users', ['user_id'], ['id'])
+    op.drop_constraint(None, 'likes', type_='foreignkey')
+    op.drop_constraint(None, 'likes', type_='foreignkey')
+    op.create_foreign_key(op.f('likes_post_id_fkey'), 'likes', 'posts', ['post_id'], ['id'])
+    op.create_foreign_key(op.f('likes_user_id_fkey'), 'likes', 'users', ['user_id'], ['id'])
+    op.drop_constraint(None, 'leave', type_='foreignkey')
+    op.drop_constraint(None, 'leave', type_='foreignkey')
+    op.drop_constraint(None, 'leave', type_='foreignkey')
+    op.create_foreign_key(op.f('leave_mentor_id_fkey'), 'leave', 'users', ['mentor_id'], ['id'])
+    op.create_foreign_key(op.f('leave_user_id_fkey'), 'leave', 'users', ['user_id'], ['id'])
+    op.create_foreign_key(op.f('leave_lead_id_fkey'), 'leave', 'users', ['lead_id'], ['id'])
+    op.alter_column('leave', 'lead_id',
+               existing_type=sa.UUID(),
+               nullable=False)
+    op.alter_column('leave', 'mentor_id',
+               existing_type=sa.UUID(),
+               nullable=False)
+    op.drop_constraint(None, 'knowledge_chunk', type_='foreignkey')
+    op.create_foreign_key(op.f('knowledge_chunk_kb_id_fkey'), 'knowledge_chunk', 'knowledge_base', ['kb_id'], ['id'])
+    op.drop_constraint(None, 'emotion_logs', type_='foreignkey')
+    op.create_foreign_key(op.f('emotion_logs_user_id_fkey'), 'emotion_logs', 'users', ['user_id'], ['id'])
+    op.drop_constraint(None, 'comments', type_='foreignkey')
+    op.drop_constraint(None, 'comments', type_='foreignkey')
+    op.create_foreign_key(op.f('comments_post_id_fkey'), 'comments', 'posts', ['post_id'], ['id'])
+    op.create_foreign_key(op.f('comments_user_id_fkey'), 'comments', 'users', ['user_id'], ['id'])
+    op.drop_constraint(None, 'assets', type_='foreignkey')
+    op.create_foreign_key(op.f('assets_user_id_fkey'), 'assets', 'users', ['user_id'], ['id'])
+    # ### end Alembic commands ###

alembic/versions/e96769f268bc_fix_post_s_user_cascade.py

@@ -0,0 +1,41 @@
+"""fix: post's user cascade
+
+Revision ID: e96769f268bc
+Revises: 5b5a3c7a6255
+Create Date: 2025-11-28 11:58:45.282329
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel.sql.sqltypes
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'e96769f268bc'
+down_revision: Union[str, Sequence[str], None] = '5b5a3c7a6255'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('posts', 'user_id',
+               existing_type=sa.UUID(),
+               nullable=True)
+    op.drop_constraint(op.f('posts_user_id_fkey'), 'posts', type_='foreignkey')
+    op.create_foreign_key(None, 'posts', 'users', ['user_id'], ['id'], ondelete='SET NULL')
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, 'posts', type_='foreignkey')
+    op.create_foreign_key(op.f('posts_user_id_fkey'), 'posts', 'users', ['user_id'], ['id'])
+    op.alter_column('posts', 'user_id',
+               existing_type=sa.UUID(),
+               nullable=False)
+    # ### end Alembic commands ###

src/chatbot/models.py

@@ -4,7 +4,9 @@ from typing import List
 
 from pgvector.sqlalchemy import Vector
 from sqlalchemy import Column
-from sqlmodel import Field, Relationship, SQLModel
+from sqlmodel import Field, Relationship, SQLModel, ForeignKey
+
+from sqlalchemy.dialects.postgresql import UUID
 
 
 class KnowledgeBase(SQLModel, table=True):
@@ -21,7 +23,12 @@ class KnowledgeBase(SQLModel, table=True):
 class KnowledgeChunk(SQLModel, table=True):
     __tablename__ = "knowledge_chunk"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    kb_id: uuid.UUID = Field(foreign_key="knowledge_base.id", nullable=False)
+    kb_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("knowledge_base.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     chunk_index: int
     chunk_text: str
     embedding: List[float] = Field(sa_column=Column(Vector(768)))

src/core/models.py

@@ -6,7 +6,8 @@ from enum import Enum
 from typing import List, Optional
 
 
-from sqlalchemy import CheckConstraint, UniqueConstraint
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy import CheckConstraint, UniqueConstraint, ForeignKey
 from sqlmodel import Field, Relationship, SQLModel
 
 
@@ -48,15 +49,34 @@ class Roles(SQLModel, table=True):
 class UserTeamsRole(SQLModel, table=True):
     __tablename__ = "user_teams_role"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
-    team_id: uuid.UUID = Field(foreign_key="teams.id", nullable=False)
-    role_id: uuid.UUID = Field(foreign_key="roles.id", nullable=False)
-
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
+    team_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("teams.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
+    role_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("roles.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
 
 class Assets(SQLModel, table=True):
     __tablename__ = "assets"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     name: str = Field(nullable=False)
     type: str = Field(nullable=False)
     status: AssetStatus = Field(default=AssetStatus.UNAVAILABLE)
@@ -71,7 +91,12 @@ class EmotionLogs(SQLModel, table=True):
         CheckConstraint("evening_emotion BETWEEN 1 AND 7 or evening_emotion IS NULL"),
     )
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     morning_emotion: Optional[int] = Field(default=None, ge=1, le=7)
     evening_emotion: Optional[int] = Field(default=None, ge=1, le=7)
     log_date: date = Field(default_factory=date.today)

src/feed/models.py

@@ -3,8 +3,9 @@ from datetime import datetime
 from enum import Enum
 from typing import Optional
 
-from sqlalchemy import UniqueConstraint
+from sqlalchemy import UniqueConstraint, ForeignKey, Column
 from sqlmodel import Field, SQLModel
+from sqlalchemy.dialects.postgresql import UUID
 
 
 class PostType(str, Enum):
@@ -22,7 +23,12 @@ class PostCategory(str, Enum):
 class Posts(SQLModel, table=True):
     __tablename__ = "posts"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True
+        )
+    )
     type: PostType = Field(default=PostType.NOTICE)
     category: PostCategory = Field(default=PostCategory.GLOBAL)
     caption: Optional[str] = None
@@ -34,8 +40,18 @@ class Posts(SQLModel, table=True):
 class Comments(SQLModel, table=True):
     __tablename__ = "comments"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    post_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("posts.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     comment: str = Field(nullable=False)
     created_at: datetime = Field(default_factory=datetime.now, nullable=False)
 
@@ -44,6 +60,16 @@ class Likes(SQLModel, table=True):
     __tablename__ = "likes"
     __table_args__ = (UniqueConstraint("user_id", "post_id"),)
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    post_id: uuid.UUID = Field(foreign_key="posts.id", nullable=False)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    post_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("posts.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     liked_at: datetime = Field(default_factory=datetime.now, nullable=False)

src/profile/models.py

@@ -5,7 +5,8 @@ from datetime import date, datetime
 from enum import Enum
 from typing import List, Optional
 
-from sqlmodel import Field, Relationship, SQLModel
+from sqlalchemy.dialects.postgresql import UUID
+from sqlmodel import Field, Relationship, SQLModel, ForeignKey
 
 class LeaveType(str, Enum):
     SICK = "Sick"
@@ -21,9 +22,24 @@ class LeaveStatus(str, Enum):
 class Leave(SQLModel, table=True):
     __tablename__ = "leave"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
-    mentor_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
-    lead_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
+    mentor_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True
+        )
+    )
+    lead_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="SET NULL"),
+            nullable=True
+        )
+    )
     leave_type: LeaveType = Field(default=LeaveType.SICK)
     from_date: date = Field(nullable=False)
     to_date: date = Field(nullable=False)
@@ -39,7 +55,12 @@ class Leave(SQLModel, table=True):
 class UserDevices(SQLModel, table=True):
     __tablename__ = "user_devices"
     id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True)
-    user_id: uuid.UUID = Field(foreign_key="users.id", nullable=False)
+    user_id: uuid.UUID = Field(
+        sa_column=Column(UUID(as_uuid=True),
+            ForeignKey("users.id", ondelete="CASCADE"),
+            nullable=False
+        )
+    )
     device_token: str
     last_seen: datetime = Field(default_factory=datetime.now)
     device_model: str