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
  • Work with multiple LLM providers including local models, OpenAI, Mistral, Claude and many others
WagtailAIDemo.mp4

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