griptape-nodes 0.41.0__py3-none-any.whl → 0.43.0__py3-none-any.whl

griptape_nodes/retained_mode/managers/library_manager.py

@@ -9,7 +9,6 @@ import subprocess
 import sys
 import sysconfig
 from dataclasses import dataclass, field
-from enum import StrEnum
 from importlib.resources import files
 from pathlib import Path
 from typing import TYPE_CHECKING, cast
@@ -17,6 +16,7 @@ from typing import TYPE_CHECKING, cast
 import uv
 from packaging.requirements import InvalidRequirement, Requirement
 from pydantic import ValidationError
+from rich.align import Align
 from rich.box import HEAVY_EDGE
 from rich.console import Console
 from rich.panel import Panel
@@ -58,6 +58,9 @@ from griptape_nodes.retained_mode.events.library_events import (
     GetNodeMetadataFromLibraryRequest,
     GetNodeMetadataFromLibraryResultFailure,
     GetNodeMetadataFromLibraryResultSuccess,
+    ListCapableLibraryEventHandlersRequest,
+    ListCapableLibraryEventHandlersResultFailure,
+    ListCapableLibraryEventHandlersResultSuccess,
     ListCategoriesInLibraryRequest,
     ListCategoriesInLibraryResultFailure,
     ListCategoriesInLibraryResultSuccess,
@@ -85,29 +88,44 @@ from griptape_nodes.retained_mode.events.library_events import (
     UnloadLibraryFromRegistryResultSuccess,
 )
 from griptape_nodes.retained_mode.events.object_events import ClearAllObjectStateRequest
+from griptape_nodes.retained_mode.events.payload_registry import PayloadRegistry
 from griptape_nodes.retained_mode.griptape_nodes import GriptapeNodes
+from griptape_nodes.retained_mode.managers.library_lifecycle.library_directory import LibraryDirectory
+from griptape_nodes.retained_mode.managers.library_lifecycle.library_provenance.local_file import (
+    LibraryProvenanceLocalFile,
+)
+from griptape_nodes.retained_mode.managers.library_lifecycle.library_status import LibraryStatus
 from griptape_nodes.retained_mode.managers.os_manager import OSManager
 
 if TYPE_CHECKING:
+    from collections.abc import Callable
     from types import ModuleType
 
     from griptape_nodes.node_library.advanced_node_library import AdvancedNodeLibrary
-    from griptape_nodes.retained_mode.events.base_events import ResultPayload
+    from griptape_nodes.retained_mode.events.base_events import Payload, RequestPayload, ResultPayload
     from griptape_nodes.retained_mode.managers.event_manager import EventManager
 
 logger = logging.getLogger("griptape_nodes")
+console = Console()
 
 
-class LibraryManager:
-    SANDBOX_LIBRARY_NAME = "Sandbox Library"
+def _find_griptape_uv_bin() -> str:
+    """Find the uv binary, checking dedicated Griptape installation first, then system uv.
 
-    class LibraryStatus(StrEnum):
-        """Status of the library that was attempted to be loaded."""
+    Returns:
+        Path to the uv binary to use
+    """
+    # Check for dedicated Griptape uv installation first
+    dedicated_uv_path = xdg_data_home() / "griptape_nodes" / "bin" / "uv"
+    if dedicated_uv_path.exists():
+        return str(dedicated_uv_path)
 
-        GOOD = "GOOD"  # No errors detected during loading. Registered.
-        FLAWED = "FLAWED"  # Some errors detected, but recoverable. Registered.
-        UNUSABLE = "UNUSABLE"  # Errors detected and not recoverable. Not registered.
-        MISSING = "MISSING"  # File not found. Not registered.
+    # Fall back to system uv installation
+    return uv.find_uv_bin()
+
+
+class LibraryManager:
+    SANDBOX_LIBRARY_NAME = "Sandbox Library"
 
     @dataclass
     class LibraryInfo:
@@ -116,7 +134,7 @@ class LibraryManager:
         Includes the status of the library, the file path, and any problems encountered during loading.
         """
 
-        status: LibraryManager.LibraryStatus
+        status: LibraryStatus
         library_path: str
         library_name: str | None = None
         library_version: str | None = None
@@ -124,6 +142,13 @@ class LibraryManager:
 
     _library_file_path_to_info: dict[str, LibraryInfo]
 
+    @dataclass
+    class RegisteredEventHandler:
+        """Information regarding an event handler from a registered library."""
+
+        handler: Callable[[RequestPayload], ResultPayload]
+        library_data: LibrarySchema
+
     # Stable module namespace mappings for workflow serialization
     # These mappings ensure that dynamically loaded modules can be reliably imported
     # in generated workflow code by providing stable, predictable import paths.
@@ -148,10 +173,16 @@ class LibraryManager:
         self._dynamic_to_stable_module_mapping = {}
         self._stable_to_dynamic_module_mapping = {}
         self._library_to_stable_modules = {}
+        self._library_event_handler_mappings: dict[type[Payload], dict[str, LibraryManager.RegisteredEventHandler]] = {}
+        # LibraryDirectory owns the FSMs and manages library lifecycle
+        self._library_directory = LibraryDirectory()
 
         event_manager.assign_manager_to_request_type(
             ListRegisteredLibrariesRequest, self.on_list_registered_libraries_request
         )
+        event_manager.assign_manager_to_request_type(
+            ListCapableLibraryEventHandlersRequest, self.on_list_capable_event_handlers
+        )
         event_manager.assign_manager_to_request_type(
             ListNodeTypesInLibraryRequest, self.on_list_node_types_in_library_request
         )
@@ -223,10 +254,10 @@ class LibraryManager:
 
         # Status emojis mapping
         status_emoji = {
-            LibraryManager.LibraryStatus.GOOD: "✅",
-            LibraryManager.LibraryStatus.FLAWED: "🟡",
-            LibraryManager.LibraryStatus.UNUSABLE: "❌",
-            LibraryManager.LibraryStatus.MISSING: "❓",
+            LibraryStatus.GOOD: "✅",
+            LibraryStatus.FLAWED: "🟡",
+            LibraryStatus.UNUSABLE: "❌",
+            LibraryStatus.MISSING: "❓",
         }
 
         # Add rows for each library info
@@ -277,6 +308,33 @@ class LibraryManager:
                 return library_info
         return None
 
+    def on_register_event_handler(
+        self,
+        request_type: type[RequestPayload],
+        handler: Callable[[RequestPayload], ResultPayload],
+        library_data: LibrarySchema,
+    ) -> None:
+        """Register an event handler for a specific request type from a library."""
+        if self._library_event_handler_mappings.get(request_type) is None:
+            self._library_event_handler_mappings[request_type] = {}
+        self._library_event_handler_mappings[request_type][library_data.name] = LibraryManager.RegisteredEventHandler(
+            handler=handler, library_data=library_data
+        )
+
+    def get_registered_event_handlers(self, request_type: type[Payload]) -> dict[str, RegisteredEventHandler]:
+        """Get all registered event handlers for a specific request type."""
+        return self._library_event_handler_mappings.get(request_type, {})
+
+    def on_list_capable_event_handlers(self, request: ListCapableLibraryEventHandlersRequest) -> ResultPayload:
+        """Get all registered event handlers for a specific request type."""
+        request_type = PayloadRegistry.get_type(request.request_type)
+        if request_type is None:
+            return ListCapableLibraryEventHandlersResultFailure(
+                exception=KeyError(f"Request type '{request.request_type}' is not registered in the PayloadRegistry.")
+            )
+        handler_mappings = self.get_registered_event_handlers(request_type)
+        return ListCapableLibraryEventHandlersResultSuccess(handlers=list(handler_mappings.keys()))
+
     def on_list_registered_libraries_request(self, _request: ListRegisteredLibrariesRequest) -> ResultPayload:
         # Make a COPY of the list
         snapshot_list = LibraryRegistry.list_libraries()
@@ -350,7 +408,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=file_path,
                 library_name=None,
-                status=LibraryManager.LibraryStatus.MISSING,
+                status=LibraryStatus.MISSING,
                 problems=[
                     "Library could not be found at the file path specified. It will be removed from the configuration."
                 ],
@@ -366,7 +424,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=file_path,
                 library_name=None,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=["Library file not formatted as proper JSON."],
             )
         except Exception as err:
@@ -375,7 +433,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=file_path,
                 library_name=None,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[f"Exception occurred when attempting to load the library: {err}."],
             )
 
@@ -399,7 +457,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=file_path,
                 library_name=library_name,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=problems,
             )
         except Exception as err:
@@ -408,7 +466,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=file_path,
                 library_name=library_name,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[f"Library file did not match the library schema specified due to: {err}"],
             )
 
@@ -482,7 +540,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=sandbox_library_dir_as_posix,
                 library_name=LibraryManager.SANDBOX_LIBRARY_NAME,
-                status=LibraryManager.LibraryStatus.MISSING,
+                status=LibraryStatus.MISSING,
                 problems=["Sandbox directory does not exist."],
             )
 
@@ -537,7 +595,7 @@ class LibraryManager:
             return LoadLibraryMetadataFromFileResultFailure(
                 library_path=sandbox_library_dir_as_posix,
                 library_name=LibraryManager.SANDBOX_LIBRARY_NAME,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=["Could not get engine version for sandbox library generation."],
             )
 
@@ -621,7 +679,7 @@ class LibraryManager:
             self._library_file_path_to_info[file_path] = LibraryManager.LibraryInfo(
                 library_path=file_path,
                 library_name=None,
-                status=LibraryManager.LibraryStatus.MISSING,
+                status=LibraryStatus.MISSING,
                 problems=[
                     "Library could not be found at the file path specified. It will be removed from the configuration."
                 ],
@@ -655,7 +713,7 @@ class LibraryManager:
             self._library_file_path_to_info[file_path] = LibraryManager.LibraryInfo(
                 library_path=file_path,
                 library_name=library_data.name,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[
                     f"Library's version string '{library_data.metadata.library_version}' wasn't valid. Must be in major.minor.patch format."
                 ],
@@ -682,7 +740,7 @@ class LibraryManager:
                     library_path=file_path,
                     library_name=library_data.name,
                     library_version=library_version,
-                    status=LibraryManager.LibraryStatus.UNUSABLE,
+                    status=LibraryStatus.UNUSABLE,
                     problems=[
                         f"Failed to load Advanced Library module from '{library_data.advanced_library_path}': {err}"
                     ],
@@ -706,7 +764,7 @@ class LibraryManager:
                 library_path=file_path,
                 library_name=library_data.name,
                 library_version=library_version,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[
                     "Failed because a library with this name was already registered. Check the Settings to ensure duplicate libraries are not being loaded."
                 ],
@@ -735,13 +793,35 @@ class LibraryManager:
                         library_path=file_path,
                         library_name=library_data.name,
                         library_version=library_version,
-                        status=LibraryManager.LibraryStatus.UNUSABLE,
+                        status=LibraryStatus.UNUSABLE,
                         problems=[str(e)],
                     )
                     details = f"Attempted to load Library JSON file from '{json_path}'. Failed when creating the virtual environment: {e}."
                     logger.error(details)
                     return RegisterLibraryFromFileResultFailure()
                 if self._can_write_to_venv_location(library_venv_python_path):
+                    # Check disk space before installing dependencies
+                    config_manager = GriptapeNodes.ConfigManager()
+                    min_space_gb = config_manager.get_config_value("minimum_disk_space_gb_libraries")
+                    if not OSManager.check_available_disk_space(Path(venv_path), min_space_gb):
+                        error_msg = OSManager.format_disk_space_error(Path(venv_path))
+                        logger.error(
+                            "Attempted to load Library JSON from '%s'. Failed when installing dependencies (requires %.1f GB): %s",
+                            json_path,
+                            min_space_gb,
+                            error_msg,
+                        )
+                        self._library_file_path_to_info[file_path] = LibraryManager.LibraryInfo(
+                            library_path=file_path,
+                            library_name=library_data.name,
+                            library_version=library_version,
+                            status=LibraryStatus.UNUSABLE,
+                            problems=[
+                                f"Insufficient disk space for dependencies (requires {min_space_gb} GB): {error_msg}"
+                            ],
+                        )
+                        return RegisterLibraryFromFileResultFailure()
+
                     # Grab the python executable from the virtual environment so that we can pip install there
                     logger.info(
                         "Installing dependencies for library '%s' with pip in venv at %s", library_data.name, venv_path
@@ -771,11 +851,12 @@ class LibraryManager:
         except subprocess.CalledProcessError as e:
             # Failed to create the library
             error_details = f"return code={e.returncode}, stdout={e.stdout}, stderr={e.stderr}"
+
             self._library_file_path_to_info[file_path] = LibraryManager.LibraryInfo(
                 library_path=file_path,
                 library_name=library_data.name,
                 library_version=library_version,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[f"Dependency installation failed: {error_details}"],
             )
             details = f"Attempted to load Library JSON file from '{json_path}'. Failed when installing dependencies: {error_details}"
@@ -807,7 +888,9 @@ class LibraryManager:
                 else:
                     # We had an existing category. Union our changes into it (not replacing anything that matched).
                     existing_category_contents = get_category_result.contents
-                    existing_category_contents.update(library_data_setting.contents)
+                    existing_category_contents |= {
+                        k: v for k, v in library_data_setting.contents.items() if k not in existing_category_contents
+                    }
                     set_category_request = SetConfigCategoryRequest(
                         category=library_data_setting.category, contents=existing_category_contents
                     )
@@ -830,15 +913,15 @@ class LibraryManager:
         self._library_file_path_to_info[file_path] = library_load_results
 
         match library_load_results.status:
-            case LibraryManager.LibraryStatus.GOOD:
+            case LibraryStatus.GOOD:
                 details = f"Successfully loaded Library '{library_data.name}' from JSON file at {json_path}"
                 logger.info(details)
                 return RegisterLibraryFromFileResultSuccess(library_name=library_data.name)
-            case LibraryManager.LibraryStatus.FLAWED:
+            case LibraryStatus.FLAWED:
                 details = f"Successfully loaded Library JSON file from '{json_path}', but one or more nodes failed to load. Check the log for more details."
                 logger.warning(details)
                 return RegisterLibraryFromFileResultSuccess(library_name=library_data.name)
-            case LibraryManager.LibraryStatus.UNUSABLE:
+            case LibraryStatus.UNUSABLE:
                 details = f"Attempted to load Library JSON file from '{json_path}'. Failed because no nodes were loaded. Check the log for more details."
                 logger.error(details)
                 return RegisterLibraryFromFileResultFailure()
@@ -863,10 +946,23 @@ class LibraryManager:
                 logger.error(details)
                 return RegisterLibraryFromRequirementSpecifierResultFailure()
             if self._can_write_to_venv_location(library_python_venv_path):
+                # Check disk space before installing dependencies
+                config_manager = GriptapeNodes.ConfigManager()
+                min_space_gb = config_manager.get_config_value("minimum_disk_space_gb_libraries")
+                if not OSManager.check_available_disk_space(Path(venv_path), min_space_gb):
+                    error_msg = OSManager.format_disk_space_error(Path(venv_path))
+                    logger.error(
+                        "Attempted to install library '%s'. Failed when installing dependencies (requires %.1f GB): %s",
+                        request.requirement_specifier,
+                        min_space_gb,
+                        error_msg,
+                    )
+                    return RegisterLibraryFromRequirementSpecifierResultFailure()
+
                 logger.info("Installing dependency '%s' with pip in venv at %s", package_name, venv_path)
                 subprocess.run(  # noqa: S603
                     [
-                        uv.find_uv_bin(),
+                        _find_griptape_uv_bin(),
                         "pip",
                         "install",
                         request.requirement_specifier,
@@ -922,6 +1018,19 @@ class LibraryManager:
         if library_venv_path.exists():
             logger.debug("Virtual environment already exists at %s", library_venv_path)
         else:
+            # Check disk space before creating virtual environment
+            config_manager = GriptapeNodes.ConfigManager()
+            min_space_gb = config_manager.get_config_value("minimum_disk_space_gb_libraries")
+            if not OSManager.check_available_disk_space(library_venv_path.parent, min_space_gb):
+                error_msg = OSManager.format_disk_space_error(library_venv_path.parent)
+                logger.error(
+                    "Attempted to create virtual environment (requires %.1f GB). Failed: %s", min_space_gb, error_msg
+                )
+                error_message = (
+                    f"Disk space error creating virtual environment (requires {min_space_gb} GB): {error_msg}"
+                )
+                raise RuntimeError(error_message)
+
             try:
                 logger.info("Creating virtual environment at %s with Python %s", library_venv_path, python_version)
                 subprocess.run(  # noqa: S603
@@ -1394,6 +1503,8 @@ class LibraryManager:
         return node_class
 
     def load_all_libraries_from_config(self) -> None:
+        # Comment out lines 1503-1545 and call the _load libraries from provenance system to test the other functionality.
+
         # Load metadata for all libraries to determine which ones can be safely loaded
         metadata_request = LoadMetadataForAllLibrariesRequest()
         metadata_result = self.load_metadata_for_all_libraries_request(metadata_request)
@@ -1441,6 +1552,9 @@ class LibraryManager:
         self._remove_missing_libraries_from_config(config_category=user_libraries_section)
 
     def on_app_initialization_complete(self, _payload: AppInitializationComplete) -> None:
+        GriptapeNodes.EngineIdentityManager().initialize_engine_id()
+        GriptapeNodes.SessionManager().get_saved_session_id()
+
         # App just got init'd. See if there are library JSONs to load!
         self.load_all_libraries_from_config()
 
@@ -1479,6 +1593,107 @@ class LibraryManager:
         # Go tell the Workflow Manager that it's turn is now.
         GriptapeNodes.WorkflowManager().on_libraries_initialization_complete()
 
+        # Print the engine ready message
+        engine_version_request = GetEngineVersionRequest()
+        engine_version_result = GriptapeNodes.get_instance().handle_engine_version_request(engine_version_request)
+        if isinstance(engine_version_result, GetEngineVersionResultSuccess):
+            engine_version = (
+                f"v{engine_version_result.major}.{engine_version_result.minor}.{engine_version_result.patch}"
+            )
+        else:
+            engine_version = "<UNKNOWN ENGINE VERSION>"
+
+        # Get current session ID
+        session_id = GriptapeNodes.get_session_id()
+        session_info = f" | Session: {session_id[:8]}..." if session_id else " | No Session"
+
+        nodes_app_url = os.getenv("GRIPTAPE_NODES_UI_BASE_URL", "https://nodes.griptape.ai")
+        message = Panel(
+            Align.center(
+                f"[bold green]Engine is ready to receive events[/bold green]\n"
+                f"[bold blue]Return to: [link={nodes_app_url}]{nodes_app_url}[/link] to access the Workflow Editor[/bold blue]",
+                vertical="middle",
+            ),
+            title="🚀 Griptape Nodes Engine Started",
+            subtitle=f"[green]{engine_version}{session_info}[/green]",
+            border_style="green",
+            padding=(1, 4),
+        )
+        console.print(message)
+
+    def _load_libraries_from_provenance_system(self) -> None:
+        """Load libraries using the new provenance-based system with FSM.
+
+        This method converts libraries_to_register entries into LibraryProvenanceLocalFile
+        objects and processes them through the LibraryDirectory and LibraryLifecycleFSM systems.
+        """
+        # Get config manager
+        config_mgr = GriptapeNodes.ConfigManager()
+
+        # Get the current libraries_to_register list
+        user_libraries_section = "app_events.on_app_initialization_complete.libraries_to_register"
+        libraries_to_register: list[str] = config_mgr.get_config_value(user_libraries_section)
+
+        if not libraries_to_register:
+            logger.info("No libraries to register from config")
+            return
+
+        # Convert string paths to LibraryProvenanceLocalFile objects
+        for library_path in libraries_to_register:
+            # Skip non-JSON files for now (requirement specifiers will need different handling)
+            if not library_path.endswith(".json"):
+                logger.debug("Skipping non-JSON library path: %s", library_path)
+                continue
+
+            # Create provenance object
+            provenance = LibraryProvenanceLocalFile(file_path=library_path)
+
+            # Add to directory as user candidate (defaults to active=True)
+            # This automatically creates FSM and runs evaluation
+            self._library_directory.add_user_candidate(provenance)
+
+            logger.debug("Added library provenance: %s", provenance.get_display_name())
+
+        # Get all candidates for evaluation
+        all_candidates = self._library_directory.get_all_candidates()
+
+        logger.info("Evaluated %d library candidates through FSM lifecycle", len(all_candidates))
+
+        # Report on conflicts found
+        self._report_library_name_conflicts()
+
+        # Get candidates that are ready for installation
+        installable_candidates = self._library_directory.get_installable_candidates()
+
+        # Log any skipped libraries
+        active_candidates = self._library_directory.get_active_candidates()
+        for candidate in active_candidates:
+            if candidate not in installable_candidates:
+                blockers = self._library_directory.get_installation_blockers(candidate.provenance)
+                if blockers:
+                    blocker_messages = [blocker.message for blocker in blockers]
+                    combined_message = "; ".join(blocker_messages)
+                    logger.info("Skipping library '%s' - %s", candidate.provenance.get_display_name(), combined_message)
+
+        logger.info("Installing and loading %d installable library candidates", len(installable_candidates))
+
+        # Process installable candidates through installation and loading
+        for candidate in installable_candidates:
+            if self._library_directory.install_library(candidate.provenance):
+                self._library_directory.load_library(candidate.provenance)
+
+    def _report_library_name_conflicts(self) -> None:
+        """Report on library name conflicts found during evaluation."""
+        conflicting_names = self._library_directory.get_all_conflicting_library_names()
+        for library_name in conflicting_names:
+            conflicting_provenances = self._library_directory.get_conflicting_provenances(library_name)
+            logger.warning(
+                "Library name conflict detected for '%s' across %d libraries: %s",
+                library_name,
+                len(conflicting_provenances),
+                [p.get_display_name() for p in conflicting_provenances],
+            )
+
     def _load_advanced_library_module(
         self,
         library_data: LibrarySchema,
@@ -1561,7 +1776,7 @@ class LibraryManager:
         has_disqualifying_issues = False
         for issue in version_issues:
             problems.append(issue.message)
-            if issue.severity == LibraryManager.LibraryStatus.UNUSABLE:
+            if issue.severity == LibraryStatus.UNUSABLE:
                 has_disqualifying_issues = True
 
         # Early exit if any version issues are disqualifying
@@ -1570,7 +1785,7 @@ class LibraryManager:
                 library_path=library_file_path,
                 library_name=library_data.name,
                 library_version=library_version,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=problems,
             )
 
@@ -1637,13 +1852,13 @@ class LibraryManager:
 
         # Create a LibraryInfo object based on load successes and problem count.
         if not any_nodes_loaded_successfully:
-            status = LibraryManager.LibraryStatus.UNUSABLE
+            status = LibraryStatus.UNUSABLE
         elif problems:
             # Success, but errors.
-            status = LibraryManager.LibraryStatus.FLAWED
+            status = LibraryStatus.FLAWED
         else:
             # Flawless victory.
-            status = LibraryManager.LibraryStatus.GOOD
+            status = LibraryStatus.GOOD
 
         # Create a LibraryInfo object based on load successes and problem count.
         return LibraryManager.LibraryInfo(
@@ -1735,7 +1950,7 @@ class LibraryManager:
                 library_path=sandbox_library_dir_as_posix,
                 library_name=library_data.name,
                 library_version=library_data.metadata.library_version,
-                status=LibraryManager.LibraryStatus.UNUSABLE,
+                status=LibraryStatus.UNUSABLE,
                 problems=[
                     "Failed because a library with this name was already registered. Check the Settings to ensure duplicate libraries are not being loaded."
                 ],
@@ -1790,7 +2005,7 @@ class LibraryManager:
 
         paths_to_remove = set()
         for library_path, library_info in self._library_file_path_to_info.items():
-            if library_info.status == LibraryManager.LibraryStatus.MISSING:
+            if library_info.status == LibraryStatus.MISSING:
                 # Remove this file path from the config.
                 paths_to_remove.add(library_path.lower())