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/time/time.py
View File

@@ -9,9 +9,8 @@ class WikiPediaProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
try:
CurrentTimeTool().invoke(
user_id='',
user_id="",
tool_parameters={},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))

25
api/core/tools/provider/builtin/time/tools/current_time.py
View File

@@ -8,21 +8,22 @@ from core.tools.tool.builtin_tool import BuiltinTool
class CurrentTimeTool(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 timezone
tz = tool_parameters.get('timezone', 'UTC')
fm = tool_parameters.get('format') or '%Y-%m-%d %H:%M:%S %Z'
if tz == 'UTC':
return self.create_text_message(f'{datetime.now(timezone.utc).strftime(fm)}')
tz = tool_parameters.get("timezone", "UTC")
fm = tool_parameters.get("format") or "%Y-%m-%d %H:%M:%S %Z"
if tz == "UTC":
return self.create_text_message(f"{datetime.now(timezone.utc).strftime(fm)}")
try:
tz = pytz_timezone(tz)
except:
return self.create_text_message(f'Invalid timezone: {tz}')
return self.create_text_message(f'{datetime.now(tz).strftime(fm)}')
return self.create_text_message(f"Invalid timezone: {tz}")
return self.create_text_message(f"{datetime.now(tz).strftime(fm)}")

21
api/core/tools/provider/builtin/time/tools/weekday.py
View File

@@ -7,25 +7,26 @@ from core.tools.tool.builtin_tool import BuiltinTool
class WeekdayTool(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]]:
"""
Calculate the day of the week for a given date
Calculate the day of the week for a given date
"""
year = tool_parameters.get('year')
month = tool_parameters.get('month')
day = tool_parameters.get('day')
year = tool_parameters.get("year")
month = tool_parameters.get("month")
day = tool_parameters.get("day")
date_obj = self.convert_datetime(year, month, day)
if not date_obj:
return self.create_text_message(f'Invalid date: Year {year}, Month {month}, Day {day}.')
return self.create_text_message(f"Invalid date: Year {year}, Month {month}, Day {day}.")
weekday_name = calendar.day_name[date_obj.weekday()]
month_name = calendar.month_name[month]
readable_date = f"{month_name} {date_obj.day}, {date_obj.year}"
return self.create_text_message(f'{readable_date} is {weekday_name}.')
return self.create_text_message(f"{readable_date} is {weekday_name}.")
@staticmethod
def convert_datetime(year, month, day) -> datetime | None: