chore: ruff

This commit is contained in:
psychedelicious
2025-08-19 17:51:39 +10:00
committed by Mary Hipp Rogers
parent 4b4ec29a09
commit 39d5879405
4 changed files with 17 additions and 17 deletions

View File

@@ -2,15 +2,21 @@ from typing import Optional
from fastapi import Body, HTTPException, Path, Query
from fastapi.routing import APIRouter
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
from invokeai.app.services.videos_common import VideoDTO, VideoRecordChanges, DeleteVideosResult, StarredVideosResult, UnstarredVideosResult, VideoIdsResult
from invokeai.app.services.shared.pagination import OffsetPaginatedResults
from invokeai.app.services.shared.sqlite.sqlite_common import SQLiteDirection
from invokeai.app.services.videos_common import (
DeleteVideosResult,
StarredVideosResult,
UnstarredVideosResult,
VideoDTO,
VideoIdsResult,
VideoRecordChanges,
)
videos_router = APIRouter(prefix="/v1/videos", tags=["videos"])
@videos_router.patch(
"/i/{video_id}",
operation_id="update_video",
@@ -42,10 +48,8 @@ async def get_video_dto(
async def delete_videos_from_list(
video_ids: list[str] = Body(description="The list of ids of videos to delete", embed=True),
) -> DeleteVideosResult:
raise HTTPException(status_code=501, detail="Not implemented")
@videos_router.post("/star", operation_id="star_videos_in_list", response_model=StarredVideosResult)
async def star_videos_in_list(
@@ -61,11 +65,8 @@ async def unstar_videos_in_list(
raise HTTPException(status_code=501, detail="Not implemented")
@videos_router.delete(
"/uncategorized", operation_id="delete_uncategorized_videos", response_model=DeleteVideosResult
)
async def delete_uncategorized_videos(
) -> DeleteVideosResult:
@videos_router.delete("/uncategorized", operation_id="delete_uncategorized_videos", response_model=DeleteVideosResult)
async def delete_uncategorized_videos() -> DeleteVideosResult:
"""Deletes all videos that are uncategorized"""
raise HTTPException(status_code=501, detail="Not implemented")
@@ -115,4 +116,4 @@ async def get_videos_by_ids(
) -> list[VideoDTO]:
"""Gets video DTOs for the specified video ids. Maintains order of input ids."""
raise HTTPException(status_code=501, detail="Not implemented")
raise HTTPException(status_code=501, detail="Not implemented")

View File

@@ -27,8 +27,8 @@ from invokeai.app.api.routers import (
session_queue,
style_presets,
utilities,
workflows,
videos,
workflows,
)
from invokeai.app.api.sockets import SocketIO
from invokeai.app.services.config.config_default import get_config
@@ -126,7 +126,7 @@ app.include_router(utilities.utilities_router, prefix="/api")
app.include_router(model_manager.model_manager_router, prefix="/api")
app.include_router(download_queue.download_queue_router, prefix="/api")
app.include_router(images.images_router, prefix="/api")
app.include_router(videos.videos_router, prefix="/api")
app.include_router(videos.videos_router, prefix="/api")
app.include_router(boards.boards_router, prefix="/api")
app.include_router(board_images.board_images_router, prefix="/api")
app.include_router(model_relationships.model_relationships_router, prefix="/api")

View File

@@ -307,6 +307,7 @@ class RunwayVideoOutput(BaseInvocationOutput):
runway_task_id=runway_task_id,
)
@invocation_output("video_output")
class VideoOutput(BaseInvocationOutput):
"""Base class for nodes that output a video"""
@@ -326,8 +327,6 @@ class VideoOutput(BaseInvocationOutput):
)
# endregion
# region DenoiseMask

View File

@@ -66,9 +66,9 @@ from invokeai.app.invocations.primitives import (
IntegerOutput,
LatentsCollectionOutput,
LatentsOutput,
RunwayVideoOutput,
StringCollectionOutput,
StringOutput,
RunwayVideoOutput,
VideoOutput,
)
from invokeai.app.invocations.scheduler import SchedulerOutput