Function ensure_node

masfactory/visualizer/serialize.py:158–235

(node, *, parent: str | None = None)

        return file_path, line

    def ensure_node(node, *, parent: str | None = None) -> str:
        nid = node_id(node)
        if nid not in node_types:
            nodes.append(nid)
            resolved_type = node_type(node)
            node_types[nid] = resolved_type

            # Human-friendly aliases for internal control nodes.
            if resolved_type == "entry":
                node_aliases[nid] = ["entry"]
            elif resolved_type == "exit":
                node_aliases[nid] = ["exit"]
            elif resolved_type == "Controller":
                node_aliases[nid] = ["controller"]
            elif resolved_type == "TerminateNode":
                node_aliases[nid] = ["terminate"]

            try:
                fp, ln = _source_info(node)
                if fp:
                    node_file_paths[nid] = fp
                if ln is not None:
                    node_line_numbers[nid] = ln
            except Exception:
                pass

            pull = _keys_semantics(getattr(node, "_pull_keys", None))
            push = _keys_semantics(getattr(node, "_push_keys", None))
            if pull is not None:
                node_pull_keys[nid] = pull
            if push is not None:
                node_push_keys[nid] = push

            # Prefer _default_attributes snapshot (this matches what the node actually uses).
            default_attrs = getattr(node, "_default_attributes", None)
            if isinstance(default_attrs, dict):
                node_attributes[nid] = _safe_obj(default_attrs)

            # Message keys on edges (input/output)
            try:
                in_keys = getattr(node, "input_keys", None)
                out_keys = getattr(node, "output_keys", None)
                if isinstance(in_keys, dict) and len(in_keys) > 0:
                    node_input_keys[nid] = _safe_obj(in_keys)
                if isinstance(out_keys, dict) and len(out_keys) > 0:
                    node_output_keys[nid] = _safe_obj(out_keys)
            except Exception:
                pass

            if isinstance(node, Agent):
                try:
                    ins = getattr(node, "instructions", None)
                    if isinstance(ins, str) and ins.strip():
                        node_instructions[nid] = ins
                except Exception:
                    pass
                try:
                    tmpl = getattr(node, "_prompt_template", None)