Skip to content

wagtail/wagtail-ai

Repository files navigation

wagtail-ai

Wagtail AI

Get help with your content using AI superpowers.

License: MIT PyPI version ai CI

Wagtail AI integrates Wagtail with AI's APIs (think ChatGPT) to help you write and correct your content.

Right now, it can:

  • Finish what you've started - write some text and tell Wagtail AI to finish it off for you
  • Correct your spelling/grammar
  • Let you add your own custom prompts
  • Automatically generate alt-tags for your uploaded images
  • Work with multiple LLM providers including local models, OpenAI, Mistral, Claude and many others

Demos

Rich-text integration

WagtailAIDemo.mp4

Alt-text generation

Screen.Recording.2024-03-08.at.17.07.18.mov

Requirements & Costs

Wagtail AI supports many different LLMs, with OpenAI models available by default. To use these, you'll need an OpenAI account and an API key. There'll also be some cost involved.

For the OpenAI API used here (gpt-3.5-turbo), the cost is

  • $0.0005 per 1000 tokens for input tokens (prompt)
  • $0.0015 per 1000 tokens for output tokens (answer)

Here is an estimated cost breakdown for the correction prompt on a 1000-word paragraph.

We assume that:

  • Prompt is 30 words and the existing paragraph is 1000 words (Input)
  • Each word is 1.3 tokens (Tokens multiplier)
  • We get back 1000 words back (Output)

Then:

  • Input tokens : (35 + 1000) x 1.3 = 1345.5 tokens.
  • Output tokens : 1000 x 1.3 = 1300
  • Input tokens cost : 1345.5 / 1000 * $0.0005 = $0.00067275
  • Output tokens cost : 1300 / 1000 * $0.0015 = $0.00195
  • Total cost : $0.00262275

Links

Supported Versions

  • Wagtail 5.2
  • Django 4.2
  • Python 3.11