repair pretest

app/api/v1/endpoints/pretest.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, Depends, UploadFile, File, Form, Header
+from fastapi import APIRouter, Depends, UploadFile, File, Form, Header, HTTPException
 from sqlalchemy.ext.asyncio import AsyncSession
 from app.core.database import get_db
 from app.repositories.material_repository import MaterialRepository
@@ -20,7 +20,11 @@ async def check_onboarding(
     """Cek apakah user sudah pretest"""
     talent_repo = TalentRepository(db)
     talent = await talent_repo.get_by_id(current_user["idtalent"])
-    show_onboarding = talent.pretest_score is None if talent else False
+    
+    if not talent:
+        raise HTTPException(status_code=404, detail="User profile not found")
+    show_onboarding = talent.pretest_score is None
+    
     return ResponseBase(data={"show_onboarding": show_onboarding})
 
 @router.get("/start", response_model=ResponseBase)
@@ -63,7 +67,7 @@ async def compare_pretest(
 
 @router.post("/submit", response_model=ResponseBase)
 async def submit_pretest(
-    payload: dict, # {"score": float}
+    payload: dict,
     current_user: dict = Depends(get_current_user),
     db: AsyncSession = Depends(get_db)
 ):