mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-22 19:47:15 +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.
86 lines
2.3 KiB
Python
86 lines
2.3 KiB
Python
import initialize
|
|
from helpers import dotenv, extension, runtime
|
|
from helpers.print_style import PrintStyle
|
|
from helpers.server_startup import run_uvicorn_with_retries
|
|
from helpers.ui_server import UiServerRuntime, configure_process_environment
|
|
|
|
|
|
def run():
|
|
configure_process_environment()
|
|
PrintStyle().print("Initializing Python framework...")
|
|
PrintStyle().print("Checking for data migration...")
|
|
run_migration_checks()
|
|
|
|
PrintStyle().print("Preparing web server runtime...")
|
|
server_runtime, host, port = prepare_web_runtime()
|
|
|
|
PrintStyle().print("Initializing Agent Zero components...")
|
|
init_a0()
|
|
|
|
PrintStyle().print("Starting UI/API server...")
|
|
start_web_server(server_runtime, host, port)
|
|
|
|
|
|
def run_migration_checks() -> None:
|
|
initialize.initialize_migration()
|
|
|
|
|
|
def prepare_web_runtime() -> tuple[UiServerRuntime, str, int]:
|
|
host = (
|
|
runtime.get_arg("host") or dotenv.get_dotenv_value("WEB_UI_HOST") or "localhost"
|
|
)
|
|
port = runtime.get_web_ui_port()
|
|
server_runtime = UiServerRuntime.create()
|
|
server_runtime.register_http_routes()
|
|
server_runtime.register_transport_handlers()
|
|
|
|
return server_runtime, host, port
|
|
|
|
|
|
def start_web_server(server_runtime: UiServerRuntime, host: str, port: int) -> None:
|
|
run_uvicorn_with_retries(
|
|
host=host,
|
|
port=port,
|
|
build_asgi_app=server_runtime.build_asgi_app,
|
|
flush_callback=create_flush_callback(),
|
|
access_log=server_runtime.access_log_enabled(),
|
|
ws="wsproto",
|
|
)
|
|
|
|
|
|
def create_flush_callback():
|
|
def flush_and_shutdown_callback() -> None:
|
|
"""
|
|
TODO(dev): add cleanup + flush-to-disk logic here.
|
|
"""
|
|
return
|
|
|
|
flush_ran = False
|
|
|
|
def _run_flush(reason: str) -> None:
|
|
nonlocal flush_ran
|
|
if flush_ran:
|
|
return
|
|
flush_ran = True
|
|
try:
|
|
flush_and_shutdown_callback()
|
|
except Exception as e:
|
|
PrintStyle.warning(f"Shutdown flush failed ({reason}): {e}")
|
|
|
|
return _run_flush
|
|
|
|
|
|
@extension.extensible
|
|
def init_a0():
|
|
init_chats = initialize.initialize_chats()
|
|
init_chats.result_sync()
|
|
|
|
initialize.initialize_mcp()
|
|
initialize.initialize_job_loop()
|
|
initialize.initialize_preload()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
runtime.initialize()
|
|
dotenv.load_dotenv()
|
|
run()
|