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

griptape_nodes/retained_mode/managers/workflow_manager.py

@@ -1332,6 +1332,9 @@ class WorkflowManager:
             right=ast.Constant(value=None),
         )
 
+        # Generate the ensure flow context function call
+        ensure_context_call = self._generate_ensure_flow_context_call()
+
         # Create conditional logic: workflow_executor = workflow_executor or LocalWorkflowExecutor()
         executor_assign = ast.Assign(
             targets=[ast.Name(id="workflow_executor", ctx=ast.Store())],
@@ -1373,7 +1376,7 @@ class WorkflowManager:
         func_def = ast.FunctionDef(
             name="execute_workflow",
             args=args,
-            body=[executor_assign, run_call, return_stmt],
+            body=[ensure_context_call, executor_assign, run_call, return_stmt],
             decorator_list=[],
             returns=return_annotation,
             type_params=[],
@@ -1592,7 +1595,201 @@ class WorkflowManager:
         )
         ast.fix_missing_locations(if_node)
 
-        return [func_def, if_node]
+        # Generate the ensure flow context function
+        ensure_context_func = self._generate_ensure_flow_context_function(import_recorder)
+
+        return [ensure_context_func, func_def, if_node]
+
+    def _generate_ensure_flow_context_function(
+        self,
+        import_recorder: ImportRecorder,
+    ) -> ast.FunctionDef:
+        """Generates the _ensure_workflow_context function for the serialized workflow file."""
+        import_recorder.add_from_import("griptape_nodes.retained_mode.events.flow_events", "GetTopLevelFlowRequest")
+        import_recorder.add_from_import(
+            "griptape_nodes.retained_mode.events.flow_events", "GetTopLevelFlowResultSuccess"
+        )
+
+        # Function signature: def _ensure_workflow_context():
+        func_def = ast.FunctionDef(
+            name="_ensure_workflow_context",
+            args=ast.arguments(
+                posonlyargs=[],
+                args=[],
+                vararg=None,
+                kwonlyargs=[],
+                kw_defaults=[],
+                kwarg=None,
+                defaults=[],
+            ),
+            body=[],
+            decorator_list=[],
+            returns=None,
+            type_params=[],
+        )
+
+        context_manager_assign = ast.Assign(
+            targets=[ast.Name(id="context_manager", ctx=ast.Store())],
+            value=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="GriptapeNodes", ctx=ast.Load()),
+                    attr="ContextManager",
+                    ctx=ast.Load(),
+                ),
+                args=[],
+                keywords=[],
+            ),
+        )
+
+        # if not context_manager.has_current_flow():
+        has_flow_check = ast.UnaryOp(
+            op=ast.Not(),
+            operand=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="context_manager", ctx=ast.Load()),
+                    attr="has_current_flow",
+                    ctx=ast.Load(),
+                ),
+                args=[],
+                keywords=[],
+            ),
+        )
+
+        # top_level_flow_request = GetTopLevelFlowRequest()  # noqa: ERA001
+        flow_request_assign = ast.Assign(
+            targets=[ast.Name(id="top_level_flow_request", ctx=ast.Store())],
+            value=ast.Call(
+                func=ast.Name(id="GetTopLevelFlowRequest", ctx=ast.Load()),
+                args=[],
+                keywords=[],
+            ),
+        )
+
+        # top_level_flow_result = GriptapeNodes.handle_request(top_level_flow_request)  # noqa: ERA001
+        flow_result_assign = ast.Assign(
+            targets=[ast.Name(id="top_level_flow_result", ctx=ast.Store())],
+            value=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="GriptapeNodes", ctx=ast.Load()),
+                    attr="handle_request",
+                    ctx=ast.Load(),
+                ),
+                args=[ast.Name(id="top_level_flow_request", ctx=ast.Load())],
+                keywords=[],
+            ),
+        )
+
+        # isinstance check and flow_name is not None
+        isinstance_check = ast.Call(
+            func=ast.Name(id="isinstance", ctx=ast.Load()),
+            args=[
+                ast.Name(id="top_level_flow_result", ctx=ast.Load()),
+                ast.Name(id="GetTopLevelFlowResultSuccess", ctx=ast.Load()),
+            ],
+            keywords=[],
+        )
+
+        flow_name_check = ast.Compare(
+            left=ast.Attribute(
+                value=ast.Name(id="top_level_flow_result", ctx=ast.Load()),
+                attr="flow_name",
+                ctx=ast.Load(),
+            ),
+            ops=[ast.IsNot()],
+            comparators=[ast.Constant(value=None)],
+        )
+
+        success_condition = ast.BoolOp(
+            op=ast.And(),
+            values=[isinstance_check, flow_name_check],
+        )
+
+        # flow_manager = GriptapeNodes.FlowManager()  # noqa: ERA001
+        flow_manager_assign = ast.Assign(
+            targets=[ast.Name(id="flow_manager", ctx=ast.Store())],
+            value=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="GriptapeNodes", ctx=ast.Load()),
+                    attr="FlowManager",
+                    ctx=ast.Load(),
+                ),
+                args=[],
+                keywords=[],
+            ),
+        )
+
+        # flow_obj = flow_manager.get_flow_by_name(top_level_flow_result.flow_name)  # noqa: ERA001
+        flow_obj_assign = ast.Assign(
+            targets=[ast.Name(id="flow_obj", ctx=ast.Store())],
+            value=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="flow_manager", ctx=ast.Load()),
+                    attr="get_flow_by_name",
+                    ctx=ast.Load(),
+                ),
+                args=[
+                    ast.Attribute(
+                        value=ast.Name(id="top_level_flow_result", ctx=ast.Load()),
+                        attr="flow_name",
+                        ctx=ast.Load(),
+                    )
+                ],
+                keywords=[],
+            ),
+        )
+
+        # context_manager.push_flow(flow_obj)  # noqa: ERA001
+        push_flow_call = ast.Expr(
+            value=ast.Call(
+                func=ast.Attribute(
+                    value=ast.Name(id="context_manager", ctx=ast.Load()),
+                    attr="push_flow",
+                    ctx=ast.Load(),
+                ),
+                args=[ast.Name(id="flow_obj", ctx=ast.Load())],
+                keywords=[],
+            ),
+        )
+
+        # Build the inner if statement for success condition
+        success_if = ast.If(
+            test=success_condition,
+            body=[
+                flow_manager_assign,
+                flow_obj_assign,
+                push_flow_call,
+            ],
+            orelse=[],
+        )
+
+        # Build the main if statement
+        main_if = ast.If(
+            test=has_flow_check,
+            body=[
+                flow_request_assign,
+                flow_result_assign,
+                success_if,
+            ],
+            orelse=[],
+        )
+
+        # Set the function body
+        func_def.body = [context_manager_assign, main_if]
+        ast.fix_missing_locations(func_def)
+
+        return func_def
+
+    def _generate_ensure_flow_context_call(
+        self,
+    ) -> ast.Expr:
+        """Generates the call to _ensure_workflow_context() function."""
+        return ast.Expr(
+            value=ast.Call(
+                func=ast.Name(id="_ensure_workflow_context", ctx=ast.Load()),
+                args=[],
+                keywords=[],
+            )
+        )
 
     def _generate_workflow_run_prerequisite_code(
         self,

griptape_nodes/traits/file_system_picker.py

@@ -0,0 +1,127 @@
+from collections.abc import Callable
+from dataclasses import dataclass, field
+from typing import Any
+
+from griptape_nodes.exe_types.core_types import Parameter, Trait
+
+
+@dataclass(eq=False)
+class FileSystemPicker(Trait):
+    allow_files: bool = False
+    allow_directories: bool = True
+    multiple: bool = False
+    file_types: list[str] = field(default_factory=list)
+    file_extensions: list[str] = field(default_factory=list)
+    exclude_patterns: list[str] = field(default_factory=list)
+    include_patterns: list[str] = field(default_factory=list)
+    max_file_size: int | None = None
+    min_file_size: int | None = None
+    workspace_only: bool = True
+    initial_path: str | None = None
+    element_id: str = field(default_factory=lambda: "FileSystemPicker")
+
+    def __init__(  # noqa: PLR0913
+        self,
+        *,
+        allow_files: bool = False,
+        allow_directories: bool = True,
+        multiple: bool = False,
+        file_types: list[str] | None = None,
+        file_extensions: list[str] | None = None,
+        exclude_patterns: list[str] | None = None,
+        include_patterns: list[str] | None = None,
+        max_file_size: int | None = None,
+        min_file_size: int | None = None,
+        workspace_only: bool = True,
+        initial_path: str | None = None,
+    ) -> None:
+        super().__init__()
+        self.allow_files = allow_files
+        self.allow_directories = allow_directories
+        self.multiple = multiple
+        self.file_types = file_types or []
+        self.file_extensions = file_extensions or []
+        self.exclude_patterns = exclude_patterns or []
+        self.include_patterns = include_patterns or []
+        self.max_file_size = max_file_size
+        self.min_file_size = min_file_size
+        self.workspace_only = workspace_only
+        self.initial_path = initial_path
+
+    @classmethod
+    def get_trait_keys(cls) -> list[str]:
+        return ["fileSystemPicker", "file_picker", "folder_picker"]
+
+    def ui_options_for_trait(self) -> dict[str, Any]:
+        """Generate the fileSystemPicker UI options dictionary."""
+        options: dict[str, Any] = {
+            "allowFiles": self.allow_files,
+            "allowDirectories": self.allow_directories,
+            "multiple": self.multiple,
+            "workspaceOnly": self.workspace_only,
+        }
+
+        # Add file types/extensions
+        if self.file_types:
+            options["fileTypes"] = self.file_types
+        elif self.file_extensions:
+            options["fileExtensions"] = self.file_extensions
+
+        # Add patterns
+        if self.exclude_patterns:
+            options["excludePatterns"] = self.exclude_patterns
+        if self.include_patterns:
+            options["includePatterns"] = self.include_patterns
+
+        # Add size limits
+        if self.max_file_size is not None:
+            options["maxFileSize"] = self.max_file_size
+        if self.min_file_size is not None:
+            options["minFileSize"] = self.min_file_size
+
+        # Add initial path
+        if self.initial_path:
+            options["initialPath"] = self.initial_path
+
+        return {"fileSystemPicker": options}
+
+    def validators_for_trait(self) -> list[Callable[[Parameter, Any], Any]]:
+        """Validate file system picker configuration."""
+
+        def validate(param: Parameter, value: Any) -> None:  # noqa: ARG001
+            # Validate that at least one selection type is enabled
+            if not self.allow_files and not self.allow_directories:
+                msg = "At least one of allow_files or allow_directories must be True"
+                raise ValueError(msg)
+
+            # Validate file size limits
+            if (
+                self.max_file_size is not None
+                and self.min_file_size is not None
+                and self.max_file_size < self.min_file_size
+            ):
+                msg = "max_file_size cannot be less than min_file_size"
+                raise ValueError(msg)
+
+            # Validate that file types/extensions are valid
+            all_file_types = self.file_types + self.file_extensions
+            for file_type in all_file_types:
+                if not file_type.startswith("."):
+                    msg = f"File type '{file_type}' must start with a dot (e.g., '.py')"
+                    raise ValueError(msg)
+
+        return [validate]
+
+    def converters_for_trait(self) -> list[Callable]:
+        """Convert file system picker values if needed."""
+
+        def converter(value: Any) -> Any:
+            # If value is a string and we expect a list, convert it
+            if isinstance(value, str) and self.multiple:
+                return [value] if value else []
+            return value
+
+        return [converter]
+
+
+# These Traits get added to a list on the parameter. When they are added they apply their functions to the parameter.

griptape_nodes/updater/__init__.py

@@ -38,7 +38,7 @@ def _download_and_run_installer() -> None:
     """Runs the update commands for the engine."""
     console.print("[bold green]Updating self...[/bold green]")
     try:
-        subprocess.run(  # noqa: S603
+        subprocess.run(
             ["uv", "tool", "upgrade", "griptape-nodes"],  # noqa: S607
             text=True,
             capture_output=True,
@@ -57,7 +57,7 @@ def _sync_libraries() -> None:
     """Syncs the libraries for the engine."""
     console.print("[bold green]Syncing libraries...[/bold green]")
     try:
-        subprocess.run(  # noqa: S603
+        subprocess.run(
             ["griptape-nodes", "libraries", "sync"],  # noqa: S607
             text=True,
             capture_output=True,

griptape_nodes/utils/image_preview.py

@@ -0,0 +1,128 @@
+"""Image preview utilities for generating thumbnails and previews."""
+
+import base64
+import io
+from pathlib import Path
+
+from PIL import Image
+
+from griptape_nodes.retained_mode.griptape_nodes import logger
+
+
+def create_image_preview(
+    image_path: Path, max_width: int = 512, max_height: int = 512, quality: int = 85, image_format: str = "WEBP"
+) -> str | None:
+    """Create a small preview image from a file path.
+
+    Args:
+        image_path: Path to the image file
+        max_width: Maximum width for the preview
+        max_height: Maximum height for the preview
+        quality: WebP quality (1-100)
+        image_format: Output format (WEBP, JPEG, PNG, etc.)
+
+    Returns:
+        Base64 encoded data URL of the preview, or None if failed
+    """
+    try:
+        # Open and resize the image
+        with Image.open(image_path) as img:
+            # Convert to RGB if necessary (for WebP/JPEG output)
+            if image_format.upper() in ("WEBP", "JPEG") and img.mode in ("RGBA", "LA", "P"):
+                converted_img = img.convert("RGB")
+            else:
+                converted_img = img
+
+            # Calculate new size maintaining aspect ratio
+            converted_img.thumbnail((max_width, max_height), Image.Resampling.LANCZOS)
+
+            # Save to bytes buffer
+            buffer = io.BytesIO()
+            converted_img.save(buffer, format=image_format, quality=quality, optimize=True)
+            buffer.seek(0)
+
+            # Convert to base64
+            image_bytes = buffer.getvalue()
+            base64_data = base64.b64encode(image_bytes).decode("utf-8")
+
+            # Create data URL
+            mime_type = f"image/{image_format.lower()}"
+            data_url = f"data:{mime_type};base64,{base64_data}"
+
+            logger.debug(f"Created preview for {image_path}: {img.size} -> {len(image_bytes)} bytes")
+            return data_url
+
+    except Exception as e:
+        logger.warning(f"Failed to create preview for {image_path}: {e}")
+        return None
+
+
+def create_image_preview_from_bytes(
+    image_bytes: bytes, max_width: int = 512, max_height: int = 512, quality: int = 85, image_format: str = "WEBP"
+) -> str | None:
+    """Create a small preview image from bytes.
+
+    Args:
+        image_bytes: Raw image bytes
+        max_width: Maximum width for the preview
+        max_height: Maximum height for the preview
+        quality: WebP quality (1-100)
+        image_format: Output format (WEBP, JPEG, PNG, etc.)
+
+    Returns:
+        Base64 encoded data URL of the preview, or None if failed
+    """
+    try:
+        # Open image from bytes
+        with Image.open(io.BytesIO(image_bytes)) as img:
+            # Convert to RGB if necessary (for WebP/JPEG output)
+            if image_format.upper() in ("WEBP", "JPEG") and img.mode in ("RGBA", "LA", "P"):
+                converted_img = img.convert("RGB")
+            else:
+                converted_img = img
+
+            # Calculate new size maintaining aspect ratio
+            converted_img.thumbnail((max_width, max_height), Image.Resampling.LANCZOS)
+
+            # Save to bytes buffer
+            buffer = io.BytesIO()
+            converted_img.save(buffer, format=image_format, quality=quality, optimize=True)
+            buffer.seek(0)
+
+            # Convert to base64
+            preview_bytes = buffer.getvalue()
+            base64_data = base64.b64encode(preview_bytes).decode("utf-8")
+
+            # Create data URL
+            mime_type = f"image/{image_format.lower()}"
+            data_url = f"data:{mime_type};base64,{base64_data}"
+
+            logger.debug(f"Created preview from bytes: {len(image_bytes)} -> {len(preview_bytes)} bytes")
+            return data_url
+
+    except Exception as e:
+        logger.warning(f"Failed to create preview from bytes: {e}")
+        return None
+
+
+def get_image_info(image_path: Path) -> dict | None:
+    """Get basic information about an image file.
+
+    Args:
+        image_path: Path to the image file
+
+    Returns:
+        Dictionary with image info (width, height, format, mode), or None if failed
+    """
+    try:
+        with Image.open(image_path) as img:
+            return {
+                "width": img.width,
+                "height": img.height,
+                "format": img.format,
+                "mode": img.mode,
+                "size_bytes": image_path.stat().st_size,
+            }
+    except Exception as e:
+        logger.warning(f"Failed to get image info for {image_path}: {e}")
+        return None

griptape_nodes/version_compatibility/versions/v0_39_0/modified_parameters_set_removal.py

@@ -7,7 +7,7 @@ from griptape_nodes.retained_mode.events.app_events import (
     GetEngineVersionResultSuccess,
 )
 from griptape_nodes.retained_mode.griptape_nodes import GriptapeNodes, Version
-from griptape_nodes.retained_mode.managers.library_manager import LibraryManager
+from griptape_nodes.retained_mode.managers.library_lifecycle.library_status import LibraryStatus
 from griptape_nodes.retained_mode.managers.version_compatibility_manager import (
     LibraryVersionCompatibilityCheck,
     LibraryVersionCompatibilityIssue,
@@ -47,13 +47,13 @@ class ModifiedParametersSetRemovalCheck(LibraryVersionCompatibilityCheck):
                     message=f"This library (built for engine version {library_version_str}) is incompatible with Griptape Nodes 0.39+. "
                     "The 'modified_parameters_set' parameter has been removed from BaseNode methods: 'after_incoming_connection', 'after_outgoing_connection', 'after_incoming_connection_removed', 'after_outgoing_connection_removed', 'before_value_set', and 'after_value_set'. "
                     "If this library overrides any of these methods, it will not load or function properly. Please update to a newer version of this library or contact the library author immediately.",
-                    severity=LibraryManager.LibraryStatus.UNUSABLE,
+                    severity=LibraryStatus.UNUSABLE,
                 ),
                 LibraryVersionCompatibilityIssue(
                     message=f"This library (built for engine version {library_version_str}) is incompatible with Griptape Nodes 0.39+."
                     "The 'ui_options' field has been modified on all Elements. In order to function properly, all nodes must update ui_options by setting its value to a new dictionary. Updating ui_options by accessing the private field _ui_options will no longer create UI updates in the editor."
                     "If this library accesses the private _ui_options field, it will not update the editor properly. Please update to a newer version of this library or contact the library author immediately.",
-                    severity=LibraryManager.LibraryStatus.UNUSABLE,
+                    severity=LibraryStatus.UNUSABLE,
                 ),
             ]
         if current_engine_version >= Version(0, 38, 0):
@@ -63,13 +63,13 @@ class ModifiedParametersSetRemovalCheck(LibraryVersionCompatibilityCheck):
                     message=f"WARNING: The 'modified_parameters_set' parameter will be removed in Griptape Nodes 0.39 from BaseNode methods: 'after_incoming_connection', 'after_outgoing_connection', 'after_incoming_connection_removed', 'after_outgoing_connection_removed', 'before_value_set', and 'after_value_set'. "
                     f"This library (built for engine version {library_version_str}) must be updated before the 0.39 release. "
                     "If this library overrides any of these methods, it will fail to load in 0.39. If not, no action is necessary. Please contact the library author to confirm whether this library is impacted.",
-                    severity=LibraryManager.LibraryStatus.FLAWED,
+                    severity=LibraryStatus.FLAWED,
                 ),
                 LibraryVersionCompatibilityIssue(
                     message="WARNING: The 'ui_options' field has been modified in Griptape Nodes 0.38 on all BaseNodeElements."
                     "In order to function properly, all nodes must update ui_options by setting its value to a new dictionary. Updating ui_options by accessing the private field _ui_options will no longer create UI updates in the editor."
                     "If this library accesses the private _ui_options field, it will not update the editor properly. Please update to a newer version of this library or contact the library author immediately.",
-                    severity=LibraryManager.LibraryStatus.FLAWED,
+                    severity=LibraryStatus.FLAWED,
                 ),
             ]