23 lines
740 B
Python
23 lines
740 B
Python
"""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 []},
|
|
}
|