diff --git a/invokeai/app/api_app.py b/invokeai/app/api_app.py index dffb2ec139..aaef1d78be 100644 --- a/invokeai/app/api_app.py +++ b/invokeai/app/api_app.py @@ -15,7 +15,7 @@ from fastapi_events.middleware import EventHandlerASGIMiddleware from pydantic.schema import schema from .api.dependencies import ApiDependencies -from .api.routers import image_files, image_records, sessions, models +from .api.routers import image_files, image_records, sessions, models, images from .api.sockets import SocketIO from .invocations.baseinvocation import BaseInvocation from .services.config import InvokeAIAppConfig @@ -71,12 +71,13 @@ async def shutdown_event(): app.include_router(sessions.session_router, prefix="/api") -app.include_router(image_files.image_files_router, prefix="/api") - app.include_router(models.models_router, prefix="/api") +app.include_router(image_files.image_files_router, prefix="/api") + app.include_router(image_records.image_records_router, prefix="/api") +app.include_router(images.images_router, prefix="/api") # Build a custom OpenAPI to include all outputs # TODO: can outputs be included on metadata of invocation schemas somehow?