From a091856178131112b179a2e87529732310aa5358 Mon Sep 17 00:00:00 2001
From: aboutphilippe <philippemarketing@gmail.com>
Date: Tue, 1 Oct 2024 22:38:15 +0200
Subject: [PATCH] keyframe fix

---
 lumaai/functions/generate.ts | 38 +++++++++++++++++-------------------
 lumaai/package.json          |  2 +-
 2 files changed, 19 insertions(+), 21 deletions(-)

diff --git a/lumaai/functions/generate.ts b/lumaai/functions/generate.ts
index 83cda9d..46c650c 100644
--- a/lumaai/functions/generate.ts
+++ b/lumaai/functions/generate.ts
@@ -22,31 +22,29 @@ export async function lumaaiGenerate({
       apiKey,
     });
 
+    let keyframes;
+    if (extendGenerationId) {
+      keyframes = {
+        frame0: {
+          type: "generation",
+          id: extendGenerationId,
+        },
+      };
+    } else if (fromImageUrl) {
+      keyframes = {
+        frame0: {
+          type: "image",
+          url: fromImageUrl,
+        },
+      };
+    }
+
     const generation = await client.generations
       .create({
         aspect_ratio: aspectRatio,
         prompt: prompt,
-        ...(extendGenerationId
-          ? {
-              keyframes: {
-                frame0: {
-                  type: "generation",
-                  id: extendGenerationId,
-                },
-              },
-            }
-          : {}),
-        ...(fromImageUrl
-          ? {
-              keyframes: {
-                frame0: {
-                  type: "image",
-                  url: fromImageUrl,
-                },
-              },
-            }
-          : {}),
         loop,
+        ...(keyframes && { keyframes }),
       })
       .catch(async (err) => {
         if (err instanceof LumaAI.APIError) {
diff --git a/lumaai/package.json b/lumaai/package.json
index ce7f257..f42ee35 100644
--- a/lumaai/package.json
+++ b/lumaai/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@restackio/integrations-lumaai",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "main": "dist/service.js",
   "types": "dist/service.d.ts",
   "scripts": {