Update src/services/interview_history_router.py

src/services/interview_history_router.py

@@ -42,12 +42,18 @@ async def create_interview_history(history: InterviewHistoryCreate, db: AsyncIOM
 
 @router.get("/interview-histories/{history_id}", response_model=InterviewHistoryResponse)
 async def get_interview_history_by_id(history_id: str, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
-    history = await InterviewHistoryModel.get(db, InterviewHistoryModel.collection_name, {"_id": history_id})
+    if not ObjectId.is_valid(history_id):
+        raise HTTPException(status_code=400, detail="Invalid ID format")
+    
+    history = await InterviewHistoryModel.get(db, InterviewHistoryModel.collection_name, {"_id": ObjectId(history_id)})
     if history is None:
         raise HTTPException(status_code=404, detail="Interview history not found")
     return history
 
 @router.put("/interview-histories/{history_id}", response_model=InterviewHistoryResponse)
 async def update_interview_history(history_id: str, history: InterviewHistoryUpdate, db: AsyncIOMotorDatabase = Depends(lambda: mongo_db)):
-    await InterviewHistoryModel.update(db, InterviewHistoryModel.collection_name, {"_id": history_id}, history.model_dump(exclude_unset=True))
+    if not ObjectId.is_valid(history_id):
+        raise HTTPException(status_code=400, detail="Invalid ID format")
+        
+    await InterviewHistoryModel.update(db, InterviewHistoryModel.collection_name, {"_id": ObjectId(history_id)}, history.model_dump(exclude_unset=True))
     return await get_interview_history_by_id(history_id, db)