Added AI agent node (baserow#4154)

* Added AI integration node

* Fix AIAgent node

* Renaming AiAgentWorkflowAction->AIAgentWorkflowAction. Renaming AiAgentWorkflowActionType->AIAgentWorkflowActionType. Tweaking the changelog so it's about builder.

* Fix tests

---------

Co-authored-by: Jeremie Pardou <571533+jrmi@users.noreply.github.com>
Co-authored-by: peter_baserow <peter@baserow.io>

Author: 3 people

backend/src/baserow/contrib/automation/apps.py

@@ -20,6 +20,7 @@ def ready(self):
             UpdateAutomationNodeActionType,
         )
         from baserow.contrib.automation.nodes.node_types import (
+            AIAgentActionNodeType,
             CoreHttpRequestNodeType,
             CoreHTTPTriggerNodeType,
             CoreIteratorNodeType,
@@ -167,6 +168,7 @@ def ready(self):
             )
             automation_node_type_registry.register(CorePeriodicTriggerNodeType())
             automation_node_type_registry.register(CoreHTTPTriggerNodeType())
+            automation_node_type_registry.register(AIAgentActionNodeType())
 
             from baserow.core.trash.registries import trash_operation_type_registry
 

backend/src/baserow/contrib/automation/migrations/0022_aiagentactionnode.py

@@ -0,0 +1,36 @@
+# Generated by Django 5.0.14 on 2025-11-03 16:06
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        (
+            "automation",
+            "0021_coreiteratoractionnode_alter_automationnode_options_and_more",
+        ),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="AIAgentActionNode",
+            fields=[
+                (
+                    "automationnode_ptr",
+                    models.OneToOneField(
+                        auto_created=True,
+                        on_delete=django.db.models.deletion.CASCADE,
+                        parent_link=True,
+                        primary_key=True,
+                        serialize=False,
+                        to="automation.automationnode",
+                    ),
+                ),
+            ],
+            options={
+                "abstract": False,
+            },
+            bases=("automation.automationnode",),
+        ),
+    ]

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

@@ -224,3 +224,7 @@ class CoreRouterActionNode(AutomationActionNode):
 
 class CoreIteratorActionNode(AutomationActionNode):
     ...
+
+
+class AIAgentActionNode(AutomationActionNode):
+    ...

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

