Handle exceptions like GenerativeAIPromptError in UI (baserow#4368)

Author: dimmur-brw

backend/src/baserow/contrib/database/ws/rows/signals.py

@@ -131,7 +131,7 @@ def rows_ai_values_generation_error(
                 "row_ids": [row.id for row in rows],
                 "error": error_message,
             },
-            getattr(user, "web_socket_id", None),
+            None,
             table_id=table.id,
         )
     )

backend/tests/baserow/contrib/database/ws/test_ws_rows_signals.py

@@ -9,6 +9,7 @@
 from rest_framework import serializers
 from rest_framework.fields import Field
 
+from baserow.contrib.database.rows import signals as row_signals
 from baserow.contrib.database.rows.actions import UpdateRowsActionType
 from baserow.contrib.database.rows.handler import RowHandler
 from baserow.contrib.database.rows.registries import (
@@ -383,3 +384,62 @@ def test_rows_history_updated(
         mock_broadcast_many_channel_group.mock_calls == row_history_many_broadcast_calls
     )
     assert mock_broadcast_channel_group.mock_calls == table_and_row_broadcast_calls
+
+
+@pytest.mark.django_db(transaction=True)
+@patch("baserow.ws.registries.broadcast_to_channel_group")
+def test_rows_ai_values_generation_error(mock_broadcast_to_channel_group, data_fixture):
+    user = data_fixture.create_user()
+    table = data_fixture.create_database_table(user=user)
+    field = data_fixture.create_text_field(table=table)
+    model = table.get_model()
+    row1 = model.objects.create()
+    row2 = model.objects.create()
+
+    row_signals.rows_ai_values_generation_error.send(
+        sender=None,
+        user=user,
+        rows=[row1, row2],
+        field=field,
+        table=table,
+        error_message="Test error message",
+    )
+
+    mock_broadcast_to_channel_group.delay.assert_called_once()
+    args = mock_broadcast_to_channel_group.delay.call_args
+    assert args[0][0] == f"table-{table.id}"
+    assert args[0][1]["type"] == "rows_ai_values_generation_error"
+    assert args[0][1]["table_id"] == table.id
+    assert args[0][1]["field_id"] == field.id
+    assert args[0][1]["row_ids"] == [row1.id, row2.id]
+    assert args[0][1]["error"] == "Test error message"
+    assert args[0][2] is None
+
+
+@pytest.mark.django_db(transaction=True)
+@patch("baserow.ws.registries.broadcast_to_channel_group")
+def test_rows_ai_values_generation_error_with_empty_rows(
+    mock_broadcast_to_channel_group, data_fixture
+):
+    user = data_fixture.create_user()
+    table = data_fixture.create_database_table(user=user)
+    field = data_fixture.create_text_field(table=table)
+
+    row_signals.rows_ai_values_generation_error.send(
+        sender=None,
+        user=user,
+        rows=[],
+        field=field,
+        table=table,
+        error_message="Model not available",
+    )
+
+    mock_broadcast_to_channel_group.delay.assert_called_once()
+    args = mock_broadcast_to_channel_group.delay.call_args
+    assert args[0][0] == f"table-{table.id}"
+    assert args[0][1]["type"] == "rows_ai_values_generation_error"
+    assert args[0][1]["table_id"] == table.id
+    assert args[0][1]["field_id"] == field.id
+    assert args[0][1]["row_ids"] == []
+    assert args[0][1]["error"] == "Model not available"
+    assert args[0][2] is None

changelog/entries/unreleased/bug/4350_handle_exceptions_like_generativeaiprompterror_in_ui.json

@@ -0,0 +1,9 @@
+{
+    "type": "bug",
+    "message": "Handle exceptions like GenerativeAIPromptError in UI",
+    "issue_origin": "github",
+    "issue_number": 4350,
+    "domain": "database",
+    "bullet_points": [],
+    "created_at": "2025-12-01"
+}

web-frontend/modules/database/store/view/grid.js

@@ -738,6 +738,14 @@ export const mutations = {
       Vue.delete(state.pendingFieldOps, key)
     })
   },
+  CLEAR_ALL_PENDING_FIELD_OPERATIONS_FOR_FIELD(state, { fieldId }) {
+    const keysToDelete = Object.keys(state.pendingFieldOps).filter(
+      (key) => state.pendingFieldOps[key][0] === fieldId
+    )
+    keysToDelete.forEach((key) => {
+      Vue.delete(state.pendingFieldOps, key)
+    })
+  },
   UPDATE_ROW_HEIGHT(state, value) {
     state.rowHeight = value
   },
@@ -3471,7 +3479,12 @@ export const actions = {
     commit('SET_PENDING_FIELD_OPERATIONS', { fieldId, rowIds, value })
   },
   AIValuesGenerationError({ commit, dispatch }, { fieldId, rowIds }) {
-    commit('SET_PENDING_FIELD_OPERATIONS', { fieldId, rowIds, value: false })
+    // If rowIds is empty, clear ALL pending operations for this field.
+    if (rowIds.length === 0) {
+      commit('CLEAR_ALL_PENDING_FIELD_OPERATIONS_FOR_FIELD', { fieldId })
+    } else {
+      commit('SET_PENDING_FIELD_OPERATIONS', { fieldId, rowIds, value: false })
+    }
     dispatch(
       'toast/error',
       {