Function _parse_entities_plan

openkb/agent/compiler.py:656–673

Extract the entities group from a plan dict, with graceful fallback. Returns ``{"create": [...], "update": [...], "related": [...]}``. A missing/malformed ``entities`` key yields empty lists, so older or partial LLM responses never raise.

(parsed: object, valid_types: frozenset | None = None)

def _parse_entities_plan(parsed: object, valid_types: frozenset | None = None) -> dict:
    """Extract the entities group from a plan dict, with graceful fallback.

    Returns ``{"create": [...], "update": [...], "related": [...]}``. A
    missing/malformed ``entities`` key yields empty lists, so older or
    partial LLM responses never raise.
    """
    empty = {"create": [], "update": [], "related": []}
    if not isinstance(parsed, dict):
        return empty
    group = parsed.get("entities")
    if not isinstance(group, dict):
        return empty
    return {
        "create": _filter_entity_items(group.get("create", []), valid_types),
        "update": _filter_entity_items(group.get("update", []), valid_types),
        "related": _filter_related_slugs(group.get("related", [])),
    }


# ---------------------------------------------------------------------------