feat: support time format (#4138)

This commit is contained in:
Yeuoly 2024-05-07 13:02:00 +08:00 committed by GitHub
parent 049abd698f
commit e7fe7ec0f6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 3 deletions

View File

@ -17,11 +17,12 @@ class CurrentTimeTool(BuiltinTool):
"""
# 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("%Y-%m-%d %H:%M:%S %Z")}')
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("%Y-%m-%d %H:%M:%S %Z")}')
return self.create_text_message(f'{datetime.now(tz).strftime(fm)}')

View File

@ -12,6 +12,19 @@ description:
pt_BR: A tool for getting the current time.
llm: A tool for getting the current time.
parameters:
- name: format
type: string
required: false
label:
en_US: Format
zh_Hans: 格式
pt_BR: Format
human_description:
en_US: Time format in strftime standard.
zh_Hans: strftime 标准的时间格式。
pt_BR: Time format in strftime standard.
form: form
default: "%Y-%m-%d %H:%M:%S"
- name: timezone
type: select
required: false