griptape-nodes 0.44.0__py3-none-any.whl → 0.45.1__py3-none-any.whl

griptape_nodes/retained_mode/events/workflow_events.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from typing import Literal
 
 from griptape_nodes.node_library.workflow_registry import WorkflowMetadata
 from griptape_nodes.retained_mode.events.base_events import (
@@ -364,3 +365,233 @@ class PublishWorkflowResultSuccess(ResultPayloadSuccess):
 @PayloadRegistry.register
 class PublishWorkflowResultFailure(ResultPayloadFailure):
     """Workflow publish failed. Common causes: workflow not found, publish error, file system error."""
+
+
+@dataclass
+@PayloadRegistry.register
+class BranchWorkflowRequest(RequestPayload):
+    """Create a branch (copy) of an existing workflow with branch tracking.
+
+    Use when: Creating workflow variants, branching workflows for experimentation,
+    creating personal copies of shared workflows, preparing for workflow collaboration.
+
+    Args:
+        workflow_name: Name of the workflow to branch
+        branched_workflow_name: Name for the branched workflow (None for auto-generated)
+
+    Results: BranchWorkflowResultSuccess (with branch name) | BranchWorkflowResultFailure (branch error)
+    """
+
+    workflow_name: str
+    branched_workflow_name: str | None = None
+
+
+@dataclass
+@PayloadRegistry.register
+class BranchWorkflowResultSuccess(WorkflowAlteredMixin, ResultPayloadSuccess):
+    """Workflow branched successfully.
+
+    Args:
+        branched_workflow_name: Name of the created branch
+        original_workflow_name: Name of the original workflow
+    """
+
+    branched_workflow_name: str
+    original_workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class BranchWorkflowResultFailure(ResultPayloadFailure):
+    """Workflow branch failed. Common causes: workflow not found, name conflict, save error."""
+
+
+@dataclass
+@PayloadRegistry.register
+class MergeWorkflowBranchRequest(RequestPayload):
+    """Merge a branch back into its source workflow, removing the branch when complete.
+
+    Use when: Integrating branch changes back into the original workflow, consolidating
+    successful branch experiments, applying approved branch modifications to source.
+
+    Args:
+        workflow_name: Name of the branch workflow to merge back into its source
+
+    Results: MergeWorkflowBranchResultSuccess (with merge details) | MergeWorkflowBranchResultFailure (merge error)
+    """
+
+    workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class MergeWorkflowBranchResultSuccess(WorkflowAlteredMixin, ResultPayloadSuccess):
+    """Branch merge back to source completed successfully.
+
+    Args:
+        merged_workflow_name: Name of the source workflow after merge
+    """
+
+    merged_workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class MergeWorkflowBranchResultFailure(ResultPayloadFailure):
+    """Workflow branch merge failed."""
+
+
+@dataclass
+@PayloadRegistry.register
+class ResetWorkflowBranchRequest(RequestPayload):
+    """Reset a branch to match its source workflow, discarding branch changes.
+
+    Use when: Discarding branch modifications, reverting branch to source state,
+    abandoning branch experiments, syncing branch with latest source changes.
+
+    Args:
+        workflow_name: Name of the branch workflow to reset to its source
+
+    Results: ResetWorkflowBranchResultSuccess (with reset details) | ResetWorkflowBranchResultFailure (reset error)
+    """
+
+    workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class ResetWorkflowBranchResultSuccess(WorkflowAlteredMixin, ResultPayloadSuccess):
+    """Branch reset to source completed successfully.
+
+    Args:
+        reset_workflow_name: Name of the branch workflow after reset
+    """
+
+    reset_workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class ResetWorkflowBranchResultFailure(ResultPayloadFailure):
+    """Workflow branch reset failed. Common causes: workflows not branch-related, reset conflict, save error."""
+
+
+@dataclass
+@PayloadRegistry.register
+class CompareWorkflowsRequest(RequestPayload):
+    """Compare two workflows to determine if one is ahead, behind, or up-to-date relative to the other.
+
+    Use when: Checking if branched workflows need updates, determining if local changes exist,
+    managing workflow synchronization, preparing for merge operations.
+
+    Args:
+        workflow_name: Name of the workflow to evaluate
+        compare_workflow_name: Name of the workflow to compare against
+
+    Results: CompareWorkflowsResultSuccess (with status details) | CompareWorkflowsResultFailure (evaluation error)
+    """
+
+    workflow_name: str
+    compare_workflow_name: str
+
+
+@dataclass
+@PayloadRegistry.register
+class CompareWorkflowsResultSuccess(WorkflowNotAlteredMixin, ResultPayloadSuccess):
+    """Workflow comparison completed successfully.
+
+    Args:
+        workflow_name: Name of the evaluated workflow
+        compare_workflow_name: Name of the workflow being compared against (if any)
+        status: Status relative to source - "up_to_date", "ahead", "behind", "diverged", or "no_source"
+        workflow_last_modified: Last modified timestamp of the workflow
+        source_last_modified: Last modified timestamp of the source (if exists)
+        details: Additional details about the comparison
+    """
+
+    workflow_name: str
+    compare_workflow_name: str | None
+    status: Literal["up_to_date", "ahead", "behind", "diverged", "no_source"]
+    workflow_last_modified: str | None
+    source_last_modified: str | None
+    details: str
+
+
+@dataclass
+@PayloadRegistry.register
+class CompareWorkflowsResultFailure(ResultPayloadFailure):
+    """Workflow comparison failed. Common causes: workflow not found, source not accessible, comparison error."""
+
+
+@dataclass
+@PayloadRegistry.register
+class MoveWorkflowRequest(RequestPayload):
+    """Move a workflow to a different directory in the workspace.
+
+    Use when: Organizing workflows into directories, restructuring workflow hierarchies,
+    moving workflows to categorized folders, cleaning up workspace organization.
+
+    Args:
+        workflow_name: Name of the workflow to move
+        target_directory: Target directory path relative to workspace root
+
+    Results: MoveWorkflowResultSuccess (with new path) | MoveWorkflowResultFailure (move error)
+    """
+
+    workflow_name: str
+    target_directory: str
+
+
+@dataclass
+@PayloadRegistry.register
+class MoveWorkflowResultSuccess(WorkflowAlteredMixin, ResultPayloadSuccess):
+    """Workflow moved successfully.
+
+    Args:
+        moved_file_path: New file path after the move
+    """
+
+    moved_file_path: str
+
+
+@dataclass
+@PayloadRegistry.register
+class MoveWorkflowResultFailure(ResultPayloadFailure):
+    """Workflow move failed. Common causes: workflow not found, invalid target directory, file system error."""
+
+
+@dataclass
+@PayloadRegistry.register
+class RegisterWorkflowsFromConfigRequest(RequestPayload):
+    """Register workflows from configuration section.
+
+    Use when: Loading workflows from configuration after library initialization,
+    registering workflows from synced directories, batch workflow registration.
+
+    Args:
+        config_section: Configuration section path containing workflow paths to register
+
+    Results: RegisterWorkflowsFromConfigResultSuccess (with count) | RegisterWorkflowsFromConfigResultFailure (registration error)
+    """
+
+    config_section: str
+
+
+@dataclass
+@PayloadRegistry.register
+class RegisterWorkflowsFromConfigResultSuccess(WorkflowNotAlteredMixin, ResultPayloadSuccess):
+    """Workflows registered from configuration successfully.
+
+    Args:
+        succeeded_workflows: List of workflow names that were successfully registered
+        failed_workflows: List of workflow names that failed to register
+    """
+
+    succeeded_workflows: list[str]
+    failed_workflows: list[str]
+
+
+@dataclass
+@PayloadRegistry.register
+class RegisterWorkflowsFromConfigResultFailure(WorkflowNotAlteredMixin, ResultPayloadFailure):
+    """Workflow registration from configuration failed. Common causes: configuration not found, invalid paths, registration errors."""

griptape_nodes/retained_mode/griptape_nodes.py

@@ -69,6 +69,7 @@ if TYPE_CHECKING:
     from griptape_nodes.retained_mode.managers.static_files_manager import (
         StaticFilesManager,
     )
+    from griptape_nodes.retained_mode.managers.sync_manager import SyncManager
     from griptape_nodes.retained_mode.managers.version_compatibility_manager import (
         VersionCompatibilityManager,
     )
@@ -138,6 +139,7 @@ class GriptapeNodes(metaclass=SingletonMeta):
     _version_compatibility_manager: VersionCompatibilityManager
     _session_manager: SessionManager
     _engine_identity_manager: EngineIdentityManager
+    _sync_manager: SyncManager
 
     def __init__(self) -> None:
         from griptape_nodes.retained_mode.managers.agent_manager import AgentManager
@@ -161,6 +163,7 @@ class GriptapeNodes(metaclass=SingletonMeta):
         from griptape_nodes.retained_mode.managers.static_files_manager import (
             StaticFilesManager,
         )
+        from griptape_nodes.retained_mode.managers.sync_manager import SyncManager
         from griptape_nodes.retained_mode.managers.version_compatibility_manager import (
             VersionCompatibilityManager,
         )
@@ -189,6 +192,7 @@ class GriptapeNodes(metaclass=SingletonMeta):
             self._version_compatibility_manager = VersionCompatibilityManager(self._event_manager)
             self._session_manager = SessionManager(self._event_manager)
             self._engine_identity_manager = EngineIdentityManager(self._event_manager)
+            self._sync_manager = SyncManager(self._event_manager, self._config_manager)
 
             # Assign handlers now that these are created.
             self._event_manager.assign_manager_to_request_type(
@@ -326,6 +330,10 @@ class GriptapeNodes(metaclass=SingletonMeta):
     def EngineIdentityManager(cls) -> EngineIdentityManager:
         return GriptapeNodes.get_instance()._engine_identity_manager
 
+    @classmethod
+    def SyncManager(cls) -> SyncManager:
+        return GriptapeNodes.get_instance()._sync_manager
+
     @classmethod
     def clear_data(cls) -> None:
         # Get canvas

griptape_nodes/retained_mode/managers/library_manager.py

@@ -13,7 +13,6 @@ from importlib.resources import files
 from pathlib import Path
 from typing import TYPE_CHECKING, cast
 
-import uv
 from packaging.requirements import InvalidRequirement, Requirement
 from pydantic import ValidationError
 from rich.align import Align
@@ -96,6 +95,7 @@ from griptape_nodes.retained_mode.managers.library_lifecycle.library_provenance.
 )
 from griptape_nodes.retained_mode.managers.library_lifecycle.library_status import LibraryStatus
 from griptape_nodes.retained_mode.managers.os_manager import OSManager
+from griptape_nodes.utils.uv_utils import find_uv_bin
 from griptape_nodes.utils.version_utils import get_complete_version_string
 
 if TYPE_CHECKING:
@@ -110,21 +110,6 @@ logger = logging.getLogger("griptape_nodes")
 console = Console()
 
 
-def _find_griptape_uv_bin() -> str:
-    """Find the uv binary, checking dedicated Griptape installation first, then system uv.
-
-    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)
-
-    # Fall back to system uv installation
-    return uv.find_uv_bin()
-
-
 class LibraryManager:
     SANDBOX_LIBRARY_NAME = "Sandbox Library"
 
@@ -960,10 +945,12 @@ class LibraryManager:
                     )
                     return RegisterLibraryFromRequirementSpecifierResultFailure()
 
+                uv_path = find_uv_bin()
+
                 logger.info("Installing dependency '%s' with pip in venv at %s", package_name, venv_path)
                 subprocess.run(  # noqa: S603
                     [
-                        _find_griptape_uv_bin(),
+                        uv_path,
                         "pip",
                         "install",
                         request.requirement_specifier,
@@ -1033,9 +1020,10 @@ class LibraryManager:
                 raise RuntimeError(error_message)
 
             try:
+                uv_path = find_uv_bin()
                 logger.info("Creating virtual environment at %s with Python %s", library_venv_path, python_version)
                 subprocess.run(  # noqa: S603
-                    [sys.executable, "-m", "uv", "venv", str(library_venv_path), "--python", python_version],
+                    [uv_path, "venv", str(library_venv_path), "--python", python_version],
                     check=True,
                     capture_output=True,
                     text=True,

griptape_nodes/retained_mode/managers/node_manager.py

@@ -1435,8 +1435,8 @@ class NodeManager:
             result = SetParameterValueResultFailure()
             return result
 
-        # Validate that parameters can be set at all
-        if not parameter.settable:
+        # Validate that parameters can be set at all (note: we want the value to be set during initial setup, but not after)
+        if not parameter.settable and not request.initial_setup:
             details = f"Attempted to set parameter value for '{node_name}.{request.parameter_name}'. Failed because that Parameter was flagged as not settable."
             logger.error(details)
             result = SetParameterValueResultFailure()

griptape_nodes/retained_mode/managers/operation_manager.py

@@ -45,6 +45,7 @@ if TYPE_CHECKING:
         GetNodeMetadataRequest,
         GetNodeResolutionStateRequest,
         ListParametersOnNodeRequest,
+        SetLockNodeStateRequest,
         SetNodeMetadataRequest,
     )
     from griptape_nodes.retained_mode.events.parameter_events import (
@@ -417,6 +418,12 @@ class PayloadConverter:
         """Handle RenameParameterRequest payloads."""
         return f"""cmd.rename_param(node_name="{payload.node_name}",parameter_name="{payload.parameter_name}",new_parameter_name="{payload.new_parameter_name}")"""
 
+    @staticmethod
+    def _handle_SetLockNodeStateRequest(payload: SetLockNodeStateRequest) -> str:
+        """Handle SetLockNodeStateRequest payloads."""
+        node_name_param = f'node_name="{payload.node_name}"' if payload.node_name is not None else "node_name=None"
+        return f"""cmd.set_lock_node_state({node_name_param}, lock={payload.lock})"""
+
     # GENERIC HANDLERS FOR PAYLOADS WITHOUT SPECIFIC HANDLERS
 
 

griptape_nodes/retained_mode/managers/settings.py

@@ -33,6 +33,7 @@ class AppEvents(BaseModel):
             "SingleExecutionStepRequest",
             "SingleNodeStepRequest",
             "ContinueExecutionStepRequest",
+            "SetLockNodeStateRequest",
         ]
     )
 
@@ -95,3 +96,7 @@ class Settings(BaseModel):
     minimum_disk_space_gb_workflows: float = Field(
         default=1.0, description="Minimum disk space in GB required for saving workflows"
     )
+    synced_workflows_directory: str = Field(
+        default="synced_workflows",
+        description="Path to the synced workflows directory, relative to the workspace directory.",
+    )