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

19
api/core/tools/provider/builtin/webscraper/tools/webscraper.py
View File

@@ -6,23 +6,24 @@ from core.tools.tool.builtin_tool import BuiltinTool
class WebscraperTool(BuiltinTool):
def _invoke(self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
def _invoke(
self,
user_id: str,
tool_parameters: dict[str, Any],
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
"""
invoke tools
invoke tools
"""
try:
url = tool_parameters.get('url', '')
user_agent = tool_parameters.get('user_agent', '')
url = tool_parameters.get("url", "")
user_agent = tool_parameters.get("user_agent", "")
if not url:
return self.create_text_message('Please input url')
return self.create_text_message("Please input url")
# get webpage
result = self.get_url(url, user_agent=user_agent)
if tool_parameters.get('generate_summary'):
if tool_parameters.get("generate_summary"):
# summarize and return
return self.create_text_message(self.summary(user_id=user_id, content=result))
else:

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

@@ -13,12 +13,11 @@ class WebscraperProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
'url': 'https://www.google.com',
'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
"url": "https://www.google.com",
"user_agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ",
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))