Fix import of unknown ai generative types (baserow#5190)

Author: stribny

changelog/entries/unreleased/bug/5163_ai_field_import_baserow_type_fail.json

@@ -0,0 +1,9 @@
+{
+    "type": "bug",
+    "message": "Fix import of unknown ai generative types",
+    "issue_origin": "github",
+    "issue_number": 5163,
+    "domain": "database",
+    "bullet_points": [],
+    "created_at": "2026-04-13"
+}

premium/backend/src/baserow_premium/fields/field_types.py

@@ -335,10 +335,11 @@ def _validate_field_kwargs(
         self, ai_output_type, ai_type, model_type, ai_file_field_id, workspace=None
     ):
         ai_field_output_registry.get(ai_output_type)
-        ai_type = generative_ai_model_type_registry.get(ai_type)
-        models = ai_type.get_enabled_models(workspace=workspace)
-        if model_type not in models:
-            raise ModelDoesNotBelongToType(model_name=model_type)
+        if ai_type is not None:
+            ai_type = generative_ai_model_type_registry.get(ai_type)
+            models = ai_type.get_enabled_models(workspace=workspace)
+            if model_type not in models:
+                raise ModelDoesNotBelongToType(model_name=model_type)
         if ai_file_field_id is not None and not ai_type.supports_files:
             raise GenerativeAITypeDoesNotSupportFileField()
 
@@ -508,6 +509,14 @@ def import_serialized(
             serialized_values = serialized_values.copy()
             serialized_values.pop("ai_auto_update_user_id", None)
             serialized_values["ai_auto_update"] = False
+
+        ai_type = serialized_values.get("ai_generative_ai_type")
+        if ai_type is not None:
+            try:
+                generative_ai_model_type_registry.get(ai_type)
+            except GenerativeAITypeDoesNotExist:
+                serialized_values["ai_generative_ai_type"] = None
+
         return super().import_serialized(
             table,
             serialized_values,

premium/backend/tests/baserow_premium_tests/fields/test_ai_field_type.py

@@ -1382,6 +1382,39 @@ def test_create_ai_field_auto_doesnt_update_user_if_set(premium_data_fixture):
     assert ai_field.ai_auto_update_user_id == user.id  # not changed
 
 
+@pytest.mark.django_db
+@pytest.mark.field_ai
+def test_import_serialized_ai_field_missing_ai_generative_ai_type(premium_data_fixture):
+    user = premium_data_fixture.create_user()
+    table = premium_data_fixture.create_database_table(user=user)
+    premium_data_fixture.register_fake_generate_ai_type()
+    premium_data_fixture.create_text_field(
+        table=table, order=0, name="text", primary=True
+    )
+    ai_field = premium_data_fixture.create_ai_field(
+        table=table,
+        order=1,
+        name="ai",
+        ai_generative_ai_type="missing",
+        ai_generative_ai_model="test_1",
+        ai_prompt="Tell me a joke",
+    )
+    field_type = field_type_registry.get_by_model(ai_field)
+    serialized = field_type.export_serialized(ai_field)
+
+    imported_field = field_type.import_serialized(
+        table,
+        serialized,
+        ImportExportConfig(include_permission_data=False),
+        id_mapping={},
+        deferred_fk_update_collector=DeferredForeignKeyUpdater(),
+    )
+
+    imported_field = AIField.objects.get(id=imported_field.id)
+    assert imported_field.ai_generative_ai_type is None
+    assert imported_field.ai_generative_ai_model == "test_1"
+
+
 @pytest.mark.django_db
 @pytest.mark.field_ai
 def test_import_serialized_ai_field_with_auto_update_user(premium_data_fixture):