mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
feat(nodes): create invocation_api.py
This is the public API for invocations. Everything a custom node might need should be re-exported from this file.
This commit is contained in:
109
invokeai/invocation_api/__init__.py
Normal file
109
invokeai/invocation_api/__init__.py
Normal file
@ -0,0 +1,109 @@
|
||||
"""
|
||||
This file re-exports all the public API for invocations. This is the only file that should be imported by custom nodes.
|
||||
|
||||
TODO(psyche): Do we want to dogfood this?
|
||||
"""
|
||||
|
||||
from invokeai.app.invocations.baseinvocation import (
|
||||
BaseInvocation,
|
||||
BaseInvocationOutput,
|
||||
invocation,
|
||||
invocation_output,
|
||||
)
|
||||
from invokeai.app.invocations.fields import (
|
||||
BoardField,
|
||||
ColorField,
|
||||
ConditioningField,
|
||||
DenoiseMaskField,
|
||||
FieldDescriptions,
|
||||
FieldKind,
|
||||
ImageField,
|
||||
Input,
|
||||
InputField,
|
||||
LatentsField,
|
||||
MetadataField,
|
||||
OutputField,
|
||||
UIComponent,
|
||||
UIType,
|
||||
WithMetadata,
|
||||
WithWorkflow,
|
||||
)
|
||||
from invokeai.app.invocations.primitives import (
|
||||
BooleanCollectionOutput,
|
||||
BooleanOutput,
|
||||
ColorCollectionOutput,
|
||||
ColorOutput,
|
||||
ConditioningCollectionOutput,
|
||||
ConditioningOutput,
|
||||
DenoiseMaskOutput,
|
||||
FloatCollectionOutput,
|
||||
FloatOutput,
|
||||
ImageCollectionOutput,
|
||||
ImageOutput,
|
||||
IntegerCollectionOutput,
|
||||
IntegerOutput,
|
||||
LatentsCollectionOutput,
|
||||
LatentsOutput,
|
||||
StringCollectionOutput,
|
||||
StringOutput,
|
||||
)
|
||||
from invokeai.app.services.image_records.image_records_common import ImageCategory
|
||||
from invokeai.app.services.shared.invocation_context import InvocationContext
|
||||
from invokeai.backend.stable_diffusion.diffusion.conditioning_data import (
|
||||
BasicConditioningInfo,
|
||||
ConditioningFieldData,
|
||||
ExtraConditioningInfo,
|
||||
SDXLConditioningInfo,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
# invokeai.app.invocations.baseinvocation
|
||||
"BaseInvocation",
|
||||
"BaseInvocationOutput",
|
||||
"invocation",
|
||||
"invocation_output",
|
||||
# invokeai.app.services.shared.invocation_context
|
||||
"InvocationContext",
|
||||
# invokeai.app.invocations.fields
|
||||
"BoardField",
|
||||
"ColorField",
|
||||
"ConditioningField",
|
||||
"DenoiseMaskField",
|
||||
"FieldDescriptions",
|
||||
"FieldKind",
|
||||
"ImageField",
|
||||
"Input",
|
||||
"InputField",
|
||||
"LatentsField",
|
||||
"MetadataField",
|
||||
"OutputField",
|
||||
"UIComponent",
|
||||
"UIType",
|
||||
"WithMetadata",
|
||||
"WithWorkflow",
|
||||
# invokeai.app.invocations.primitives
|
||||
"BooleanCollectionOutput",
|
||||
"BooleanOutput",
|
||||
"ColorCollectionOutput",
|
||||
"ColorOutput",
|
||||
"ConditioningCollectionOutput",
|
||||
"ConditioningOutput",
|
||||
"DenoiseMaskOutput",
|
||||
"FloatCollectionOutput",
|
||||
"FloatOutput",
|
||||
"ImageCollectionOutput",
|
||||
"ImageOutput",
|
||||
"IntegerCollectionOutput",
|
||||
"IntegerOutput",
|
||||
"LatentsCollectionOutput",
|
||||
"LatentsOutput",
|
||||
"StringCollectionOutput",
|
||||
"StringOutput",
|
||||
# invokeai.app.services.image_records.image_records_common
|
||||
"ImageCategory",
|
||||
# invokeai.backend.stable_diffusion.diffusion.conditioning_data
|
||||
"BasicConditioningInfo",
|
||||
"ConditioningFieldData",
|
||||
"ExtraConditioningInfo",
|
||||
"SDXLConditioningInfo",
|
||||
]
|
Reference in New Issue
Block a user