remove unused imports and rename migration file

Author: avirajsingh7

…ange_vector_store_id_to_vector_store_.py …e_vector_store_id_to_vector_store_ids.pybackend/app/alembic/versions/f2589428c1d0_change_vector_store_id_to_vector_store_.py renamed to backend/app/alembic/versions/f2589428c1d0_change_vector_store_id_to_vector_store_ids.py backend/app/alembic/versions/f2589428c1d0_change_vector_store_id_to_vector_store_.py renamed to backend/app/alembic/versions/f2589428c1d0_change_vector_store_id_to_vector_store_ids.py

@@ -7,7 +7,6 @@
 """
 from alembic import op
 import sqlalchemy as sa
-import sqlmodel.sql.sqltypes
 from sqlalchemy.dialects import postgresql
 
 # revision identifiers, used by Alembic.
@@ -37,15 +36,29 @@ def upgrade():
 
 
 def downgrade():
-    # ### commands auto generated by Alembic - please adjust! ###
+    # Add back the single vector_store_id column as nullable for safe data migration
     op.add_column(
         "openai_assistant",
         sa.Column(
             "vector_store_id",
             sa.VARCHAR(length=255),
             autoincrement=False,
-            nullable=False,
+            nullable=True,  # Allow nulls temporarily for safe migration
         ),
     )
+
+    op.execute(
+        """
+        UPDATE openai_assistant
+        SET vector_store_id = vector_store_ids[1]
+        WHERE vector_store_ids IS NOT NULL AND array_length(vector_store_ids, 1) > 0
+        """
+    )
+
     op.drop_column("openai_assistant", "vector_store_ids")
-    # ### end Alembic commands ###
+
+    op.alter_column(
+        "openai_assistant",
+        "vector_store_id",
+        existing_type=sa.VARCHAR(length=255),
+    )

backend/app/tests/api/routes/test_assistants.py

@@ -1,10 +1,7 @@
 import pytest
 from fastapi.testclient import TestClient
-from sqlmodel import Session
-from unittest.mock import MagicMock, patch
+from unittest.mock import patch
 from app.main import app
-from app.tests.utils.utils import random_email
-from app.core.security import get_password_hash
 from app.tests.utils.openai import mock_openai_assistant
 
 client = TestClient(app)
@@ -18,7 +15,6 @@ def normal_user_api_key_header():
 @patch("app.api.routes.assistants.fetch_assistant_from_openai")
 def test_ingest_assistant_success(
     mock_fetch_assistant,
-    db: Session,
     normal_user_api_key_header: str,
 ):
     """Test successful assistant ingestion from OpenAI."""
@@ -43,7 +39,6 @@ def test_ingest_assistant_success(
 @patch("app.api.routes.assistants.configure_openai")
 def test_ingest_assistant_openai_not_configured(
     mock_configure_openai,
-    db: Session,
     normal_user_api_key_header: dict,
 ):
     """Test assistant ingestion failure when OpenAI is not configured."""

backend/app/tests/utils/utils.py

@@ -1,7 +1,7 @@
 import random
 import string
 from uuid import UUID
-from typing import Optional, Type, TypeVar
+from typing import Type, TypeVar
 
 import pytest
 from fastapi.testclient import TestClient
@@ -59,7 +59,7 @@ def get_non_existent_id(session: Session, model: Type[T]) -> int:
     return (result or 0) + 1
 
 
-def get_project(session: Session, name: Optional[str] = None) -> Project:
+def get_project(session: Session, name: str | None = None) -> Project:
     """
     Retrieve an active project from the database.
 
@@ -69,11 +69,11 @@ def get_project(session: Session, name: Optional[str] = None) -> Project:
     if name:
         statement = (
             select(Project)
-            .where(Project.name == name, Project.is_active == True)
+            .where(Project.name == name, Project.is_active)
             .limit(1)
         )
     else:
-        statement = select(Project).where(Project.is_active == True).limit(1)
+        statement = select(Project).where(Project.is_active).limit(1)
 
     project = session.exec(statement).first()