fix: add folder_id and sort_order to AIChatSession model, fix UUID conversion in routes, add migration 0003

app/plugins/builtins/ai_assistant/routes.py

@@ -457,11 +457,16 @@ async def create_session(
         if default_agent:
             agent_id = default_agent.id
 
+    folder_id = None
+    if data.folder_id:
+        folder_id = uuid.UUID(data.folder_id)
+
     session = AIChatSession(
         user_id=user_id,
         agent_id=agent_id,
         title=data.title,
         is_sidebar=data.is_sidebar,
+        folder_id=folder_id,
         tenant_id=tenant_id,
     )
     db.add(session)
@@ -486,6 +491,11 @@ async def update_session(
     update_data = data.model_dump(exclude_unset=True)
     if "agent_id" in update_data and update_data["agent_id"]:
         update_data["agent_id"] = uuid.UUID(update_data["agent_id"])
+    if "folder_id" in update_data:
+        if update_data["folder_id"]:
+            update_data["folder_id"] = uuid.UUID(update_data["folder_id"])
+        else:
+            update_data["folder_id"] = None
     for field, val in update_data.items():
         setattr(session, field, val)
     await db.commit()
@@ -643,8 +653,11 @@ async def update_folder(
         raise HTTPException(status_code=404, detail="Folder not found")
 
     update_data = data.model_dump(exclude_unset=True)
-    if "parent_id" in update_data and update_data["parent_id"]:
-        update_data["parent_id"] = uuid.UUID(update_data["parent_id"])
+    if "parent_id" in update_data:
+        if update_data["parent_id"]:
+            update_data["parent_id"] = uuid.UUID(update_data["parent_id"])
+        else:
+            update_data["parent_id"] = None
     for field, val in update_data.items():
         setattr(folder, field, val)
     await db.commit()