""" FILE: app/core/ingestion/ingestion_chunk_payload.py DESCRIPTION: Baut das JSON-Objekt für 'mindnet_chunks'. Fix v2.4.3: Integration der zentralen Registry (WP-14) für konsistente Defaults. WP-24c v4.3.0: candidate_pool wird explizit übernommen für Chunk-Attribution. VERSION: 2.4.4 (WP-24c v4.3.0) STATUS: Active """ from __future__ import annotations from typing import Any, Dict, List, Optional # ENTSCHEIDENDER FIX: Import der neutralen Registry-Logik zur Vermeidung von Circular Imports from app.core.registry import load_type_registry # --------------------------------------------------------------------------- # Resolution Helpers (Audited) # --------------------------------------------------------------------------- def _as_list(x): """Sichert die Listen-Integrität für Metadaten wie Tags.""" if x is None: return [] return x if isinstance(x, list) else [x] def _resolve_val(note_type: str, reg: dict, key: str, default: Any) -> Any: """ Hierarchische Suche in der Registry: Type-Spezifisch > Globaler Default. WP-14: Erlaubt dynamische Konfiguration via types.yaml. """ types = reg.get("types", {}) if isinstance(types, dict): t_cfg = types.get(note_type, {}) if isinstance(t_cfg, dict): # Fallback für Key-Varianten (z.B. chunking_profile vs chunk_profile) val = t_cfg.get(key) or t_cfg.get(key.replace("ing", "")) if val is not None: return val defs = reg.get("defaults", {}) or reg.get("global", {}) if isinstance(defs, dict): val = defs.get(key) or defs.get(key.replace("ing", "")) if val is not None: return val return default # --------------------------------------------------------------------------- # Haupt-API # --------------------------------------------------------------------------- def make_chunk_payloads(note: Dict[str, Any], note_path: str, chunks_from_chunker: List[Any], **kwargs) -> List[Dict[str, Any]]: """ Erstellt die Payloads für die Chunks inklusive Audit-Resolution. Nutzt nun die zentrale Registry für alle Fallbacks. """ if isinstance(note, dict) and "frontmatter" in note: fm = note["frontmatter"] else: fm = note or {} # WP-14 Fix: Nutzt übergebene Registry oder lädt sie global reg = kwargs.get("types_cfg") or load_type_registry() note_type = fm.get("type") or "concept" title = fm.get("title") or fm.get("id") or "Untitled" tags = _as_list(fm.get("tags") or []) # Audit: Resolution Hierarchie (Frontmatter > Registry) cp = fm.get("chunking_profile") or fm.get("chunk_profile") if not cp: cp = _resolve_val(note_type, reg, "chunking_profile", "sliding_standard") rw = fm.get("retriever_weight") if rw is None: rw = _resolve_val(note_type, reg, "retriever_weight", 1.0) try: rw = float(rw) except: rw = 1.0 out: List[Dict[str, Any]] = [] for idx, ch in enumerate(chunks_from_chunker): is_dict = isinstance(ch, dict) cid = getattr(ch, "id", None) if not is_dict else ch.get("id") nid = getattr(ch, "note_id", None) if not is_dict else ch.get("note_id") index = getattr(ch, "index", idx) if not is_dict else ch.get("index", idx) text = getattr(ch, "text", "") if not is_dict else ch.get("text", "") window = getattr(ch, "window", text) if not is_dict else ch.get("window", text) prev_id = getattr(ch, "neighbors_prev", None) if not is_dict else ch.get("neighbors_prev") next_id = getattr(ch, "neighbors_next", None) if not is_dict else ch.get("neighbors_next") section = getattr(ch, "section_title", "") if not is_dict else ch.get("section", "") # WP-24c v4.3.0: candidate_pool muss erhalten bleiben für Chunk-Attribution candidate_pool = getattr(ch, "candidate_pool", []) if not is_dict else ch.get("candidate_pool", []) pl: Dict[str, Any] = { "note_id": nid or fm.get("id"), "chunk_id": cid, "title": title, "index": int(index), "ord": int(index) + 1, "type": note_type, "tags": tags, "text": text, "window": window, "neighbors_prev": _as_list(prev_id), "neighbors_next": _as_list(next_id), "section": section, "path": note_path, "source_path": kwargs.get("file_path") or note_path, "retriever_weight": rw, "chunk_profile": cp, "candidate_pool": candidate_pool # WP-24c v4.3.0: Kritisch für Chunk-Attribution } # Audit: Cleanup Pop (Vermeidung von redundanten Alias-Feldern) for alias in ("chunk_num", "Chunk_Number"): pl.pop(alias, None) out.append(pl) return out