Update replicate.yaml

replicate.yaml

@@ -78,17 +78,19 @@ nodes:
         type: image
     package: replicate
     script: |
-      (async () => {
+      export async function run({ inputs, state }) {
+
+          const { RepeatNode, NextNode } = DEFINITIONS;
 
           const REPLICATE_TOKEN = env?.variables?.get('REPLICATE_TOKEN');
-          if(!REPLICATE_TOKEN) {
+          if (!REPLICATE_TOKEN) {
               error.fatal('Please, set your API token for Replicate AI');
           }
 
           const { prompt, imagesCount, aspectRatio } = inputs;
 
-          if(!state) {
-              const {data: {id: task}} = await httpClient({
+          if (!state) {
+              const { data: { id: task } } = await httpClient({
                   method: 'post',
                   url: 'https://api.replicate.com/v1/models/black-forest-labs/flux-schnell/predictions',
                   data: {
@@ -104,7 +106,7 @@ nodes:
                       'Prefer': 'wait',
                   }
               });
-              return repeat({state: {task}, delay: 2000});
+              return RepeatNode.from({ state: { task }, delay: 2000 });
           } else {
               const { task } = state;
 
@@ -118,18 +120,18 @@ nodes:
               });
 
               const { status, error } = data;
-              switch(status) {
+              switch (status) {
                   case 'processing':
-                      return repeat({state: {task}, delay: 3000});
+                      return repeat({ state: { task }, delay: 3000 });
                   case 'failed':
                       error.fatal(error);
                   case 'succeeded':
                       const { output: images } = data;
                       const [image1, image2, image3, image4] = images;
-                      return next({outputs: {images, image1, image2, image3, image4}});
+                      return NextNode.from({ outputs: { images, image1, image2, image3, image4 } });
               }
           }
-      })();
+      }
     source: catalog
     title: en=Generate on Flux;ru=Генерация Flux
     version: 5