feat: support define tags in tool yaml (#4763)

This commit is contained in:
Yeuoly
2024-05-29 15:19:14 +08:00
committed by GitHub
parent b189faca52
commit 74f38eacda
82 changed files with 187 additions and 255 deletions

View File

@@ -1,6 +1,5 @@
from typing import Any
from core.tools.entities.values import ToolLabelEnum
from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin.stablediffusion.tools.stable_diffusion import StableDiffusionTool
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@@ -16,8 +15,4 @@ class StableDiffusionProvider(BuiltinToolProviderController):
).validate_models()
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))
def _get_tool_labels(self) -> list[ToolLabelEnum]:
return [
ToolLabelEnum.IMAGE
]

View File

@@ -10,6 +10,8 @@ identity:
zh_Hans: Stable Diffusion 是一个可以在本地部署的图片生成的工具。
pt_BR: Stable Diffusion is a tool for generating images which can be deployed locally.
icon: icon.png
tags:
- image
credentials_for_provider:
base_url:
type: secret-input