feat(AI Assistant): scaffold automations (baserow#4172)

* kuma scaffolds automations

* Address copilot feedback

* Address feedback

Author: silvestrid

backend/.flake8

@@ -11,6 +11,7 @@ per-file-ignores =
 exclude =
     .git,
     __pycache__,
+    src/baserow/config/settings/local.py,
     src/baserow/core/formula/parser/generated
 
 

backend/src/baserow/contrib/automation/nodes/node_types.py

@@ -245,7 +245,8 @@ def after_create(self, node: CoreRouterActionNode):
         :param node: The router node instance that was just created.
         """
 
-        node.service.edges.create(label=_("Branch"))
+        if not len(node.service.edges.all()):
+            node.service.edges.create(label=_("Branch"))
 
     def prepare_values(
         self,

backend/src/baserow/contrib/automation/workflows/service.py

@@ -54,6 +54,30 @@ def get_workflow(self, user: AbstractUser, workflow_id: int) -> AutomationWorkfl
 
         return workflow
 
+    def list_workflows(
+        self, user: AbstractUser, automation_id: int
+    ) -> List[AutomationWorkflow]:
+        """
+        Lists all the workflows that belong to the given automation.
+
+        :param user: The user requesting the workflows.
+        :param automation_id: The automation to which the workflows belong.
+        :return: A list of AutomationWorkflow instances.
+        """
+
+        automation = AutomationHandler().get_automation(automation_id)
+
+        all_workflows = self.handler.get_workflows(
+            automation, base_queryset=AutomationWorkflow.objects
+        )
+
+        return CoreHandler().filter_queryset(
+            user,
+            ReadAutomationWorkflowOperationType.type,
+            all_workflows,
+            workspace=automation.workspace,
+        )
+
     def create_workflow(
         self,
         user: AbstractUser,

enterprise/backend/src/baserow_enterprise/apps.py

@@ -303,18 +303,20 @@ def ready(self):
         notification_type_registry.register(TwoWaySyncDeactivatedNotificationType())
 
         from baserow_enterprise.assistant.tools import (
-            CreateDatabaseToolType,
+            CreateBuildersToolType,
             CreateFieldsToolType,
             CreateTablesToolType,
             CreateViewFiltersToolType,
             CreateViewsToolType,
+            CreateWorkflowsToolType,
             GenerateDatabaseFormulaToolType,
             GetRowsToolsToolType,
             GetTablesSchemaToolType,
-            ListDatabasesToolType,
+            ListBuildersToolType,
             ListRowsToolType,
             ListTablesToolType,
             ListViewsToolType,
+            ListWorkflowsToolType,
             NavigationToolType,
             SearchDocsToolType,
         )
@@ -325,8 +327,8 @@ def ready(self):
         assistant_tool_registry.register(SearchDocsToolType())
         assistant_tool_registry.register(NavigationToolType())
 
-        assistant_tool_registry.register(ListDatabasesToolType())
-        assistant_tool_registry.register(CreateDatabaseToolType())
+        assistant_tool_registry.register(ListBuildersToolType())
+        assistant_tool_registry.register(CreateBuildersToolType())
         assistant_tool_registry.register(ListTablesToolType())
         assistant_tool_registry.register(CreateTablesToolType())
         assistant_tool_registry.register(GetTablesSchemaToolType())
@@ -338,6 +340,9 @@ def ready(self):
         assistant_tool_registry.register(CreateViewsToolType())
         assistant_tool_registry.register(CreateViewFiltersToolType())
 
+        assistant_tool_registry.register(ListWorkflowsToolType())
+        assistant_tool_registry.register(CreateWorkflowsToolType())
+
         # The signals must always be imported last because they use the registries
         # which need to be filled first.
         import baserow_enterprise.audit_log.signals  # noqa: F

enterprise/backend/src/baserow_enterprise/assistant/assistant.py

@@ -145,6 +145,7 @@ def _init_lm_client(self):
             model=lm_model,
             cache=not settings.DEBUG,
             max_retries=5,
+            max_tokens=32000,
         )
 
     def _init_assistant(self):

enterprise/backend/src/baserow_enterprise/assistant/prompts.py

@@ -38,15 +38,15 @@
 AUTOMATION_BUILDER_CONCEPTS = """
 ### AUTOMATIONS (no-code automation builder)
 
-**Structure**: Automation → Workflows → Triggers + Actions + Routers (Nodes)
+**Structure**: Automation → Workflows → Trigger + Actions + Routers (Nodes)
 
 **Key concepts**:
-• **Triggers**: Events that start automations (e.g., row created/updated, view accessed)
+• **Trigger**: The single event that starts the workflow (e.g., row created/updated/deleted)
 • **Actions**: Tasks performed (e.g., create/update rows, send emails, call webhooks)
 • **Routers**: Conditional logic (if/else, switch) to control flow
 • **Execution**: Runs in the background; monitor via logs
 • **History**: Track runs, successes, failures
-• **Publishing**: Requires domain configuration
+• **Publishing**: Requires at least one configured action
 """
 
 ASSISTANT_SYSTEM_PROMPT = (

enterprise/backend/src/baserow_enterprise/assistant/tools/__init__.py

@@ -1,3 +1,5 @@
+from .automation.tools import *  # noqa: F401, F403
+from .core.tools import *  # noqa: F401, F403
 from .database.tools import *  # noqa: F401, F403
 from .navigation.tools import *  # noqa: F401, F403
 from .search_docs.tools import *  # noqa: F401, F403

enterprise/backend/src/baserow_enterprise/assistant/tools/automation/__init__.py

@@ -0,0 +1,6 @@
+from .tools import CreateWorkflowsToolType, ListWorkflowsToolType
+
+__all__ = [
+    "ListWorkflowsToolType",
+    "CreateWorkflowsToolType",
+]

enterprise/backend/src/baserow_enterprise/assistant/tools/automation/tools.py

@@ -0,0 +1,118 @@
+from typing import TYPE_CHECKING, Any, Callable
+
+from django.contrib.auth.models import AbstractUser
+from django.db import transaction
+from django.utils.translation import gettext as _
+
+from baserow.contrib.automation.workflows.service import AutomationWorkflowService
+from baserow.core.models import Workspace
+from baserow_enterprise.assistant.tools.registries import AssistantToolType
+from baserow_enterprise.assistant.types import WorkflowNavigationType
+
+from . import utils
+from .types import WorkflowCreate
+
+if TYPE_CHECKING:
+    from baserow_enterprise.assistant.assistant import ToolHelpers
+
+
+def get_list_workflows_tool(
+    user: AbstractUser, workspace: Workspace, tool_helpers: "ToolHelpers"
+) -> Callable[[int], dict[str, list[dict]]]:
+    """
+    List all workflows in an automation.
+    """
+
+    def list_workflows(automation_id: int) -> dict[str, Any]:
+        """
+        List all workflows in an automation application.
+
+        :param automation_id: The ID of the automation application
+        :return: Dictionary with workflows list
+        """
+
+        nonlocal user, workspace, tool_helpers
+
+        tool_helpers.update_status(_("Listing workflows..."))
+
+        automation = utils.get_automation(automation_id, user, workspace)
+        workflows = AutomationWorkflowService().list_workflows(user, automation.id)
+
+        return {
+            "workflows": [
+                {"id": w.id, "name": w.name, "state": w.state} for w in workflows
+            ]
+        }
+
+    return list_workflows
+
+
+def get_create_workflows_tool(
+    user: AbstractUser, workspace: Workspace, tool_helpers: "ToolHelpers"
+) -> Callable[[int, list[WorkflowCreate]], dict[str, list[dict]]]:
+    """
+    Create new workflows.
+    """
+
+    def create_workflows(
+        automation_id: int, workflows: list[WorkflowCreate]
+    ) -> dict[str, Any]:
+        """
+        Create one or more workflows in an automation.
+
+        :param automation_id: The automation application ID
+        :param workflows: List of workflows to create
+        :return: Dictionary with created workflows
+        """
+
+        nonlocal user, workspace, tool_helpers
+
+        tool_helpers.update_status(_("Creating workflows..."))
+
+        created = []
+
+        automation = utils.get_automation(automation_id, user, workspace)
+        for wf in workflows:
+            with transaction.atomic():
+                orm_workflow = utils.create_workflow(user, automation, wf)
+                created.append(
+                    {
+                        "id": orm_workflow.id,
+                        "name": orm_workflow.name,
+                        "state": orm_workflow.state,
+                    }
+                )
+        # Navigate to the last created workflow
+        tool_helpers.navigate_to(
+            WorkflowNavigationType(
+                type="automation-workflow",
+                automation_id=automation.id,
+                workflow_id=orm_workflow.id,
+                workflow_name=orm_workflow.name,
+            )
+        )
+
+        return {"created_workflows": created}
+
+    return create_workflows
+
+
+# ============================================================================
+# TOOL TYPE REGISTRY
+# ============================================================================
+
+
+class ListWorkflowsToolType(AssistantToolType):
+    type = "list_workflows"
+
+    @classmethod
+    def get_tool(cls, user, workspace, tool_helpers):
+        return get_list_workflows_tool(user, workspace, tool_helpers)
+
+
+class CreateWorkflowsToolType(AssistantToolType):
+    type = "create_workflows"
+
+    @classmethod
+    def get_tool(cls, user, workspace, tool_helpers):
+        return get_create_workflows_tool(user, workspace, tool_helpers)

enterprise/backend/src/baserow_enterprise/assistant/tools/automation/types/__init__.py

@@ -0,0 +1,24 @@
+from .node import (
+    AiAgentNodeCreate,
+    CreateRowActionCreate,
+    DeleteRowActionCreate,
+    NodeBase,
+    RouterNodeCreate,
+    SendEmailActionCreate,
+    TriggerNodeCreate,
+    UpdateRowActionCreate,
+)
+from .workflow import WorkflowCreate, WorkflowItem
+
+__all__ = [
+    "WorkflowCreate",
+    "WorkflowItem",
+    "NodeBase",
+    "RouterNodeCreate",
+    "CreateRowActionCreate",
+    "UpdateRowActionCreate",
+    "DeleteRowActionCreate",
+    "SendEmailActionCreate",
+    "AiAgentNodeCreate",
+    "TriggerNodeCreate",
+]