Method __init__

generators/node_validator.py:44–119

Initialize NodeBase with node database and optionally embedding model. Args: cfg: Configuration object with node_base settings

(self, cfg: Any)

    """

    def __init__(self, cfg: Any):
        """
        Initialize NodeBase with node database and optionally embedding model.

        Args:
            cfg: Configuration object with node_base settings
        """
        logging.info("Initializing NodeBase...")

        node_list_path = cfg.node_base.node_list_path
        logging.debug(f"Loading node list from: {node_list_path}")
        try:
            with open(node_list_path) as f:
                self.node_list = json.load(f)
            logging.info(f"Loaded {len(self.node_list)} node types")
        except Exception as e:
            logging.error(f"Failed to load node list: {e}")
            raise

        node_info_db_path = cfg.node_base.node_info_db_path
        logging.debug(f"Loading node info database from: {node_info_db_path}")
        try:
            self.node_info_db = TinyDB(node_info_db_path)
            node_info_count = len(self.node_info_db.all())
            logging.info(f"Loaded {node_info_count} node info entries")
        except (TypeError, ValueError) as e:
            # Handle case where file is in old format (plain JSON array instead of TinyDB format)
            logging.warning("Node info database appears to be in old format, attempting to convert...")
            try:
                with open(node_info_db_path, encoding="utf-8") as f:
                    old_data = json.load(f)

                if isinstance(old_data, list):
                    db = TinyDB(node_info_db_path)
                    db.truncate()
                    db.insert_multiple(old_data)
                    db.close()
                    logging.info(f"Converted {len(old_data)} entries from old format to TinyDB format")

                    self.node_info_db = TinyDB(node_info_db_path)
                    node_info_count = len(self.node_info_db.all())
                    logging.info(f"Loaded {node_info_count} node info entries")
                else:
                    raise ValueError(f"Unexpected data format in {node_info_db_path}")
            except Exception as conv_e:
                logging.error(f"Failed to convert node info database: {conv_e}")
                logging.error("Please run 'UpdateNodeCatalog' node to regenerate the catalog files")
                raise ValueError(
                    f"Node info database format is invalid. Please run 'UpdateNodeCatalog' node to regenerate it.\n"
                    f"Original error: {e}\n"
                    f"Conversion error: {conv_e}"
                ) from conv_e
        except Exception as e:
            logging.error(f"Failed to load node info database: {e}")
            raise

        self.cache_node_meta_info = {}