From cb27571e9fbd7bfc1895c2f3f5e05e4906cb0088 Mon Sep 17 00:00:00 2001
From: Yeuoly <45712896+Yeuoly@users.noreply.github.com>
Date: Tue, 30 Jan 2024 17:00:50 +0800
Subject: [PATCH] fix: missing prompt (#2294)

---
 .../provider/builtin/stablediffusion/tools/stable_diffusion.py  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
index 02fc7448e0..7626048f98 100644
--- a/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
+++ b/api/core/tools/provider/builtin/stablediffusion/tools/stable_diffusion.py
@@ -188,6 +188,8 @@ class StableDiffusionTool(BuiltinTool):
 
         if lora:
             draw_options['prompt'] = f'{lora},{prompt}'
+        else:
+            draw_options['prompt'] = prompt
 
         draw_options['width'] = width
         draw_options['height'] = height