diff --git a/invokeai/app/services/image_records/image_records_common.py b/invokeai/app/services/image_records/image_records_common.py index d91b0653ba..bb63b64468 100644 --- a/invokeai/app/services/image_records/image_records_common.py +++ b/invokeai/app/services/image_records/image_records_common.py @@ -216,6 +216,7 @@ class ImageCollectionCounts(BaseModel): class ImageNamesResult(BaseModel): """Response containing ordered image names with metadata for optimistic updates.""" + image_names: list[str] = Field(description="Ordered list of image names") starred_count: int = Field(description="Number of starred images (when starred_first=True)") total_count: int = Field(description="Total number of images matching the query") diff --git a/invokeai/app/services/image_records/image_records_sqlite.py b/invokeai/app/services/image_records/image_records_sqlite.py index fe83c772b5..72c18feb98 100644 --- a/invokeai/app/services/image_records/image_records_sqlite.py +++ b/invokeai/app/services/image_records/image_records_sqlite.py @@ -479,8 +479,4 @@ class SqliteImageRecordStorage(ImageRecordStorageBase): result = cast(list[sqlite3.Row], cursor.fetchall()) image_names = [row[0] for row in result] - return ImageNamesResult( - image_names=image_names, - starred_count=starred_count, - total_count=len(image_names) - ) + return ImageNamesResult(image_names=image_names, starred_count=starred_count, total_count=len(image_names))