Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactoring handlers to do null check

Browse Source
This commit is contained in:
Stefan Tobler
2024-02-17 23:53:38 -05:00
committed by psychedelicious
parent ef12631450
commit 2e53aa48c9
3 changed files with 27 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/app/api/routers/images.py
View File

@ -1,6 +1,6 @@
import io
import traceback
from typing import Optional, cast
from typing import Optional
from fastapi import BackgroundTasks, Body, HTTPException, Path, Query, Request, Response, UploadFile
from fastapi.responses import FileResponse
@ -396,10 +396,9 @@ async def download_images_from_list(
raise HTTPException(status_code=400, detail="No images or board id specified.")
bulk_download_item_id: str = ApiDependencies.invoker.services.bulk_download.generate_item_id(board_id)
# Type narrowing handled above ^, we know that image_names is not None, trying to keep null checks at the boundaries
background_tasks.add_task(
ApiDependencies.invoker.services.bulk_download.handler,
cast(list[str], image_names),
image_names,
board_id,
bulk_download_item_id,
)