diff --git a/invokeai/invocation_api/__init__.py b/invokeai/invocation_api/__init__.py
index d5f34f5ffd..66723f728c 100644
--- a/invokeai/invocation_api/__init__.py
+++ b/invokeai/invocation_api/__init__.py
@@ -24,6 +24,7 @@ from invokeai.app.invocations.fields import (
     Input,
     InputField,
     LatentsField,
+    FluxConditioningField,
     MetadataField,
     OutputField,
     UIComponent,
@@ -56,6 +57,8 @@ from invokeai.app.invocations.primitives import (
     DenoiseMaskOutput,
     FloatCollectionOutput,
     FloatOutput,
+    FluxConditioningCollectionOutput,
+    FluxConditioningOutput,
     ImageCollectionOutput,
     ImageOutput,
     IntegerCollectionOutput,