fix: ci issues(missing duckduckgo-search==6.2.11, ruff lint issue)

langgenius · Sep 19, 2024 · 473deed · 473deed
1 parent 2721cb8
commit 473deed
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py b/api/core/tools/provider/builtin/comfyui/tools/comfyui_stable_diffusion.py
@@ -290,7 +290,7 @@ def text2img(
         draw_options["6"]["inputs"]["text"] = prompt
         draw_options["7"]["inputs"]["text"] = negative_prompt
         # if the model is SD3 or FLUX series, the Latent class should be corresponding to SD3 Latent
-        if model_type in (ModelType.SD3.name, ModelType.FLUX.name):
+        if model_type in {ModelType.SD3.name, ModelType.FLUX.name}:
             draw_options["5"]["class_type"] = "EmptySD3LatentImage"
 
         if lora_list:

diff --git a/api/poetry.lock b/api/poetry.lock