admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 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/tavily/tavily.py
View File

@@ -13,7 +13,7 @@ class TavilyProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"query": "Sachin Tendulkar",
"search_depth": "basic",
@@ -22,9 +22,8 @@ class TavilyProvider(BuiltinToolProviderController):
"include_raw_content": False,
"max_results": 5,
"include_domains": "",
"exclude_domains": ""
"exclude_domains": "",
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

28
api/core/tools/provider/builtin/tavily/tools/tavily_search.py
View File

@@ -36,15 +36,23 @@ class TavilySearch:
"""
params["api_key"] = self.api_key
if 'exclude_domains' in params and isinstance(params['exclude_domains'], str) and params['exclude_domains'] != 'None':
params['exclude_domains'] = params['exclude_domains'].split()
if (
"exclude_domains" in params
and isinstance(params["exclude_domains"], str)
and params["exclude_domains"] != "None"
):
params["exclude_domains"] = params["exclude_domains"].split()
else:
params['exclude_domains'] = []
if 'include_domains' in params and isinstance(params['include_domains'], str) and params['include_domains'] != 'None':
params['include_domains'] = params['include_domains'].split()
params["exclude_domains"] = []
if (
"include_domains" in params
and isinstance(params["include_domains"], str)
and params["include_domains"] != "None"
):
params["include_domains"] = params["include_domains"].split()
else:
params['include_domains'] = []
params["include_domains"] = []
response = requests.post(f"{TAVILY_API_URL}/search", json=params)
response.raise_for_status()
return response.json()
@@ -91,9 +99,7 @@ class TavilySearchTool(BuiltinTool):
A tool for searching Tavily using a given query.
"""
def _invoke(
self, user_id: str, tool_parameters: dict[str, Any]
) -> ToolInvokeMessage | list[ToolInvokeMessage]:
def _invoke(self, user_id: str, tool_parameters: dict[str, Any]) -> ToolInvokeMessage | list[ToolInvokeMessage]:
"""
Invokes the Tavily search tool with the given user ID and tool parameters.
@@ -115,4 +121,4 @@ class TavilySearchTool(BuiltinTool):
if not results:
return self.create_text_message(f"No results found for '{query}' in Tavily")
else:
return self.create_text_message(text=results)
return self.create_text_message(text=results)