init auth

api/endpoints/generation_router.py

@@ -11,11 +11,15 @@ from api.models.GenerationRequest import GenerationResponse, GenerationRequest,
 from api.service.generation_service import GenerationService
 from models.Generation import Generation
 
+from starlette import status
+
 import logging
 
 logger = logging.getLogger(__name__)
 
-router = APIRouter(prefix='/api/generations', tags=["Generation"])
+from api.endpoints.auth import get_current_user
+
+router = APIRouter(prefix='/api/generations', tags=["Generation"], dependencies=[Depends(get_current_user)])
 
 
 @router.post("/prompt-assistant", response_model=PromptResponse)
@@ -69,3 +73,12 @@ async def get_generation(generation_id: str,
 async def get_running_generations(request: Request,
                                   generation_service: GenerationService = Depends(get_generation_service)):
     return await generation_service.get_running_generations()
+
+
+@router.delete("/{generation_id}", status_code=status.HTTP_204_NO_CONTENT, dependencies=[Depends(get_current_user)])
+async def delete_generation(generation_id: str, generation_service: GenerationService = Depends(get_generation_service)):
+    logger.info(f"delete_generation called for ID: {generation_id}")
+    deleted = await generation_service.delete_generation(generation_id)
+    if not deleted:
+        raise HTTPException(status_code=404, detail="Generation not found")
+    return None