refactor: redesign skill compilation and document assembly process

api/core/skill/assembler/assemblers.py

@@ -0,0 +1,80 @@
+from collections.abc import Mapping
+
+from core.app.entities.app_asset_entities import AppAssetFileTree
+from core.skill.assembler.common import (
+    build_skill_graph,
+    compute_transitive_dependance,
+    expand_referenced_skill_ids,
+    get_metadata,
+    process_skill_content,
+)
+from core.skill.entities.skill_bundle import Skill, SkillBundle, SkillDependance
+from core.skill.entities.skill_document import SkillDocument
+
+
+class SkillBundleAssembler:
+    _file_tree: AppAssetFileTree
+
+    def __init__(self, file_tree: AppAssetFileTree) -> None:
+        self._file_tree = file_tree
+
+    def assemble_bundle(
+        self,
+        documents: Mapping[str, SkillDocument],
+        assets_id: str,
+    ) -> SkillBundle:
+        direct_skills: dict[str, Skill] = {}
+        for skill_id, doc in documents.items():
+            metadata = get_metadata(doc.content, doc.metadata)
+            direct_dependance = SkillDependance.from_metadata(metadata)
+            direct_skills[skill_id] = Skill(
+                skill_id=skill_id,
+                direct_dependance=direct_dependance,
+                dependance=direct_dependance,
+                content=process_skill_content(doc.content, metadata, self._file_tree, skill_id),
+            )
+
+        graph = build_skill_graph(direct_skills, self._file_tree)
+        transitive_map = compute_transitive_dependance(direct_skills, graph)
+
+        compiled_skills: dict[str, Skill] = {}
+        for skill_id, skill in direct_skills.items():
+            compiled_skills[skill_id] = skill.model_copy(update={"dependance": transitive_map[skill_id]})
+
+        return SkillBundle(asset_tree=self._file_tree, assets_id=assets_id, skills=compiled_skills)
+
+
+class SkillDocumentAssembler:
+    _bundle: SkillBundle
+
+    def __init__(self, bundle: SkillBundle) -> None:
+        self._bundle = bundle
+
+    def assemble_document(self, document: SkillDocument, base_path: str = "") -> Skill:
+        metadata = get_metadata(document.content, document.metadata)
+        direct_dependance = SkillDependance.from_metadata(metadata)
+        resolved_content = process_skill_content(
+            document.content,
+            metadata,
+            self._bundle.asset_tree,
+            document.skill_id,
+            base_path,
+        )
+
+        transitive_dependance = direct_dependance
+        known_skill_ids = set(self._bundle.skills.keys())
+        referenced_skill_ids = expand_referenced_skill_ids(
+            direct_dependance.files, known_skill_ids, self._bundle.asset_tree
+        )
+        for skill_id in sorted(referenced_skill_ids):
+            referenced_skill = self._bundle.get(skill_id)
+            if referenced_skill is None:
+                continue
+            transitive_dependance = transitive_dependance | referenced_skill.dependance
+
+        return Skill(
+            skill_id=document.skill_id,
+            direct_dependance=direct_dependance,
+            dependance=transitive_dependance,
+            content=resolved_content,
+        )