Record logs into step artifacts (#1339)
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com> Co-authored-by: LawyZheng <lawyzheng1106@gmail.com> Co-authored-by: Nick Fisher <nick.fisher@avinium.com>
This commit is contained in:
83
skyvern/forge/skyvern_log_encoder.py
Normal file
83
skyvern/forge/skyvern_log_encoder.py
Normal file
@@ -0,0 +1,83 @@
|
||||
import json
|
||||
from datetime import datetime
|
||||
from typing import Any
|
||||
|
||||
import structlog
|
||||
from structlog.dev import ConsoleRenderer
|
||||
|
||||
from skyvern.forge.skyvern_json_encoder import SkyvernJSONLogEncoder
|
||||
|
||||
LOG = structlog.get_logger()
|
||||
|
||||
|
||||
class SkyvernLogEncoder:
|
||||
"""Encodes Skyvern logs from JSON format to human-readable string format"""
|
||||
|
||||
def __init__(self) -> None:
|
||||
self.renderer = ConsoleRenderer(
|
||||
pad_event=30,
|
||||
colors=False,
|
||||
)
|
||||
|
||||
@classmethod
|
||||
def _format_value(cls, value: Any) -> str:
|
||||
return SkyvernJSONLogEncoder.dumps(value, sort_keys=True)
|
||||
|
||||
@staticmethod
|
||||
def _parse_json_entry(entry: dict[str, Any]) -> dict[str, Any]:
|
||||
"""Convert a JSON log entry into our standard format."""
|
||||
event = entry.get("message", entry.get("event", ""))
|
||||
|
||||
clean_entry = {
|
||||
"timestamp": entry.get("timestamp", datetime.utcnow().isoformat() + "Z"),
|
||||
"level": entry.get("level", "info").lower(),
|
||||
"event": event,
|
||||
}
|
||||
|
||||
for key, value in entry.items():
|
||||
if key not in ("timestamp", "level", "event", "message"):
|
||||
clean_entry[key] = SkyvernLogEncoder._format_value(value)
|
||||
|
||||
return clean_entry
|
||||
|
||||
@classmethod
|
||||
def encode(cls, log_entries: list[dict[str, Any]]) -> str:
|
||||
"""
|
||||
Encode log entries into formatted string output using structlog's ConsoleRenderer.
|
||||
|
||||
Args:
|
||||
log_entries: List of log entry dictionaries
|
||||
|
||||
Returns:
|
||||
Formatted string with one log entry per line
|
||||
"""
|
||||
encoder = cls()
|
||||
formatted_lines = []
|
||||
|
||||
for entry in log_entries:
|
||||
try:
|
||||
if isinstance(entry, str):
|
||||
try:
|
||||
entry = json.loads(entry)
|
||||
except json.JSONDecodeError:
|
||||
entry = {"event": entry, "level": "info"}
|
||||
|
||||
parsed_entry = cls._parse_json_entry(entry)
|
||||
|
||||
formatted_line = encoder.renderer(None, None, parsed_entry)
|
||||
formatted_lines.append(formatted_line)
|
||||
|
||||
except Exception as e:
|
||||
LOG.error("Failed to format log entry", entry=entry, error=str(e), exc_info=True)
|
||||
# Add error line to output
|
||||
error_timestamp = datetime.utcnow().isoformat() + "Z"
|
||||
error_entry = {
|
||||
"timestamp": error_timestamp,
|
||||
"level": "error",
|
||||
"event": "Failed to format log entry",
|
||||
"entry": str(entry),
|
||||
"error": str(e),
|
||||
}
|
||||
formatted_lines.append(encoder.renderer(None, None, error_entry))
|
||||
|
||||
return "\n".join(formatted_lines)
|
||||
Reference in New Issue
Block a user