Changed db schema for more accuracy

Author: abhishekbuild

app/api/v1/multimodal.py

@@ -140,7 +140,7 @@ async def multimodal_chat(
 
     # Step 3: Audio Output
     if audio_output:
-        # Convert text to audio and Upload on S3
+        # Convert text to audio and upload to S3
         audio_output_service = AudioOutput()
         audio_s3_url = await audio_output_service.convert_text_into_audio(
             assistant_content=assistant_content,
@@ -149,6 +149,18 @@ async def multimodal_chat(
 
         response_payload["audio_output_url"] = audio_s3_url
 
+        # Save assistant audio as an attachment in DB
+        await create_attachment(
+            db=db,
+            session_id=session.id,
+            message_id=assistant_msg.id,
+            url=file_url,
+            media_type=media_type,
+            metadata_={"voice_style": voice_style.value},
+            audio_url=audio_s3_url
+        )
+
+
     # Add media file link if uploaded
     if file_url:
         response_payload["uploaded_file_url"] = file_url

app/crud/attachments.py

@@ -1,9 +1,25 @@
+from app.models.attachment import Attachment
 from sqlalchemy.ext.asyncio import AsyncSession
-from app.models.attachment import Attachment, MediaType
 
-async def create_attachment(db: AsyncSession, session_id, message_id, url: str, media_type: MediaType, metadata: dict = None) -> Attachment:
-    attach = Attachment(session_id=session_id, message_id=message_id, url=url, media_type=media_type, metadata=metadata)
-    db.add(attach)
+
+async def create_attachment(
+    db: AsyncSession,
+    session_id,
+    message_id,
+    url,
+    media_type,
+    metadata_=None,
+    audio_url=None,
+):
+    attachment = Attachment(
+        session_id=session_id,
+        message_id=message_id,
+        url=url,
+        media_type=media_type,
+        metadata_=metadata_,
+        audio_url=audio_url,
+    )
+    db.add(attachment)
     await db.commit()
-    await db.refresh(attach)
-    return attach
+    await db.refresh(attachment)
+    return attachment

app/models/attachment.py

@@ -28,8 +28,8 @@ class Attachment(Base):
         nullable=True,
     )
 
-    url = Column(String, nullable=False)
-    media_type = Column(Enum(MediaType), nullable=False)
+    url = Column(String, nullable=True)
+    media_type = Column(Enum(MediaType), nullable=True)
     metadata_ = Column(JSONB, nullable=True)
 
     # To store generated assistant audio responses

migrations/versions/0adeb6494274_make_media_type_nullable.py

@@ -0,0 +1,36 @@
+"""Make media_type nullable
+
+Revision ID: 0adeb6494274
+Revises: f8e5c60d6d7c
+Create Date: 2025-10-05 17:50:45.936151
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+# revision identifiers, used by Alembic.
+revision: str = '0adeb6494274'
+down_revision: Union[str, Sequence[str], None] = 'f8e5c60d6d7c'
+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('attachments', 'media_type',
+               existing_type=postgresql.ENUM('image', 'audio', name='mediatype'),
+               nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('attachments', 'media_type',
+               existing_type=postgresql.ENUM('image', 'audio', name='mediatype'),
+               nullable=False)
+    # ### end Alembic commands ###

migrations/versions/fa1dbf5dd801_make_media_type_nullable.py

@@ -0,0 +1,36 @@
+"""Make media_type nullable
+
+Revision ID: fa1dbf5dd801
+Revises: 0adeb6494274
+Create Date: 2025-10-05 17:55:01.776342
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'fa1dbf5dd801'
+down_revision: Union[str, Sequence[str], None] = '0adeb6494274'
+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('attachments', 'url',
+               existing_type=sa.VARCHAR(),
+               nullable=True)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('attachments', 'url',
+               existing_type=sa.VARCHAR(),
+               nullable=False)
+    # ### end Alembic commands ###