"""Normalize heterogeneous sources into the assistant import-package shape."""

from __future__ import annotations

from typing import Any


def export_package_from_skill_ir(ir: dict[str, Any]) -> dict[str, Any]:
    skill = ir.get("skill") or {}
    return {
        "source": {
            "type": "skill",
            "skill_name": skill.get("name"),
            "source_digest": skill.get("source_digest"),
        },
        "wiki_pages": [ir.get("wiki", {}).get("skill_page")],
        "documents": ir.get("files") or [],
        "chunks": [],
        "entities": ir.get("entities") or [],
        "relations": ir.get("relations") or [],
        "graph": {"nodes": ir.get("entities") or [], "edges": ir.get("relations") or []},
    }