diff --git a/invokeai/invocation_api/__init__.py b/invokeai/invocation_api/__init__.py
index 66723f728c..9069f6d4a7 100644
--- a/invokeai/invocation_api/__init__.py
+++ b/invokeai/invocation_api/__init__.py
@@ -20,11 +20,11 @@ from invokeai.app.invocations.fields import (
     DenoiseMaskField,
     FieldDescriptions,
     FieldKind,
+    FluxConditioningField,
     ImageField,
     Input,
     InputField,
     LatentsField,
-    FluxConditioningField,
     MetadataField,
     OutputField,
     UIComponent,
@@ -104,6 +104,7 @@ __all__ = [
     "DenoiseMaskField",
     "FieldDescriptions",
     "FieldKind",
+    "FluxConditioningField",
     "ImageField",
     "Input",
     "InputField",
@@ -143,6 +144,8 @@ __all__ = [
     "DenoiseMaskOutput",
     "FloatCollectionOutput",
     "FloatOutput",
+    "FluxConditioningCollectionOutput",
+    "FluxConditioningOutput",
     "ImageCollectionOutput",
     "ImageOutput",
     "IntegerCollectionOutput",