fix: step controls

novuhq · Dec 13, 2024 · 6a4f22d · 6a4f22d
1 parent 5bd946e
commit 6a4f22d
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/apps/dashboard/src/components/workflow-editor/steps/digest/digest-control-values.tsx b/apps/dashboard/src/components/workflow-editor/steps/digest/digest-control-values.tsx
@@ -4,10 +4,19 @@ import { getComponentByType } from '@/components/workflow-editor/steps/component
 import { useWorkflow } from '@/components/workflow-editor/workflow-provider';
 import { Separator } from '@/components/primitives/separator';
 import { SidebarContent } from '@/components/side-navigation/sidebar';
+import { CustomStepControls } from '../controls/custom-step-controls';
 
 export const DigestControlValues = () => {
-  const { step } = useWorkflow();
-  const { uiSchema } = step?.controls ?? {};
+  const { step, workflow } = useWorkflow();
+  const { uiSchema, dataSchema } = step?.controls ?? {};
+
+  if (dataSchema && workflow) {
+    return (
+      <div>
+        <CustomStepControls dataSchema={dataSchema} origin={workflow.origin} />
+      </div>
+    );
+  }
 
   if (!uiSchema || uiSchema?.group !== UiSchemaGroupEnum.DIGEST) {
     return null;