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

7
api/core/tools/provider/builtin/serper/serper.py
View File

@@ -13,11 +13,8 @@ class SerperProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
tool_parameters={
"query": "test",
"result_type": "link"
},
user_id="",
tool_parameters={"query": "test", "result_type": "link"},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

30
api/core/tools/provider/builtin/serper/tools/serper_search.py
View File

@@ -9,7 +9,6 @@ SERPER_API_URL = "https://google.serper.dev/search"
class SerperSearchTool(BuiltinTool):
def _parse_response(self, response: dict) -> dict:
result = {}
if "knowledgeGraph" in response:
@@ -17,28 +16,19 @@ class SerperSearchTool(BuiltinTool):
result["description"] = response["knowledgeGraph"].get("description", "")
if "organic" in response:
result["organic"] = [
{
"title": item.get("title", ""),
"link": item.get("link", ""),
"snippet": item.get("snippet", "")
}
{"title": item.get("title", ""), "link": item.get("link", ""), "snippet": item.get("snippet", "")}
for item in response["organic"]
]
return result
def _invoke(self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
params = {
"q": tool_parameters['query'],
"gl": "us",
"hl": "en"
}
headers = {
'X-API-KEY': self.runtime.credentials['serperapi_api_key'],
'Content-Type': 'application/json'
}
response = requests.get(url=SERPER_API_URL, params=params,headers=headers)
def _invoke(
self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
params = {"q": tool_parameters["query"], "gl": "us", "hl": "en"}
headers = {"X-API-KEY": self.runtime.credentials["serperapi_api_key"], "Content-Type": "application/json"}
response = requests.get(url=SERPER_API_URL, params=params, headers=headers)
response.raise_for_status()
valuable_res = self._parse_response(response.json())
return self.create_json_message(valuable_res)