Task flow engine implementation

api/routers/workstreams.py

@@ -5,8 +5,13 @@ from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from api.database import get_session
-from api.models.workstream import Workstream, WorkstreamStatus
-from api.schemas.workstream import WorkstreamCreate, WorkstreamRead, WorkstreamUpdate
+from api.models.workstream import Workstream
+from api.schemas.workstream import (
+    WorkstreamCreate,
+    WorkstreamRead,
+    WorkstreamStatus,
+    WorkstreamUpdate,
+)
 
 router = APIRouter(prefix="/workstreams", tags=["workstreams"])
 
@@ -86,7 +91,7 @@ async def archive_workstream(
     ws = await session.get(Workstream, workstream_id)
     if ws is None:
         raise HTTPException(status_code=404, detail="Workstream not found")
-    ws.status = WorkstreamStatus.archived
+    ws.status = "archived"
     await session.commit()
     await session.refresh(ws)
     return ws