feat: handle JSON.parse error (#220)

Co-authored-by: Shigma <[email protected]>
koishijs · Feb 3, 2024 · 86d422f · 86d422f
1 parent a2fab9a
commit 86d422f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -393,8 +393,17 @@ export function apply(ctx: Context, config: Config) {
           })
 
           if (config.type === 'sd-webui') {
-            finalPrompt = (JSON.parse((res.data as StableDiffusionWebUI.Response).info)).prompt
-            return forceDataPrefix((res.data as StableDiffusionWebUI.Response).images[0])
+            const data = res.data as StableDiffusionWebUI.Response
+            if (data?.info?.prompt) {
+              finalPrompt = data.info.prompt
+            } else {
+              try {
+                finalPrompt = (JSON.parse(data.info)).prompt
+              } catch (err) {
+                logger.warn(err)
+              }
+            }
+            return forceDataPrefix(data.images[0])
           }
           if (config.type === 'stable-horde') {
             const uuid = res.data.id