hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-24 02:03:02 +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

8
api/core/tools/provider/builtin/searxng/searxng.py
View File

@@ -13,12 +13,8 @@ class SearXNGProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
tool_parameters={
"query": "SearXNG",
"limit": 1,
"search_type": "general"
},
user_id="",
tool_parameters={"query": "SearXNG", "limit": 1, "search_type": "general"},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

19
api/core/tools/provider/builtin/searxng/tools/searxng_search.py
View File

@@ -23,18 +23,21 @@ class SearXNGSearchTool(BuiltinTool):
ToolInvokeMessage | list[ToolInvokeMessage]: The result of the tool invocation.
"""
host = self.runtime.credentials.get('searxng_base_url')
host = self.runtime.credentials.get("searxng_base_url")
if not host:
raise Exception('SearXNG api is required')
raise Exception("SearXNG api is required")
response = requests.get(host, params={
"q": tool_parameters.get('query'),
"format": "json",
"categories": tool_parameters.get('search_type', 'general')
})
response = requests.get(
host,
params={
"q": tool_parameters.get("query"),
"format": "json",
"categories": tool_parameters.get("search_type", "general"),
},
)
if response.status_code != 200:
raise Exception(f'Error {response.status_code}: {response.text}')
raise Exception(f"Error {response.status_code}: {response.text}")
res = response.json().get("results", [])
if not res: