feat: add flux dev of siliconflow image-gen tool (#8450)

This commit is contained in:
非法操作 2024-09-15 17:14:12 +08:00 committed by GitHub
parent aa5b2db10a
commit 3d083b758f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 27 additions and 5 deletions

View File

@ -5,7 +5,10 @@ import requests
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool.builtin_tool import BuiltinTool
FLUX_URL = "https://api.siliconflow.cn/v1/black-forest-labs/FLUX.1-schnell/text-to-image"
FLUX_URL = {
"schnell": "https://api.siliconflow.cn/v1/black-forest-labs/FLUX.1-schnell/text-to-image",
"dev": "https://api.siliconflow.cn/v1/image/generations",
}
class FluxTool(BuiltinTool):
@ -24,8 +27,12 @@ class FluxTool(BuiltinTool):
"seed": tool_parameters.get("seed"),
"num_inference_steps": tool_parameters.get("num_inference_steps", 20),
}
model = tool_parameters.get("model", "schnell")
url = FLUX_URL.get(model)
if model == "dev":
payload["model"] = "black-forest-labs/FLUX.1-dev"
response = requests.post(FLUX_URL, json=payload, headers=headers)
response = requests.post(url, json=payload, headers=headers)
if response.status_code != 200:
return self.create_text_message(f"Got Error Response:{response.text}")

View File

@ -6,8 +6,8 @@ identity:
icon: icon.svg
description:
human:
en_US: Generate image via SiliconFlow's flux schnell.
llm: This tool is used to generate image from prompt via SiliconFlow's flux schnell model.
en_US: Generate image via SiliconFlow's flux model.
llm: This tool is used to generate image from prompt via SiliconFlow's flux model.
parameters:
- name: prompt
type: string
@ -17,9 +17,24 @@ parameters:
zh_Hans: 提示词
human_description:
en_US: The text prompt used to generate the image.
zh_Hans: 用于生成图片的文字提示词
zh_Hans: 建议用英文的生成图片提示词以获得更好的生成效果。
llm_description: this prompt text will be used to generate image.
form: llm
- name: model
type: select
required: true
options:
- value: schnell
label:
en_US: Flux.1-schnell
- value: dev
label:
en_US: Flux.1-dev
default: schnell
label:
en_US: Choose Image Model
zh_Hans: 选择生成图片的模型
form: form
- name: image_size
type: select
required: true