somew1nd
67248c887e
Add execution incidents, action bursts and shared Flash history. Expose verification and Explorer settings with streamed check results. Support graceful cancellation, recording recovery and bundled UI resources. Add CI quality checks and clean up comments and documentation.
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
"""Best-effort startup progress events for UI clients.
|
|
|
|
Startup happens before the normal DataEngine event bridge exists. This small
|
|
sender lets the worker report those otherwise invisible phases over the same
|
|
local IPC protocol without making task startup depend on the UI server.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import socket
|
|
import time
|
|
from typing import Any
|
|
|
|
from artemis.config import read_ipc_port
|
|
|
|
|
|
def publish_startup_progress(
|
|
stage: str, message: str, session_id: str | None = None, **details: Any
|
|
) -> None:
|
|
"""Publish one startup milestone, ignoring unavailable UI/IPC endpoints."""
|
|
resolved_session_id = (
|
|
session_id or os.getenv("ARTEMIS_SESSION_ID") or os.getenv("ARTEMIS_CLOUD_SESSION_ID")
|
|
)
|
|
ipc_port = read_ipc_port()
|
|
if not resolved_session_id or not ipc_port:
|
|
return
|
|
|
|
event_data = {
|
|
"session_id": str(resolved_session_id),
|
|
"stage": stage,
|
|
"message": message,
|
|
"timestamp": time.time(),
|
|
**details,
|
|
}
|
|
encoded = (
|
|
json.dumps(
|
|
{"event_type": "startup_progress", "data": event_data},
|
|
default=str,
|
|
)
|
|
+ "\n"
|
|
).encode("utf-8")
|
|
|
|
try:
|
|
with socket.create_connection(("127.0.0.1", ipc_port), timeout=0.25) as sock:
|
|
sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
|
|
sock.sendall(encoded)
|
|
except OSError:
|
|
# Progress reporting is observability only and must never delay or fail
|
|
# the automation task itself.
|
|
return
|