horde_sdk.worker.generations¶

class ImageGenerationInitKwargs(TypedDict, total=False):
    """Optional keyword arguments accepted by `ImageSingleGeneration` from template helpers."""

    generation_id: ID_TYPES | None
    dispatch_result_ids: Sequence[ID_TYPES] | None
    result_ids: list[ID_TYPES] | None
    requires_submit: bool
    safety_rules: SafetyRules
    black_box_mode: bool
    state_error_limits: Mapping[GENERATION_PROGRESS, int] | None
    strict_transition_mode: bool
    extra_logging: bool

class AlchemyGenerationInitKwargs(TypedDict, total=False):
    """Optional keyword arguments accepted by `AlchemySingleGeneration` factory flows."""

    generation_id: str | None
    dispatch_result_ids: Sequence[ID_TYPES] | None
    result_ids: list[ID_TYPES] | None
    requires_generation: bool
    requires_post_processing: bool
    requires_safety_check: bool
    requires_submit: bool
    safety_rules: SafetyRules
    black_box_mode: bool
    state_error_limits: Mapping[GENERATION_PROGRESS, int] | None
    strict_transition_mode: bool
    extra_logging: bool

class TextGenerationInitKwargs(TypedDict, total=False):
    """Optional keyword arguments accepted by `TextSingleGeneration` helpers."""

    generation_id: ID_TYPES | None
    dispatch_result_ids: Sequence[ID_TYPES] | None
    result_ids: list[ID_TYPES] | None
    requires_post_processing: bool
    requires_safety_check: bool
    requires_submit: bool
    safety_rules: SafetyRules
    black_box_mode: bool
    state_error_limits: Mapping[GENERATION_PROGRESS, int] | None
    strict_transition_mode: bool
    extra_logging: bool