Function _write_tool_result

workers/file_processing/structure_tool_task.py:836–887

Write tool result and tool_metadata to METADATA.json. Matches BaseTool._update_exec_metadata(): - tool_metadata: list of dicts with tool_name, output_type, elapsed_time (destination connector reads output_type from here) - total_elapsed_time: cumulative elapsed time

(
    fs: Any, execution_data_dir: str, _data: dict, elapsed_time: float = 0.0
)

def _write_tool_result(
    fs: Any, execution_data_dir: str, _data: dict, elapsed_time: float = 0.0
) -> None:
    """Write tool result and tool_metadata to METADATA.json.

    Matches BaseTool._update_exec_metadata():
    - tool_metadata: list of dicts with tool_name, output_type, elapsed_time
      (destination connector reads output_type from here)
    - total_elapsed_time: cumulative elapsed time
    """
    metadata_path: Path | None = None
    try:
        metadata_path = Path(execution_data_dir) / "METADATA.json"

        # Read existing metadata if present
        existing: dict = {}
        if fs.exists(metadata_path):
            try:
                existing_raw = fs.read(path=metadata_path, mode="r")
                if existing_raw:
                    existing = json.loads(existing_raw)
            except Exception:
                pass

        # Add tool_metadata (matches BaseTool._update_exec_metadata)
        # The destination connector reads output_type from tool_metadata[-1]
        tool_meta_entry = {
            "tool_name": "structure_tool",
            "output_type": "JSON",
            "elapsed_time": elapsed_time,
        }
        if "tool_metadata" not in existing:
            existing["tool_metadata"] = [tool_meta_entry]
        else:
            existing["tool_metadata"].append(tool_meta_entry)

        existing["total_elapsed_time"] = (
            existing.get("total_elapsed_time", 0.0) + elapsed_time
        )

        fs.write(
            path=metadata_path,
            mode="w",
            data=json.dumps(existing, indent=2),
        )
    except Exception as e:
        logger.error(
            "Failed to write tool result to METADATA.json at '%s': %s",
            metadata_path,
            e,
            exc_info=True,
        )