InvokeAI/invokeai/app/services/invocation_services.py

# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
from invokeai.backend import InvokeAIGeneratorFactory

from .events import EventServiceBase
from .image_storage import ImageStorageBase
from .invocation_queue import InvocationQueueABC
from .item_storage import ItemStorageABC


class InvocationServices:
    """Services that can be used by invocations"""

    generator_factory: InvokeAIGeneratorFactory
    events: EventServiceBase
    images: ImageStorageBase
    queue: InvocationQueueABC

    # NOTE: we must forward-declare any types that include invocations, since invocations can use services
    graph_execution_manager: ItemStorageABC["GraphExecutionState"]
    processor: "InvocationProcessorABC"

    def __init__(
            self,
            generator_factory: InvokeAIGeneratorFactory,
            events: EventServiceBase,
            images: ImageStorageBase,
            queue: InvocationQueueABC,
            graph_execution_manager: ItemStorageABC["GraphExecutionState"],
            processor: "InvocationProcessorABC",
    ):
        self.generator_factory = generator_factory
        self.events = events
        self.images = images
        self.queue = queue
        self.graph_execution_manager = graph_execution_manager
        self.processor = processor