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

6
api/core/tools/provider/builtin/regex/regex.py
View File

@@ -9,10 +9,10 @@ class RegexProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
try:
RegexExpressionTool().invoke(
user_id='',
user_id="",
tool_parameters={
'content': '1+(2+3)*4',
'expression': r'(\d+)',
"content": "1+(2+3)*4",
"expression": r"(\d+)",
},
)
except Exception as e:

21
api/core/tools/provider/builtin/regex/tools/regex_extract.py
View File

@@ -6,22 +6,23 @@ from core.tools.tool.builtin_tool import BuiltinTool
class RegexExpressionTool(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
"""
# get expression
content = tool_parameters.get('content', '').strip()
content = tool_parameters.get("content", "").strip()
if not content:
return self.create_text_message('Invalid content')
expression = tool_parameters.get('expression', '').strip()
return self.create_text_message("Invalid content")
expression = tool_parameters.get("expression", "").strip()
if not expression:
return self.create_text_message('Invalid expression')
return self.create_text_message("Invalid expression")
try:
result = re.findall(expression, content)
return self.create_text_message(str(result))
except Exception as e:
return self.create_text_message(f'Failed to extract result, error: {str(e)}')
return self.create_text_message(f"Failed to extract result, error: {str(e)}")