estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 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

3
api/core/tools/provider/builtin/judge0ce/judge0ce.py
View File

@@ -13,7 +13,7 @@ class Judge0CEProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"source_code": "print('hello world')",
"language_id": 71,
@@ -21,4 +21,3 @@ class Judge0CEProvider(BuiltinToolProviderController):
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

44
api/core/tools/provider/builtin/judge0ce/tools/executeCode.py
View File

@@ -9,11 +9,13 @@ from core.tools.tool.builtin_tool import BuiltinTool
class ExecuteCodeTool(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
"""
api_key = self.runtime.credentials['X-RapidAPI-Key']
api_key = self.runtime.credentials["X-RapidAPI-Key"]
url = "https://judge0-ce.p.rapidapi.com/submissions"
@@ -22,15 +24,15 @@ class ExecuteCodeTool(BuiltinTool):
headers = {
"Content-Type": "application/json",
"X-RapidAPI-Key": api_key,
"X-RapidAPI-Host": "judge0-ce.p.rapidapi.com"
"X-RapidAPI-Host": "judge0-ce.p.rapidapi.com",
}
payload = {
"language_id": tool_parameters['language_id'],
"source_code": tool_parameters['source_code'],
"stdin": tool_parameters.get('stdin', ''),
"expected_output": tool_parameters.get('expected_output', ''),
"additional_files": tool_parameters.get('additional_files', ''),
"language_id": tool_parameters["language_id"],
"source_code": tool_parameters["source_code"],
"stdin": tool_parameters.get("stdin", ""),
"expected_output": tool_parameters.get("expected_output", ""),
"additional_files": tool_parameters.get("additional_files", ""),
}
response = post(url, data=json.dumps(payload), headers=headers, params=querystring)
@@ -38,22 +40,22 @@ class ExecuteCodeTool(BuiltinTool):
if response.status_code != 201:
raise Exception(response.text)
token = response.json()['token']
token = response.json()["token"]
url = f"https://judge0-ce.p.rapidapi.com/submissions/{token}"
headers = {
"X-RapidAPI-Key": api_key
}
headers = {"X-RapidAPI-Key": api_key}
response = requests.get(url, headers=headers)
if response.status_code == 200:
result = response.json()
return self.create_text_message(text=f"stdout: {result.get('stdout', '')}\n"
f"stderr: {result.get('stderr', '')}\n"
f"compile_output: {result.get('compile_output', '')}\n"
f"message: {result.get('message', '')}\n"
f"status: {result['status']['description']}\n"
f"time: {result.get('time', '')} seconds\n"
f"memory: {result.get('memory', '')} bytes")
return self.create_text_message(
text=f"stdout: {result.get('stdout', '')}\n"
f"stderr: {result.get('stderr', '')}\n"
f"compile_output: {result.get('compile_output', '')}\n"
f"message: {result.get('message', '')}\n"
f"status: {result['status']['description']}\n"
f"time: {result.get('time', '')} seconds\n"
f"memory: {result.get('memory', '')} bytes"
)
else:
return self.create_text_message(text=f"Error retrieving submission details: {response.text}")
return self.create_text_message(text=f"Error retrieving submission details: {response.text}")