jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chore: apply flake8-comprehensions Ruff rules to improve collection comprehensions (#5652)

Browse source 
Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
Bowen Liang 2024-06-27 11:21:31 +08:00 • committed by GitHub
commit dcb72e0067
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
58 changed files with 123 additions and 136 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/tools/provider/builtin/chart/tools/bar.py
View file

@ -16,12 +16,12 @@ class BarChartTool(BuiltinTool):
data = data.split(';')
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]
axis = tool_parameters.get('x_axis', None) or None
axis = tool_parameters.get('x_axis') or None
if axis:
axis = axis.split(';')
if len(axis) != len(data):

4
api/core/tools/provider/builtin/chart/tools/line.py
View file

@ -17,14 +17,14 @@ class LinearChartTool(BuiltinTool):
return self.create_text_message('Please input data')
data = data.split(';')
axis = tool_parameters.get('x_axis', None) or None
axis = tool_parameters.get('x_axis') or None
if axis:
axis = axis.split(';')
if len(axis) != len(data):
axis = None
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]

4
api/core/tools/provider/builtin/chart/tools/pie.py
View file

@ -16,10 +16,10 @@ class PieChartTool(BuiltinTool):
if not data:
return self.create_text_message('Please input data')
data = data.split(';')
categories = tool_parameters.get('categories', None) or None
categories = tool_parameters.get('categories') or None
# if all data is int, convert to int
if all([i.isdigit() for i in data]):
if all(i.isdigit() for i in data):
data = [int(i) for i in data]
else:
data = [float(i) for i in data]