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

4
api/core/tools/provider/builtin/onebot/onebot.py
View File

@@ -5,8 +5,6 @@ from core.tools.provider.builtin_tool_provider import BuiltinToolProviderControl
class OneBotProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
if not credentials.get("ob11_http_url"):
raise ToolProviderCredentialValidationError('OneBot HTTP URL is required.')
raise ToolProviderCredentialValidationError("OneBot HTTP URL is required.")

53
api/core/tools/provider/builtin/onebot/tools/send_group_msg.py
View File

@@ -11,54 +11,29 @@ class SendGroupMsg(BuiltinTool):
"""OneBot v11 Tool: Send Group Message"""
def _invoke(
self,
user_id: str,
tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
self, user_id: str, tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
# Get parameters
send_group_id = tool_parameters.get('group_id', '')
message = tool_parameters.get('message', '')
send_group_id = tool_parameters.get("group_id", "")
message = tool_parameters.get("message", "")
if not message:
return self.create_json_message(
{
'error': 'Message is empty.'
}
)
auto_escape = tool_parameters.get('auto_escape', False)
return self.create_json_message({"error": "Message is empty."})
auto_escape = tool_parameters.get("auto_escape", False)
try:
url = URL(self.runtime.credentials['ob11_http_url']) / 'send_group_msg'
url = URL(self.runtime.credentials["ob11_http_url"]) / "send_group_msg"
resp = requests.post(
url,
json={
'group_id': send_group_id,
'message': message,
'auto_escape': auto_escape
},
headers={
'Authorization': 'Bearer ' + self.runtime.credentials['access_token']
}
json={"group_id": send_group_id, "message": message, "auto_escape": auto_escape},
headers={"Authorization": "Bearer " + self.runtime.credentials["access_token"]},
)
if resp.status_code != 200:
return self.create_json_message(
{
'error': f'Failed to send group message: {resp.text}'
}
)
return self.create_json_message({"error": f"Failed to send group message: {resp.text}"})
return self.create_json_message(
{
'response': resp.json()
}
)
return self.create_json_message({"response": resp.json()})
except Exception as e:
return self.create_json_message(
{
'error': f'Failed to send group message: {e}'
}
)
return self.create_json_message({"error": f"Failed to send group message: {e}"})

55
api/core/tools/provider/builtin/onebot/tools/send_private_msg.py
View File

@@ -11,54 +11,29 @@ class SendPrivateMsg(BuiltinTool):
"""OneBot v11 Tool: Send Private Message"""
def _invoke(
self,
user_id: str,
tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
self, user_id: str, tool_parameters: dict[str, Any]
) -> Union[ToolInvokeMessage, list[ToolInvokeMessage]]:
# Get parameters
send_user_id = tool_parameters.get('user_id', '')
message = tool_parameters.get('message', '')
send_user_id = tool_parameters.get("user_id", "")
message = tool_parameters.get("message", "")
if not message:
return self.create_json_message(
{
'error': 'Message is empty.'
}
)
auto_escape = tool_parameters.get('auto_escape', False)
return self.create_json_message({"error": "Message is empty."})
auto_escape = tool_parameters.get("auto_escape", False)
try:
url = URL(self.runtime.credentials['ob11_http_url']) / 'send_private_msg'
url = URL(self.runtime.credentials["ob11_http_url"]) / "send_private_msg"
resp = requests.post(
url,
json={
'user_id': send_user_id,
'message': message,
'auto_escape': auto_escape
},
headers={
'Authorization': 'Bearer ' + self.runtime.credentials['access_token']
}
json={"user_id": send_user_id, "message": message, "auto_escape": auto_escape},
headers={"Authorization": "Bearer " + self.runtime.credentials["access_token"]},
)
if resp.status_code != 200:
return self.create_json_message(
{
'error': f'Failed to send private message: {resp.text}'
}
)
return self.create_json_message(
{
'response': resp.json()
}
)
return self.create_json_message({"error": f"Failed to send private message: {resp.text}"})
return self.create_json_message({"response": resp.json()})
except Exception as e:
return self.create_json_message(
{
'error': f'Failed to send private message: {e}'
}
)
return self.create_json_message({"error": f"Failed to send private message: {e}"})