Install Wagtail AI

Author: laymonage

bakerydemo/settings/base.py

@@ -44,6 +44,7 @@
     "bakerydemo.locations",
     "bakerydemo.recipes",
     "bakerydemo.search",
+    "wagtail_ai",
     "wagtail.embeds",
     "wagtail.sites",
     "wagtail.users",
@@ -222,6 +223,40 @@
 
 ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "changeme")
 
+# Wagtail AI settings
+# https://wagtail-ai.readthedocs.io
+
+WAGTAIL_AI = {
+    "BACKENDS": {
+        "default": {
+            "CLASS": "wagtail_ai.ai.llm.LLMBackend",
+            "CONFIG": {
+                # Model ID recognizable by the "LLM" library.
+                "MODEL_ID": os.environ.get(
+                    "WAGTAIL_AI_DEFAULT_MODEL_ID", "gpt-4.1-mini"
+                ),
+                "TOKEN_LIMIT": int(
+                    os.environ.get("WAGTAIL_AI_DEFAULT_TOKEN_LIMIT", 4096)
+                ),
+            },
+        },
+        "vision": {
+            "CLASS": "wagtail_ai.ai.openai.OpenAIBackend",
+            "CONFIG": {
+                "MODEL_ID": os.environ.get(
+                    "WAGTAIL_AI_VISION_MODEL_ID", "gpt-4.1-mini"
+                ),
+                "TOKEN_LIMIT": int(
+                    os.environ.get("WAGTAIL_AI_VISION_TOKEN_LIMIT", 300)
+                ),
+            },
+        },
+    },
+    "IMAGE_DESCRIPTION_BACKEND": "vision",
+}
+WAGTAILIMAGES_IMAGE_FORM_BASE = "wagtail_ai.forms.DescribeImageForm"
+
+
 # Content Security policy settings
 # http://django-csp.readthedocs.io/en/latest/configuration.html
 

requirements/base.txt

@@ -1,6 +1,7 @@
 Django>=5.2,<5.3
 django-dotenv==1.4.2
 wagtail>=7.1,<7.2
+wagtail-ai
 wagtail-font-awesome-svg>=1,<2
 django-debug-toolbar>=4.2,<5
 django-extensions==3.2.3