Skip to content

Commit 970bb16

Browse files
feat(api): add get_uncategorized_image_counts endpoint
1 parent 338c74d commit 970bb16

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

invokeai/app/api/routers/boards.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from pydantic import BaseModel, Field
66

77
from invokeai.app.api.dependencies import ApiDependencies
8-
from invokeai.app.services.board_records.board_records_common import BoardChanges
8+
from invokeai.app.services.board_records.board_records_common import BoardChanges, UncategorizedImageCounts
99
from invokeai.app.services.boards.boards_common import BoardDTO
1010
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
1111

@@ -146,3 +146,14 @@ async def list_all_board_image_names(
146146
board_id,
147147
)
148148
return image_names
149+
150+
151+
@boards_router.get(
152+
"/uncategorized/counts",
153+
operation_id="get_uncategorized_image_counts",
154+
response_model=UncategorizedImageCounts,
155+
)
156+
async def get_uncategorized_image_counts() -> UncategorizedImageCounts:
157+
"""Gets count of images and assets for uncategorized images (images with no board assocation)"""
158+
159+
return ApiDependencies.invoker.services.board_records.get_uncategorized_image_counts()

0 commit comments

Comments
 (0)