jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Feat/workflow phase2 (#4687)

Browse source
This commit is contained in:
Yeuoly 2024-05-27 22:01:11 +08:00 • committed by GitHub
commit e852a21634
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
139 changed files with 5997 additions and 779 deletions
Download patch file Download diff file Expand all files Collapse all files

10
api/core/tools/provider/builtin/brave/brave.py
View file

@ -1,5 +1,6 @@
from typing import Any
from core.tools.entities.values import ToolLabelEnum
from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin.brave.tools.brave_search import BraveSearchTool
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
@ -9,7 +10,7 @@ class BraveProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
try:
BraveSearchTool().fork_tool_runtime(
meta={
runtime={
"credentials": credentials,
}
).invoke(
@ -19,4 +20,9 @@ class BraveProvider(BuiltinToolProviderController):
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))
raise ToolProviderCredentialValidationError(str(e))
def _get_tool_labels(self) -> list[ToolLabelEnum]:
return [
ToolLabelEnum.SEARCH,
]