@@ -16,6 +16,7 @@
     AutomationNodeTriggerMustBeFirstNode,
 )
 from baserow.contrib.automation.nodes.models import (
+    AIAgentActionNode,
     AutomationNode,
     AutomationTriggerNode,
     CoreHTTPRequestActionNode,
@@ -38,6 +39,7 @@
 from baserow.contrib.automation.nodes.types import NodePositionType
 from baserow.contrib.automation.workflows.constants import WorkflowState
 from baserow.contrib.automation.workflows.models import AutomationWorkflow
+from baserow.contrib.integrations.ai.service_types import AIAgentServiceType
 from baserow.contrib.integrations.core.service_types import (
     CoreHTTPRequestServiceType,
     CoreHTTPTriggerServiceType,
@@ -172,6 +174,12 @@ class CoreSMTPEmailNodeType(AutomationNodeActionNodeType):
     service_type = CoreSMTPEmailServiceType.type
 
 
+class AIAgentActionNodeType(AutomationNodeActionNodeType):
+    type = "ai_agent"
+    model_class = AIAgentActionNode
+    service_type = AIAgentServiceType.type
+
+
 class CoreRouterActionNodeType(AutomationNodeActionNodeType):
     type = "router"
     model_class = CoreRouterActionNode

backend/src/baserow/contrib/builder/apps.py

@@ -277,6 +277,7 @@ def ready(self):
 
         from .workflow_actions.registries import builder_workflow_action_type_registry
         from .workflow_actions.workflow_action_types import (
+            AIAgentWorkflowActionType,
             CoreHttpRequestActionType,
             CoreSMTPEmailActionType,
             CreateRowWorkflowActionType,
@@ -299,6 +300,7 @@ def ready(self):
         )
         builder_workflow_action_type_registry.register(CoreHttpRequestActionType())
         builder_workflow_action_type_registry.register(CoreSMTPEmailActionType())
+        builder_workflow_action_type_registry.register(AIAgentWorkflowActionType())
 
         from .elements.collection_field_types import (
             BooleanCollectionFieldType,

backend/src/baserow/contrib/builder/domains/handler.py

@@ -251,6 +251,7 @@ def publish(self, domain: Domain, progress: Progress | None = None):
             include_permission_data=True,
             reduce_disk_space_usage=False,
             exclude_sensitive_data=False,
+            is_publishing=True,
         )
 
         default_storage = get_default_storage()

backend/src/baserow/contrib/builder/migrations/0065_aiagentworkflowaction.py

@@ -0,0 +1,42 @@
+# Generated by Django 5.0.14 on 2025-11-04 14:46
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("builder", "0064_migrate_to_formula_field_objects"),
+        ("core", "0106_schemaoperation"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="AIAgentWorkflowAction",
+            fields=[
+                (
+                    "builderworkflowaction_ptr",
+                    models.OneToOneField(
+                        auto_created=True,
+                        on_delete=django.db.models.deletion.CASCADE,
+                        parent_link=True,
+                        primary_key=True,
+                        serialize=False,
+                        to="builder.builderworkflowaction",
+                    ),
+                ),
+                (
+                    "service",
+                    models.ForeignKey(
+                        help_text="The service which this action is associated with.",
+                        on_delete=django.db.models.deletion.CASCADE,
+                        to="core.service",
+                    ),
+                ),
+            ],
+            options={
+                "abstract": False,
+            },
+            bases=("builder.builderworkflowaction",),
+        ),
+    ]

backend/src/baserow/contrib/builder/workflow_actions/models.py

@@ -121,3 +121,7 @@ class CoreHTTPRequestWorkflowAction(BuilderWorkflowServiceAction):
 
 class CoreSMTPEmailWorkflowAction(BuilderWorkflowServiceAction):
     ...
+
+
+class AIAgentWorkflowAction(BuilderWorkflowServiceAction):
+    ...

backend/src/baserow/contrib/builder/workflow_actions/workflow_action_types.py

@@ -16,6 +16,7 @@
 from baserow.contrib.builder.elements.element_types import NavigationElementManager
 from baserow.contrib.builder.formula_importer import import_formula
 from baserow.contrib.builder.workflow_actions.models import (
+    AIAgentWorkflowAction,
     CoreHTTPRequestWorkflowAction,
     CoreSMTPEmailWorkflowAction,
     LocalBaserowCreateRowWorkflowAction,
@@ -30,6 +31,7 @@
     BuilderWorkflowActionType,
 )
 from baserow.contrib.builder.workflow_actions.types import BuilderWorkflowActionDict
+from baserow.contrib.integrations.ai.service_types import AIAgentServiceType
 from baserow.contrib.integrations.core.service_types import (
     CoreHTTPRequestServiceType,
     CoreSMTPEmailServiceType,
@@ -477,3 +479,13 @@ class CoreSMTPEmailActionType(BuilderWorkflowServiceActionType):
     def get_pytest_params(self, pytest_data_fixture) -> Dict[str, int]:
         service = pytest_data_fixture.create_core_smtp_email_service()
         return {"service": service}
+
+
+class AIAgentWorkflowActionType(BuilderWorkflowServiceActionType):
+    type = "ai_agent"
+    model_class = AIAgentWorkflowAction
+    service_type = AIAgentServiceType.type
+
+    def get_pytest_params(self, pytest_data_fixture) -> Dict[str, int]:
+        service = pytest_data_fixture.create_ai_agent_service()
+        return {"service": service}