hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-15 13:56:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore(api/core): apply ruff reformatting (#7624)

Browse Source
This commit is contained in:
Bowen Liang
2024-09-10 17:00:20 +08:00
committed by GitHub
parent 178730266d
commit 2cf1187b32
724 changed files with 21180 additions and 21123 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/tools/provider/builtin/stepfun/stepfun.py
View File

@@ -13,13 +13,12 @@ class StepfunProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"prompt": "cute girl, blue eyes, white hair, anime style",
"size": "1024x1024",
"n": 1
"n": 1,
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

60
api/core/tools/provider/builtin/stepfun/tools/image.py
View File

@@ -9,61 +9,67 @@ from core.tools.tool.builtin_tool import BuiltinTool
class StepfunTool(BuiltinTool):
""" Stepfun Image Generation Tool """
def _invoke(self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
"""Stepfun Image Generation Tool"""
def _invoke(
self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
"""
invoke tools
invoke tools
"""
base_url = self.runtime.credentials.get('stepfun_base_url', 'https://api.stepfun.com')
base_url = str(URL(base_url) / 'v1')
base_url = self.runtime.credentials.get("stepfun_base_url", "https://api.stepfun.com")
base_url = str(URL(base_url) / "v1")
client = OpenAI(
api_key=self.runtime.credentials['stepfun_api_key'],
api_key=self.runtime.credentials["stepfun_api_key"],
base_url=base_url,
)
extra_body = {}
model = tool_parameters.get('model', 'step-1x-medium')
model = tool_parameters.get("model", "step-1x-medium")
if not model:
return self.create_text_message('Please input model name')
return self.create_text_message("Please input model name")
# prompt
prompt = tool_parameters.get('prompt', '')
prompt = tool_parameters.get("prompt", "")
if not prompt:
return self.create_text_message('Please input prompt')
return self.create_text_message("Please input prompt")
seed = tool_parameters.get('seed', 0)
seed = tool_parameters.get("seed", 0)
if seed > 0:
extra_body['seed'] = seed
steps = tool_parameters.get('steps', 0)
extra_body["seed"] = seed
steps = tool_parameters.get("steps", 0)
if steps > 0:
extra_body['steps'] = steps
negative_prompt = tool_parameters.get('negative_prompt', '')
extra_body["steps"] = steps
negative_prompt = tool_parameters.get("negative_prompt", "")
if negative_prompt:
extra_body['negative_prompt'] = negative_prompt
extra_body["negative_prompt"] = negative_prompt
# call openapi stepfun model
response = client.images.generate(
prompt=prompt,
model=model,
size=tool_parameters.get('size', '1024x1024'),
n=tool_parameters.get('n', 1),
extra_body= extra_body
size=tool_parameters.get("size", "1024x1024"),
n=tool_parameters.get("n", 1),
extra_body=extra_body,
)
print(response)
result = []
for image in response.data:
result.append(self.create_image_message(image=image.url))
result.append(self.create_json_message({
"url": image.url,
}))
result.append(
self.create_json_message(
{
"url": image.url,
}
)
)
return result
@staticmethod
def _generate_random_id(length=8):
characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
random_id = ''.join(random.choices(characters, k=length))
characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
random_id = "".join(random.choices(characters, k=length))
return random_id