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

78
api/core/tools/provider/builtin/wolframalpha/tools/wolframalpha.py
View File

@@ -8,29 +8,24 @@ from core.tools.tool.builtin_tool import BuiltinTool
class WolframAlphaTool(BuiltinTool):
_base_url = 'https://api.wolframalpha.com/v2/query'
_base_url = "https://api.wolframalpha.com/v2/query"
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
"""
query = tool_parameters.get('query', '')
query = tool_parameters.get("query", "")
if not query:
return self.create_text_message('Please input query')
appid = self.runtime.credentials.get('appid', '')
return self.create_text_message("Please input query")
appid = self.runtime.credentials.get("appid", "")
if not appid:
raise ToolProviderCredentialValidationError('Please input appid')
params = {
'appid': appid,
'input': query,
'includepodid': 'Result',
'format': 'plaintext',
'output': 'json'
}
raise ToolProviderCredentialValidationError("Please input appid")
params = {"appid": appid, "input": query, "includepodid": "Result", "format": "plaintext", "output": "json"}
finished = False
result = None
@@ -45,34 +40,33 @@ class WolframAlphaTool(BuiltinTool):
response_data = response.json()
except Exception as e:
raise ToolInvokeError(str(e))
if 'success' not in response_data['queryresult'] or response_data['queryresult']['success'] != True:
query_result = response_data.get('queryresult', {})
if query_result.get('error'):
if 'msg' in query_result['error']:
if query_result['error']['msg'] == 'Invalid appid':
raise ToolProviderCredentialValidationError('Invalid appid')
raise ToolInvokeError('Failed to invoke tool')
if 'didyoumeans' in response_data['queryresult']:
# get the most likely interpretation
query = ''
max_score = 0
for didyoumean in response_data['queryresult']['didyoumeans']:
if float(didyoumean['score']) > max_score:
query = didyoumean['val']
max_score = float(didyoumean['score'])
params['input'] = query
if "success" not in response_data["queryresult"] or response_data["queryresult"]["success"] != True:
query_result = response_data.get("queryresult", {})
if query_result.get("error"):
if "msg" in query_result["error"]:
if query_result["error"]["msg"] == "Invalid appid":
raise ToolProviderCredentialValidationError("Invalid appid")
raise ToolInvokeError("Failed to invoke tool")
if "didyoumeans" in response_data["queryresult"]:
# get the most likely interpretation
query = ""
max_score = 0
for didyoumean in response_data["queryresult"]["didyoumeans"]:
if float(didyoumean["score"]) > max_score:
query = didyoumean["val"]
max_score = float(didyoumean["score"])
params["input"] = query
else:
finished = True
if 'souces' in response_data['queryresult']:
return self.create_link_message(response_data['queryresult']['sources']['url'])
elif 'pods' in response_data['queryresult']:
result = response_data['queryresult']['pods'][0]['subpods'][0]['plaintext']
if "souces" in response_data["queryresult"]:
return self.create_link_message(response_data["queryresult"]["sources"]["url"])
elif "pods" in response_data["queryresult"]:
result = response_data["queryresult"]["pods"][0]["subpods"][0]["plaintext"]
if not finished or not result:
return self.create_text_message('No result found')
return self.create_text_message("No result found")
return self.create_text_message(result)

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

@@ -13,11 +13,10 @@ class GoogleProvider(BuiltinToolProviderController):
"credentials": credentials,
}
).invoke(
user_id='',
user_id="",
tool_parameters={
"query": "1+2+....+111",
},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))