mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-23 04:17:34 +00:00
d1827e6c66
Add user-configurable timezone and 12/24-hour preferences, then wire them through settings, runtime snapshots, scheduler payloads, wait handling, notifications, backups, memory, plugin metadata, and frontend formatters. Keep UTC as the boundary for absolute instants while serializing user-facing dates in the configured or browser-resolved timezone. Preserve scheduler wall-clock inputs in the selected timezone, propagate TZ into desktop/runtime process environments, and restart active desktop sessions when the runtime timezone changes. Cover the risky paths with timezone regression tests for settings normalization, auto and fixed timezone resolution, scheduler round-trips, memory timestamp conversion, and desktop timezone sync.
89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
import asyncio
|
|
from datetime import timedelta
|
|
from helpers.tool import Tool, Response
|
|
from helpers.print_style import PrintStyle
|
|
from helpers.wait import managed_wait
|
|
from helpers.localization import Localization
|
|
|
|
class WaitTool(Tool):
|
|
|
|
async def execute(self, **kwargs) -> Response:
|
|
await self.agent.handle_intervention()
|
|
|
|
seconds = self.args.get("seconds", 0)
|
|
minutes = self.args.get("minutes", 0)
|
|
hours = self.args.get("hours", 0)
|
|
days = self.args.get("days", 0)
|
|
until_timestamp_str = self.args.get("until")
|
|
|
|
is_duration_wait = not bool(until_timestamp_str)
|
|
|
|
now = Localization.get().now()
|
|
target_time = None
|
|
|
|
if until_timestamp_str:
|
|
try:
|
|
target_time = Localization.get().localtime_str_to_utc_dt(until_timestamp_str)
|
|
if not target_time:
|
|
raise ValueError(f"Invalid timestamp format: {until_timestamp_str}")
|
|
except ValueError as e:
|
|
return Response(
|
|
message=str(e),
|
|
break_loop=False,
|
|
)
|
|
else:
|
|
wait_duration = timedelta(
|
|
days=int(days),
|
|
hours=int(hours),
|
|
minutes=int(minutes),
|
|
seconds=int(seconds),
|
|
)
|
|
if wait_duration.total_seconds() <= 0:
|
|
return Response(
|
|
message="Wait duration must be positive.",
|
|
break_loop=False,
|
|
)
|
|
target_time = now + wait_duration
|
|
|
|
if target_time <= now:
|
|
return Response(
|
|
message=f"Target time {Localization.get().serialize_datetime(target_time)} is in the past.",
|
|
break_loop=False,
|
|
)
|
|
|
|
PrintStyle.info(f"Waiting until {Localization.get().serialize_datetime(target_time)}...")
|
|
|
|
target_time = await managed_wait(
|
|
agent=self.agent,
|
|
target_time=target_time,
|
|
is_duration_wait=is_duration_wait,
|
|
log=self.log,
|
|
get_heading_callback=self.get_heading
|
|
)
|
|
|
|
if self.log:
|
|
self.log.update(heading=self.get_heading("Done", done=True))
|
|
|
|
message = self.agent.read_prompt(
|
|
"fw.wait_complete.md",
|
|
target_time=Localization.get().serialize_datetime(target_time)
|
|
)
|
|
|
|
return Response(
|
|
message=message,
|
|
break_loop=False,
|
|
)
|
|
|
|
def get_log_object(self):
|
|
return self.agent.context.log.log(
|
|
type="progress",
|
|
heading=self.get_heading(),
|
|
content="",
|
|
kvps=self.args,
|
|
)
|
|
|
|
def get_heading(self, text: str = "", done: bool = False):
|
|
done_icon = " icon://done_all" if done else ""
|
|
if not text:
|
|
text = f"Waiting..."
|
|
return f"icon://timer Wait: {text}{done_icon}"
|