├── .eslintrc
├── .gitignore
├── modules
    └── @apostrophecms
    │   ├── ai-helper-image
    │       ├── public
    │       │   └── placeholder.jpg
    │       ├── ui
    │       │   └── apos
    │       │   │   └── components
    │       │   │       ├── AposAiHelperImageEditor.vue
    │       │   │       └── AposAiHelperImageManager.vue
    │       └── index.js
    │   └── ai-helper-rich-text-widget
    │       ├── ui
    │           └── apos
    │           │   └── components
    │           │       └── AposAiHelperTextDialog.vue
    │       └── index.js
├── i18n
    └── aposAiHelper
    │   ├── en.json
    │   ├── pt-BR.json
    │   ├── es.json
    │   ├── sk.json
    │   ├── it.json
    │   ├── fr.json
    │   └── de.json
├── package.json
├── LICENSE.md
├── index.js
├── CHANGELOG.md
└── README.md


/.eslintrc:
--------------------------------------------------------------------------------
1 | { "extends": "apostrophe" }
2 | 


--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
1 | node_modules
2 | npm-debug.log
3 | .DS_Store
4 | package-lock.json


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-image/public/placeholder.jpg:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/apostrophecms/ai-helper/main/modules/@apostrophecms/ai-helper-image/public/placeholder.jpg


--------------------------------------------------------------------------------
/i18n/aposAiHelper/en.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Generate Image",
 3 |   "generateTextLabel": "Generate Text",
 4 |   "generateTextDescription": "Generate text with AI",
 5 |   "imagePlaceholderText": "Describe the image you want to create.",
 6 |   "textPromptLabel": "Enter a prompt to generate text. Example: \"100 words about cheese, with subheadings\"",
 7 |   "variations": "Variations",
 8 |   "select": "Select",
 9 |   "delete": "Delete",
10 |   "generateTextAction": "Generate",
11 |   "rateLimitExceeded": "Rate limit exceeded, try again later",
12 |   "invalidRequest": "Invalid request, please review OpenAI's rules",
13 |   "utilityOperationLabel": "Generate an image with AI",
14 |   "errorMessage": "An error occurred."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/pt-BR.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Gerar Imagem",
 3 |   "generateTextLabel": "Gerar Texto",
 4 |   "generateTextDescription": "Gerar texto com IA",
 5 |   "imagePlaceholderText": "Descreva a imagem que você deseja criar.",
 6 |   "textPromptLabel": "Digite um prompt para gerar texto. Exemplo: \"100 palavras sobre queijo, no estilo de Anthony Bourdain, com subtítulos\"",
 7 |   "variations": "Variações",
 8 |   "select": "Selecionar",
 9 |   "delete": "Excluir",
10 |   "generateTextAction": "Gerar",
11 |   "rateLimitExceeded": "Limite de taxa excedido, tente novamente mais tarde",
12 |   "invalidRequest": "Pedido inválido, por favor revise as regras da OpenAI",
13 |   "utilityOperationLabel": "Gerar uma imagem com IA",
14 |   "errorMessage": "Ocorreu um erro."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/es.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Generar Imagen",
 3 |   "generateTextLabel": "Generar Texto",
 4 |   "generateTextDescription": "Generar texto con IA",
 5 |   "imagePlaceholderText": "Describe la imagen que deseas crear.",
 6 |   "textPromptLabel": "Introduce un aviso para generar texto. Ejemplo: \"100 palabras sobre el queso, al estilo de Anthony Bourdain, con subtítulos\"",
 7 |   "variations": "Variaciones",
 8 |   "select": "Seleccionar",
 9 |   "delete": "Eliminar",
10 |   "generateTextAction": "Generar",
11 |   "rateLimitExceeded": "Límite de tasa excedido, intenta de nuevo más tarde",
12 |   "invalidRequest": "Solicitud inválida, revisa las reglas de OpenAI",
13 |   "utilityOperationLabel": "Generar una imagen con IA",
14 |   "errorMessage": "Ocurrió un error."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/sk.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Vygenerovať obrázok",
 3 |   "generateTextLabel": "Vygenerovať text",
 4 |   "generateTextDescription": "Vygenerovať text s AI",
 5 |   "imagePlaceholderText": "Opíšte obrázok, ktorý chcete vytvoriť.",
 6 |   "textPromptLabel": "Zadajte výzvu na vygenerovanie textu. Príklad: \"100 slov o syre, v štýle Anthony Bourdaina, so subnadpismi\"",
 7 |   "variations": "Variácie",
 8 |   "select": "Vybrať",
 9 |   "delete": "Odstrániť",
10 |   "generateTextAction": "Vygenerovať",
11 |   "rateLimitExceeded": "Prekročený limit požiadaviek, skúste to znova neskôr",
12 |   "invalidRequest": "Neplatná požiadavka, prosím skontrolujte pravidlá OpenAI",
13 |   "utilityOperationLabel": "Vygenerovať obrázok s AI",
14 |   "errorMessage": "Vyskytla sa chyba."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/it.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Genera Immagine",
 3 |   "generateTextLabel": "Genera Testo",
 4 |   "generateTextDescription": "Genera testo con AI",
 5 |   "imagePlaceholderText": "Descrivi l'immagine che vuoi creare.",
 6 |   "textPromptLabel": "Inserisci un prompt per generare testo. Esempio: \"100 parole sul formaggio, nello stile di Anthony Bourdain, con sottotitoli\"",
 7 |   "variations": "Varianti",
 8 |   "select": "Seleziona",
 9 |   "delete": "Elimina",
10 |   "generateTextAction": "Genera",
11 |   "rateLimitExceeded": "Limite di richieste superato, riprova più tardi",
12 |   "invalidRequest": "Richiesta non valida, per favore rivedi le regole di OpenAI",
13 |   "utilityOperationLabel": "Genera un'immagine con AI",
14 |   "errorMessage": "Si è verificato un errore."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/fr.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Générer une image",
 3 |   "generateTextLabel": "Générer du texte",
 4 |   "generateTextDescription": "Générer du texte avec l'IA",
 5 |   "imagePlaceholderText": "Décrivez l'image que vous souhaitez créer.",
 6 |   "textPromptLabel": "Entrez un prompt pour générer du texte. Exemple : \"100 mots sur le fromage, dans le style d'Anthony Bourdain, avec des sous-titres\"",
 7 |   "variations": "Variations",
 8 |   "select": "Sélectionner",
 9 |   "delete": "Supprimer",
10 |   "generateTextAction": "Générer",
11 |   "rateLimitExceeded": "Limite de taux dépassée, réessayez plus tard",
12 |   "invalidRequest": "Demande invalide, veuillez revoir les règles d'OpenAI",
13 |   "utilityOperationLabel": "Générer une image avec l'IA",
14 |   "errorMessage": "Une erreur s'est produite."
15 | }
16 | 


--------------------------------------------------------------------------------
/i18n/aposAiHelper/de.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "generateImage": "Bild generieren",
 3 |   "generateTextLabel": "Text generieren",
 4 |   "generateTextDescription": "Text mit KI generieren",
 5 |   "imagePlaceholderText": "Beschreiben Sie das Bild, das Sie erstellen möchten.",
 6 |   "textPromptLabel": "Geben Sie einen Prompt ein, um Text zu generieren. Beispiel: \"100 Wörter über Käse, im Stil von Anthony Bourdain, mit Unterüberschriften\"",
 7 |   "variations": "Varianten",
 8 |   "select": "Auswählen",
 9 |   "delete": "Löschen",
10 |   "generateTextAction": "Generieren",
11 |   "rateLimitExceeded": "Die Ratebegrenzung wurde überschritten, versuchen Sie es später erneut",
12 |   "invalidRequest": "Ungültige Anfrage, bitte überprüfen Sie die Regeln von OpenAI",
13 |   "utilityOperationLabel": "Ein Bild mit KI generieren",
14 |   "errorMessage": "Ein Fehler ist aufgetreten."
15 | }
16 | 


--------------------------------------------------------------------------------
/package.json:
--------------------------------------------------------------------------------
 1 | {
 2 |   "name": "@apostrophecms/ai-helper",
 3 |   "version": "1.0.0-beta.10",
 4 |   "description": "AI helpers for content creation",
 5 |   "main": "index.js",
 6 |   "scripts": {
 7 |     "test": "npm run lint",
 8 |     "lint": "eslint --ext .js,.vue ."
 9 |   },
10 |   "repository": {
11 |     "type": "git",
12 |     "url": "git+https://github.com/apostrophecms/ai-helper.git"
13 |   },
14 |   "keywords": [
15 |     "ApostropheCMS",
16 |     "apostrophe"
17 |   ],
18 |   "author": "Apostrophe Technologies, Inc.",
19 |   "license": "MIT",
20 |   "bugs": {
21 |     "url": "https://github.com/apostrophecms/ai-helper/issues"
22 |   },
23 |   "homepage": "https://github.com/apostrophecms/ai-helper#readme",
24 |   "devDependencies": {
25 |     "eslint-config-apostrophe": "^5.0.0"
26 |   },
27 |   "dependencies": {
28 |     "@paralleldrive/cuid2": "^3.0.4",
29 |     "form-data": "^4.0.0",
30 |     "marked": "^4.3.0",
31 |     "node-fetch": "^2.6.9",
32 |     "sharp": "^0.30.7"
33 |   }
34 | }
35 | 


--------------------------------------------------------------------------------
/LICENSE.md:
--------------------------------------------------------------------------------
1 | Copyright (c) 2023 Apostrophe Technologies, Inc.
2 | 
3 | Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
4 | 
5 | The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
6 | 
7 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
8 | 


--------------------------------------------------------------------------------
/index.js:
--------------------------------------------------------------------------------
 1 | const fs = require('fs');
 2 | const path = require('path');
 3 | 
 4 | module.exports = {
 5 |   init() {
 6 |     if (!(process.env.APOS_OPENAI_KEY || process.env.APOS_AI_HELPER_MOCK)) {
 7 |       // We do not document the mock because it is for internal testing
 8 |       throw new Error('APOS_OPENAI_KEY must be set in your environment');
 9 |     }
10 |   },
11 |   i18n: {
12 |     aposAiHelper: {
13 |       browser: true
14 |     }
15 |   },
16 |   bundle: {
17 |     directory: 'modules',
18 |     modules: getBundleModuleNames()
19 |   },
20 |   options: {
21 |     textModel: 'gpt-5.1',
22 |     textMaxTokens: 1000,
23 |     imageModel: 'gpt-image-1-mini'
24 |   },
25 |   methods(self) {
26 |     return {
27 |       checkPermissions(req) {
28 |         // If the user cannot edit at least one content type, they have
29 |         // no business talking to the AI
30 |         if (!Object.keys(self.apos.modules).some(type => self.apos.permission.can(req, 'edit', type))) {
31 |           throw self.apos.error('forbidden');
32 |         }
33 |       }
34 |     };
35 |   }
36 | };
37 | 
38 | function getBundleModuleNames() {
39 |   const source = path.join(__dirname, './modules/@apostrophecms');
40 |   return fs
41 |     .readdirSync(source, { withFileTypes: true })
42 |     .filter(dirent => dirent.isDirectory())
43 |     .map(dirent => `@apostrophecms/${dirent.name}`);
44 | }
45 | 


--------------------------------------------------------------------------------
/CHANGELOG.md:
--------------------------------------------------------------------------------
 1 | # Changelog
 2 | 
 3 | ## 1.0.0-beta.10 (2025-11-18)
 4 | 
 5 | * Text generation is fully compatible with the latest versions of Apostrophe 4.x.
 6 | * Default models modernized to `gpt-5.1` and `gpt-image-1-mini`.
 7 | * Compatible with current OpenAI API.
 8 | * Because newer models take much longer to generate each image and also generate much better images, the image generator now generates one image or variant image at a time.
 9 | * Bumps `eslint-config-apostrophe` to `5`, fixes linter errors, removes unused dependencies and modernizes dependencies.
10 | 
11 | ## 1.0.0-beta.9 (2024-09-05)
12 | 
13 | * Add AI and community-reviewed translation strings
14 | 
15 | ## 1.0.0-beta.8 (2024-08-09)
16 | 
17 | * Security: Apostrophe now checks to make sure the current user can edit at least one type of content on the site before making calls to the OpenAI APIs. Upgrading is recommended.
18 | * Updated to use the gpt4o text model by default.
19 | * Image model can be explicitly specified and defaults to `dall-e-2`. Note that `dall-e-3` does not support variations in the same way, so the current UI will have to change if that feature is not added upstream.
20 | 
21 | 
22 | ## 1.0.0-beta.7 (2024-07-10)
23 | 
24 | * Add missing UI translation keys.
25 | 
26 | ## 1.0.0-beta.6 (2024-02-21)
27 | 
28 | * Check that the current user `canCreate` images before showing the `Generate an image with AI` button.
29 | 
30 | ## 1.0.0-beta.5 (2023-09-20)
31 | 
32 | * `text-davinci-003` model was deprecated, update to the `gpt-3.5-turbo-instruct` model as a new default
33 | * Document how to override the default text generation model and token limit (subject to GPT's own limits)
34 | 
35 | ## 1.0.0-beta.4 (2023-09-14)
36 | 
37 | Just bumping the `latest` tag, no changes.
38 | 
39 | ## 1.0.0-beta.3 (2023-08-03)
40 | 
41 | Fixed a bug in the "variants" feature caused by premature PNG to JPEG
42 | conversion. This now happens only when inserting into the media library.
43 | 
44 | ## 1.0.0-beta.2 (2023-05-26)
45 | 
46 | Convert imported PNGs to JPEGs as this is more appropriate to web
47 | delivery of AI-generated, often realistic images. Reduces image
48 | download time to the browser by at least 80%.
49 | 
50 | ## 1.0.0-beta (2023-04-27)
51 | 
52 | Initial beta release.
53 | 


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-image/ui/apos/components/AposAiHelperImageEditor.vue:
--------------------------------------------------------------------------------
  1 | <template>
  2 |   <AposModal
  3 |     class="apos-ai-helper-image-editor"
  4 |     :modal="modal"
  5 |     modal-title="aposAiHelper:generateImage"
  6 |     @inactive="modal.active = false"
  7 |     @show-modal="modal.showModal = true"
  8 |     @esc="close"
  9 |     @no-modal="$emit('safe-close')"
 10 |   >
 11 |     <template #primaryControls>
 12 |       <AposButton
 13 |         type="default"
 14 |         label="apostrophe:close"
 15 |         @click="close"
 16 |       />
 17 |     </template>
 18 |     <template #main>
 19 |       <AposModalBody>
 20 |         <template #bodyMain>
 21 |           <div class="apos-ai-helper-form">
 22 |             <img
 23 |               class="apos-ai-helper-image"
 24 |               :src="image.url"
 25 |             >
 26 |             <div class="apos-ai-helper-image-buttons">
 27 |               <AposButton
 28 |                 :disabled="image.accepted"
 29 |                 icon="plus-icon"
 30 |                 :label="$t('aposAiHelper:select')"
 31 |                 type="primary"
 32 |                 @click.prevent="action('save')"
 33 |               />
 34 |               <AposButton
 35 |                 icon="group-icon"
 36 |                 :label="$t('aposAiHelper:variations')"
 37 |                 @click.prevent="action('variations')"
 38 |               />
 39 |               <AposButton
 40 |                 icon="delete-icon"
 41 |                 :label="$t('aposAiHelper:delete')"
 42 |                 type="danger"
 43 |                 @click.prevent="action('delete')"
 44 |               />
 45 |             </div>
 46 |             <p v-if="error">
 47 |               An error occurred.
 48 |             </p>
 49 |           </div>
 50 |         </template>
 51 |       </AposModalBody>
 52 |     </template>
 53 |   </AposModal>
 54 | </template>
 55 | 
 56 | <script>
 57 | export default {
 58 |   props: {
 59 |     image: {
 60 |       type: Object,
 61 |       required: true
 62 |     }
 63 |   },
 64 |   emits: [ 'modal-result', 'safe-close' ],
 65 |   data() {
 66 |     return {
 67 |       modal: {
 68 |         active: false,
 69 |         type: 'overlay',
 70 |         showModal: true
 71 |       },
 72 |       error: false
 73 |     };
 74 |   },
 75 |   mounted() {
 76 |     this.modal.active = true;
 77 |     const expireMs = new Date(this.image.createdAt).getTime() + 1000 * 60 * 60;
 78 |     const nowMs = Date.now();
 79 |     const timeout = expireMs - nowMs;
 80 |     this.expireTimeout = setTimeout(this.expire, Math.max(timeout, 0));
 81 |   },
 82 |   unmounted() {
 83 |     clearTimeout(this.expireTimeout);
 84 |   },
 85 |   methods: {
 86 |     expire() {
 87 |       this.modal.showModal = false;
 88 |     },
 89 |     close() {
 90 |       this.modal.showModal = false;
 91 |     },
 92 |     action(action) {
 93 |       this.modal.showModal = false;
 94 |       this.$emit('modal-result', {
 95 |         action
 96 |       });
 97 |     }
 98 |   }
 99 | };
100 | </script>
101 | 
102 | <style lang="scss" scoped>
103 | // Prevent scrollbar and ensure image shrinks to box rather than growing
104 | // in proportion to 100% width
105 | :deep(apos-modal__main), :deep(.apos-modal__body-inner), :deep(.apos-modal__body-main) {
106 |   height: 100%;
107 |   min-height: 0;
108 | }
109 | .apos-ai-helper-form {
110 |   display: flex;
111 |   flex-direction: column;
112 |   height: 100%;
113 |   gap: 32px;
114 | }
115 | .apos-ai-helper-image {
116 |   object-fit: contain;
117 |   min-height: 0;
118 | }
119 | .apos-ai-helper-image-buttons {
120 |   display: flex;
121 |   flex-direction: row;
122 |   justify-content: center;
123 |   gap: 32px;
124 | }
125 | </style>
126 | 


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-rich-text-widget/ui/apos/components/AposAiHelperTextDialog.vue:
--------------------------------------------------------------------------------
  1 | <template>
  2 |   <div
  3 |     class="apos-popover apos-ai-helper-text__dialog"
  4 |     x-placement="bottom"
  5 |     :class="{
  6 |       'apos-is-triggered': true,
  7 |       'apos-has-selection': true
  8 |     }"
  9 |   >
 10 |     <div class="apos-ai-helper-form">
 11 |       <p>
 12 |         {{ $t('aposAiHelper:textPromptLabel') }}
 13 |       </p>
 14 |       <textarea v-model="prompt" />
 15 |       <p v-if="error">
 16 |         {{ $t('aposAiHelper:errorMessage') }}
 17 |       </p>
 18 |     </div>
 19 |     <footer class="apos-ai-helper-text__footer">
 20 |       <AposButton
 21 |         type="default"
 22 |         label="apostrophe:cancel"
 23 |         :modifiers="formModifiers"
 24 |         @click="cancel"
 25 |       />
 26 |       <AposButton
 27 |         type="primary"
 28 |         label="aposAiHelper:generateTextAction"
 29 |         :disabled="!prompt.length"
 30 |         :modifiers="formModifiers"
 31 |         @click="save"
 32 |       />
 33 |     </footer>
 34 |   </div>
 35 | </template>
 36 | 
 37 | <script>
 38 | export default {
 39 |   name: 'AposAiHelperTextDialog',
 40 |   props: {
 41 |     editor: {
 42 |       type: Object,
 43 |       required: true
 44 |     },
 45 |     options: {
 46 |       type: Object,
 47 |       required: true
 48 |     }
 49 |   },
 50 |   emits: [ 'close', 'before-commands' ],
 51 |   data() {
 52 |     return {
 53 |       prompt: '',
 54 |       error: false,
 55 |       formModifiers: [ 'small', 'margin-micro' ]
 56 |     };
 57 |   },
 58 |   mounted() {
 59 |     window.addEventListener('keydown', this.keyboardHandler);
 60 |   },
 61 |   unmounted() {
 62 |     window.removeEventListener('keydown', this.keyboardHandler);
 63 |   },
 64 |   methods: {
 65 |     cancel() {
 66 |       this.$emit('close');
 67 |     },
 68 |     done() {
 69 |       this.$emit('close');
 70 |     },
 71 |     async save() {
 72 |       this.error = false;
 73 |       try {
 74 |         const headingLevels = (this.options.styles || []).filter(style => style.tag.match(/^h\d$/)).map(style => parseInt(style.tag.replace(/h/i, '')));
 75 |         const result = await apos.http.post(`${getOptions().action}/ai-helper`, {
 76 |           body: {
 77 |             prompt: this.prompt,
 78 |             headingLevels
 79 |           },
 80 |           busy: true
 81 |         });
 82 |         this.$emit('before-commands');
 83 |         // newlines shouldn't matter but they do to tiptap, so get rid of them
 84 |         const html = result.html.replace(/>\n+</g, '><');
 85 |         this.editor.commands.insertContent(html);
 86 |         this.done();
 87 |       } catch (e) {
 88 |         this.error = true;
 89 |       }
 90 |     },
 91 |     keyboardHandler(e) {
 92 |       if (e.keyCode === 27) {
 93 |         this.cancel();
 94 |       }
 95 |     }
 96 |   }
 97 | };
 98 | 
 99 | function getOptions() {
100 |   return apos.modules['@apostrophecms/rich-text-widget'];
101 | }
102 | </script>
103 | 
104 | <style lang="scss" scoped>
105 |   .apos-ai-helper-text__dialog {
106 |     width: 500px;
107 |   }
108 | 
109 |   .apos-ai-helper-text__dialog.apos-is-triggered {
110 |     opacity: 1;
111 |     pointer-events: auto;
112 |   }
113 | 
114 |   .apos-context-menu__dialog {
115 |     width: 500px;
116 |   }
117 | 
118 |   p {
119 |     font-size: 14px;
120 |     line-height: 1.25;
121 |   }
122 | 
123 |   textarea {
124 |     width: 100%;
125 |     line-height: 1.25;
126 |     padding: 4px;
127 |     height: 48px;
128 |     resize: none;
129 |   }
130 | 
131 |   .apos-ai-helper-text__footer {
132 |     display: flex;
133 |     justify-content: flex-end;
134 |     margin-top: 10px;
135 |   }
136 | 
137 |   .apos-ai-helper-text__footer .apos-button__wrapper {
138 |     margin-left: 7.5px;
139 |   }
140 | 
141 |   .apos-ai-helper-text__remove {
142 |     display: flex;
143 |     justify-content: flex-end;
144 |   }
145 | </style>
146 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
  1 | # ai-helper: AI for content creation in Apostrophe 3
  2 | 
  3 | ## Purpose
  4 | 
  5 | This module enhances Apostrophe with AI-driven helpers. Currently this module offers:
  6 | 
  7 | * A button to generate an image from a text prompt via OpenAI.
  8 | * An insert menu option to generate rich text from a prompt via OpenAI.
  9 | 
 10 | You will need to [obtain your own API key from OpenAI](https://openai.com/api/). To generate images using the default
 11 | model you will need to verify your organization in your OpenAI account.
 12 | 
 13 | ## Caveats
 14 | 
 15 | This is a beta module where we are experimenting with simple ways to integrate generative AI in the
 16 | content creation process. AI systems can generate surprising and sometimes inappropriate results.
 17 | The OpenAI API itself is subject to change.
 18 | 
 19 | ## Install
 20 | 
 21 | ```bash
 22 | npm install @apostrophecms/ai-helper
 23 | ```
 24 | 
 25 | ```javascript
 26 | // in app.js
 27 | modules: {
 28 |   '@apostrophecms/ai-helper': {
 29 |     // Optional: specify a particular GPT model name.
 30 |     // This is the default:
 31 |     textModel: 'gpt-5.1',
 32 |     // Optional: specify a particular image generation model.
 33 |     // This is the default:
 34 |     imageModel: 'gpt-image-1-mini',
 35 |     // Optional: override the maximum number of tokens,
 36 |     // up to GPT's limit for the model. This is the default:
 37 |     textMaxTokens: 1000
 38 |   }
 39 | }
 40 | ```
 41 | 
 42 | ```javascript
 43 | // Anywhere you have a rich text widget that should support
 44 | // AI-generated rich text
 45 | someAreaName: {
 46 |   widgets: {
 47 |     '@apostrophecms/rich-text': {
 48 |       toolbar: [
 49 |         'styles',
 50 |         'bold'
 51 |       ],
 52 |       insert: [
 53 |         'ai'
 54 |       ],
 55 |       // Generated text includes headings if asked for
 56 |       styles: [
 57 |         {
 58 |           name: 'Heading',
 59 |           tag: 'h2'
 60 |         },
 61 |         {
 62 |           name: 'Subheading',
 63 |           tag: 'h3'
 64 |         },
 65 |         {
 66 |           name: 'Paragraph',
 67 |           tag: 'p'
 68 |         }
 69 |       ]
 70 |     }
 71 |   }
 72 | }
 73 | ```
 74 | 
 75 | 
 76 | ```javascript
 77 | // in modules/@apostrophecms/security-headers/index.js,
 78 | // only if you are using that module in your project
 79 | module.exports = {
 80 |   options: {
 81 |     policies: {
 82 |       ai: {
 83 |         // Images served by OpenAI, for editing purposes only
 84 |         'img-src': '*.blob.core.windows.net'
 85 |       }
 86 |     }
 87 |   }
 88 | };
 89 | ```
 90 | 
 91 | ## Run
 92 | 
 93 | ```bash
 94 | export APOS_OPENAI_KEY=get-your-own-key-from-openai
 95 | node app
 96 | ```
 97 | 
 98 | ## Usage
 99 | 
100 | ### Image Generation
101 | 
102 | Add an image widget to the page. Click the edit pencil, then "Browse" as you normally would.
103 | 
104 | When the media manager appears, click the "🤖" (robot) button in the upper right corner.
105 | 
106 | When the "Generate Image" dialog appears, follow the instructions to enter a plain English
107 | description of the image you want. Then click "Generate." After a pause, four images
108 | will appear. You can do this as many times as you wish.
109 | 
110 | When you are happy with the results, click on the best of the four images to review it
111 | and click "Select" to bring it into the media library, "Variants" to generate
112 | variants of it, or "Delete" to discard it.
113 | 
114 | The image helper generates 1024x1024 images. This is the maximum size supported by OpenAI.
115 | As of this writing, smaller images are not much cheaper, and are unlikely to look good
116 | in various placements on a website.
117 | 
118 | Note that generated images not selected for use in the media library after one hour will
119 | be discarded by OpenAI. Those you select are permanently imported to the media library.
120 | 
121 | ### Text Generation
122 | 
123 | Configure a rich text widget with the `insert` subproperty configured as shown above.
124 | 
125 | Now press the "/" (slash) key at the start of any line to bring up the insert menu.
126 | Choose "Generate Text" to generate text.
127 | 
128 | Enter a prompt as suggested and click "Generate." After a pause, the generated text
129 | is inserted into the rich text widget.
130 | 
131 | Note that the generated text can include headings and links if you so request.
132 | It is also a good idea to specify a word count. You can make your request using
133 | ordinary conversational language.
134 | 


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-image/ui/apos/components/AposAiHelperImageManager.vue:
--------------------------------------------------------------------------------
  1 | <template>
  2 |   <AposModal
  3 |     class="apos-ai-helper-image-manager"
  4 |     :modal="modal"
  5 |     modal-title="aposAiHelper:generateImage"
  6 |     @inactive="modal.active = false"
  7 |     @show-modal="modal.showModal = true"
  8 |     @esc="close"
  9 |     @no-modal="$emit('safe-close')"
 10 |   >
 11 |     <template #main>
 12 |       <AposModalBody>
 13 |         <template #bodyMain>
 14 |           <form class="apos-ai-helper-form">
 15 |             <textarea
 16 |               v-model="prompt"
 17 |               :placeholder="$t('aposAiHelper:imagePlaceholderText')"
 18 |             />
 19 |             <AposButton
 20 |               :disabled="!prompt.length"
 21 |               :label="$t('aposAiHelper:generateImage')"
 22 |               @click.prevent="generate({})"
 23 |             />
 24 |             <p v-if="error">
 25 |               An error occurred.
 26 |             </p>
 27 |             <div class="apos-ai-helper-images">
 28 |               <button
 29 |                 v-for="image in images"
 30 |                 :key="image._id"
 31 |                 class="apos-ai-helper-image"
 32 |                 @click.prevent="edit(image)"
 33 |               >
 34 |                 <img
 35 |                   :src="image.url"
 36 |                 >
 37 |               </button>
 38 |             </div>
 39 |           </form>
 40 |         </template>
 41 |       </AposModalBody>
 42 |     </template>
 43 |   </AposModal>
 44 | </template>
 45 | 
 46 | <script>
 47 | export default {
 48 |   emits: [ 'modal-result', 'safe-close' ],
 49 |   data() {
 50 |     return {
 51 |       modal: {
 52 |         active: false,
 53 |         type: 'slide',
 54 |         origin: 'right',
 55 |         showModal: false,
 56 |         width: 'two-thirds'
 57 |       },
 58 |       error: false,
 59 |       prompt: '',
 60 |       images: []
 61 |     };
 62 |   },
 63 |   async mounted() {
 64 |     this.modal.active = true;
 65 |     // TODO: Error handling
 66 |     this.images = (await apos.http.get(`${apos.image.action}/ai-helper`, { busy: true }))
 67 |       .images;
 68 |     this.expireInterval = setInterval(this.expire, 1000 * 60);
 69 |   },
 70 |   unmounted() {
 71 |     clearInterval(this.expireInterval);
 72 |   },
 73 |   methods: {
 74 |     expire() {
 75 |       const expired = new Date(Date.now() - 1000 * 60 * 60).toISOString();
 76 |       if (this.images.some(image => image.createdAt <= expired)) {
 77 |         this.images = this.images.filter(image => image.createdAt > expired);
 78 |       }
 79 |     },
 80 |     close() {
 81 |       this.modal.showModal = false;
 82 |     },
 83 |     async edit(image) {
 84 |       const result = await apos.modal.execute('AposAiHelperImageEditor', {
 85 |         image
 86 |       });
 87 |       const action = result?.action;
 88 |       // If I do this as !action: return I get a linter error 🤷‍♂️
 89 |       if (action) {
 90 |         if (action === 'save') {
 91 |           await this.save(image);
 92 |         } else if (action === 'variations') {
 93 |           await this.generate({
 94 |             variantOf: image
 95 |           });
 96 |         } else if (action === 'delete') {
 97 |           await this.remove(image);
 98 |         } else {
 99 |           throw new Error(`Unimplemented action: ${action}`);
100 |         }
101 |       }
102 |     },
103 |     async generate({ variantOf }) {
104 |       this.error = false;
105 |       try {
106 |         const result = await apos.http.post(`${apos.image.action}/ai-helper`, {
107 |           body: {
108 |             prompt: variantOf?.prompt || this.prompt,
109 |             variantOf: variantOf?._id
110 |           },
111 |           busy: true
112 |         });
113 |         this.images = [ ...result.images, ...this.images ];
114 |         this.$el.querySelector('[data-apos-modal-inner]').scrollTo(0, 0);
115 |       } catch (e) {
116 |         this.error = true;
117 |       }
118 |     },
119 |     async save({ _id }) {
120 |       try {
121 |         const updated = await apos.http.patch(`${apos.image.action}/ai-helper/${_id}`, {
122 |           body: {
123 |             accepted: 1
124 |           },
125 |           busy: true
126 |         });
127 |         const image = updated._image;
128 |         this.$emit('modal-result', image);
129 |         this.modal.showModal = false;
130 |         apos.bus.$emit('content-changed', {
131 |           doc: image,
132 |           action: 'insert',
133 |           select: true
134 |         });
135 |       } catch (e) {
136 |         this.error = true;
137 |       }
138 |     },
139 |     async remove({ _id }) {
140 |       this.images = this.images.filter(image => image._id !== _id);
141 |       await apos.http.delete(`${apos.image.action}/ai-helper/${_id}`, {
142 |         busy: true
143 |       });
144 |     }
145 |   }
146 | };
147 | </script>
148 | 
149 | <style lang="scss" scoped>
150 | .apos-ai-helper-form {
151 |   display: flex;
152 |   flex-direction: column;
153 |   gap: 16px;
154 | }
155 | .apos-ai-helper-images {
156 |   display: grid;
157 |   grid-template-columns: 1fr 1fr;
158 |   gap: 16px;
159 | }
160 | .apos-ai-helper-image img {
161 |   width: 100%;
162 |   height: 100%;
163 |   aspect-ratio: 1;
164 |   object-fit: contain;
165 | }
166 | textarea {
167 |   height: 4em;
168 | }
169 | </style>
170 | 


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-rich-text-widget/index.js:
--------------------------------------------------------------------------------
  1 | const marked = require('marked');
  2 | 
  3 | // Canned sample results for cheap & offline testing
  4 | const mockResults = {
  5 |   id: 'cmpl-76iWL6JGwp0S3MoFx77100hd4cZjr',
  6 |   object: 'text_completion',
  7 |   created: 1681835461,
  8 |   model: 'gpt-3.5-turbo-instruct',
  9 |   choices: [
 10 |     {
 11 |       text: '\n# Glamping Cats\n\n## The Feline Appeal\nGlamping cats have become increasingly popular in recent years, with many people choosing to include their cats in their glamping experience. Cats are incredibly adaptive and can quickly become accustomed to new environments with minimal stress. They can also make perfect glamping companions, as their intelligence and inquisitive nature ensure that they will quickly find ways to entertain themselves and those around them.\n\n## Glamping Must-Haves\nBefore taking your cat on a glamping vacation, make sure you are prepared. Consider bringing items such as litterbox, food dish, bedding material, and toys. Each of these items can help create an enjoyable and stress-free environment for your cat while glamping.\n\n## RV Luxuries\nRVs come equipped with a variety of amenities that make glamping with cats more enjoyable. Consider portable kennels and carriers for your cat, as well as waterproof mats and blankets for lounging. RV heating capabilities can also create a comfortable and safe environment for your cat, no matter the temperature outside. \n\n## All-Weather Fun\nGlamping with cats can provide opportunities for outdoor fun regardless of the weather. Rainy days can be spent cuddling cats inside, while sunny days can offer the adventure of roaming the outdoors. Glamping can also offer cats the opportunity to meet new people and explore new environments.\n\n## A Special Bond\nUltimately, glamping with cats can create a special bond between you and your pet. Cats tend to be incredibly social creatures and thrive off of human interaction. Glamping with cats offers the perfect opportunity to strengthen your relationship and appreciate the wonders of nature.',
 12 |       index: 0,
 13 |       logprobs: null,
 14 |       finish_reason: 'stop'
 15 |     }
 16 |   ],
 17 |   usage: {
 18 |     prompt_tokens: 32,
 19 |     completion_tokens: 342,
 20 |     total_tokens: 374
 21 |   }
 22 | };
 23 | 
 24 | module.exports = {
 25 |   improve: '@apostrophecms/rich-text-widget',
 26 |   beforeSuperClass(self) {
 27 |     self.options.editorInsertMenu = {
 28 |       ...self.options.editorInsertMenu,
 29 |       ai: {
 30 |         icon: 'robot-icon',
 31 |         label: 'aposAiHelper:generateTextLabel',
 32 |         component: 'AposAiHelperTextDialog',
 33 |         description: 'aposAiHelper:generateTextDescription'
 34 |       }
 35 |     };
 36 |   },
 37 |   init(self) {
 38 |     const defaultOptions = self.options.defaultOptions;
 39 |     defaultOptions.insert ||= [];
 40 |     if (!defaultOptions.insert.includes('ai')) {
 41 |       defaultOptions.insert.push('ai');
 42 |     }
 43 |   },
 44 |   apiRoutes(self) {
 45 |     return {
 46 |       post: {
 47 |         async aiHelper(req) {
 48 |           try {
 49 |             const aiHelper = self.apos.modules['@apostrophecms/ai-helper'];
 50 |             aiHelper.checkPermissions(req);
 51 |             const prompt = self.apos.launder.string(req.body.prompt);
 52 |             const headingLevels = self.apos.launder
 53 |               .strings(req.body.headingLevels)
 54 |               .map(level => parseInt(level));
 55 |             if (!prompt.length) {
 56 |               throw self.apos.error('invalid');
 57 |             }
 58 |             const body = {
 59 |               messages: [
 60 |                 {
 61 |                   role: 'system',
 62 |                   content: 'You are a helpful text-generation assistant for CMS content. You generate text in Markdown format based on the given prompt. If the response is more than two paragraphs, use markdown subheadings.'
 63 |                 },
 64 |                 {
 65 |                   role: 'user',
 66 |                   content: prompt
 67 |                 }
 68 |               ],
 69 |               model: aiHelper.options.textModel,
 70 |               max_completion_tokens: aiHelper.options.textMaxTokens
 71 |             };
 72 |             async function post(...args) {
 73 |               const response = await self.apos.http.post(...args);
 74 |               return response;
 75 |             };
 76 |             const result = process.env.APOS_AI_HELPER_MOCK
 77 |               ? mockResults
 78 |               : await post('https://api.openai.com/v1/chat/completions', {
 79 |                 headers: {
 80 |                   Authorization: `Bearer ${process.env.APOS_OPENAI_KEY}`
 81 |                 },
 82 |                 body
 83 |               })
 84 |             ;
 85 |             const content = result?.choices?.[0]?.message?.content;
 86 |             if (!content) {
 87 |               throw self.apos.error('error');
 88 |             }
 89 |             // Remap headings to levels actually available in this widget
 90 |             const markdown = content.replace(/(^|\n)(#+) /g, (all, before, hashes) => {
 91 |               if (!headingLevels.length) {
 92 |                 return '';
 93 |               }
 94 |               const level = headingLevels[hashes.length - 1];
 95 |               return before + (level ? (before + '#'.repeat(level) + ' ') : '');
 96 |             });
 97 | 
 98 |             const html = marked.parse(markdown);
 99 |             return {
100 |               html
101 |             };
102 |           } catch (e) {
103 |             self.apos.util.warn(e);
104 |             if (e.status === 429) {
105 |               self.apos.notify(req, 'aposAiHelper:rateLimitExceeded');
106 |             } else if (e.status === 400) {
107 |               self.apos.notify(req, 'aposAiHelper:invalidRequest');
108 |             }
109 |             throw e;
110 |           }
111 |         }
112 |       }
113 |     };
114 |   }
115 | };
116 | 


--------------------------------------------------------------------------------
/modules/@apostrophecms/ai-helper-image/index.js:
--------------------------------------------------------------------------------
  1 | const fetch = require('node-fetch');
  2 | const { createId } = require('@paralleldrive/cuid2');
  3 | const path = require('path');
  4 | const util = require('util');
  5 | const FormData = require('form-data');
  6 | const fs = require('fs');
  7 | const sharp = require('sharp');
  8 | const unlink = util.promisify(fs.unlink);
  9 | const writeFile = util.promisify(fs.writeFile);
 10 | 
 11 | module.exports = {
 12 |   improve: '@apostrophecms/image',
 13 |   icons: {
 14 |     'robot-icon': 'Robot',
 15 |     'group-icon': 'Group'
 16 |   },
 17 |   utilityOperations: {
 18 |     add: {
 19 |       aiInsertImage: {
 20 |         label: 'aposAiHelper:utilityOperationLabel',
 21 |         icon: 'robot-icon',
 22 |         iconOnly: true,
 23 |         relationships: true,
 24 |         button: true,
 25 |         modalOptions: {
 26 |           modal: 'AposAiHelperImageManager'
 27 |         },
 28 |         canCreate: true
 29 |       }
 30 |     }
 31 |   },
 32 |   async init(self) {
 33 |     self.uploadfs = self.apos.uploadfs;
 34 |     self.aiHelperImages = self.apos.db.collection('aposAiHelperImages');
 35 |     await self.aiHelperImages.createIndex({
 36 |       userId: 1,
 37 |       createdAt: -1
 38 |     });
 39 |     self.scheduleCleanup();
 40 |   },
 41 |   methods(self) {
 42 |     return {
 43 |       scheduleCleanup() {
 44 |         setInterval(self.cleanup, 1000 * 60 * 60);
 45 |       },
 46 |       async cleanup() {
 47 |         const images = await self.aiHelperImages.find({
 48 |           createdAt: {
 49 |             // OpenAI image URLs were originally only good for an hour, and
 50 |             // it's not a bad policy: the ones you don't use are
 51 |             // kept for an hour
 52 |             $lt: new Date(Date.now() - 1000 * 60 * 60)
 53 |           }
 54 |         }).toArray();
 55 |         for (const image of images) {
 56 |           await self.aiHelperRemoveImage(image._id);
 57 |         }
 58 |       },
 59 |       // Fetch the image to a temporary file and return the path to that file
 60 |       async aiHelperFetchImage(req, image) {
 61 |         const response = await fetch(self.aiHelperImageUrl(req, image));
 62 |         const buffer = await response.buffer();
 63 |         const temp = path.join(self.apos.rootDir, `data/temp/${image._id}.png`);
 64 |         await writeFile(temp, buffer);
 65 |         return temp;
 66 |       },
 67 |       // Write a base64 image to uploadfs
 68 |       async aiHelperWriteImageToUploadfs(_id, base64) {
 69 |         const temp = path.join(self.apos.rootDir, `data/temp/${_id}.png`);
 70 |         try {
 71 |           await writeFile(temp, Buffer.from(base64, 'base64'));
 72 |           await util.promisify(self.uploadfs.copyIn)(temp, `/ai-helper-images/${_id}.png`);
 73 |         } finally {
 74 |           try {
 75 |             await unlink(temp);
 76 |           } catch (e) {
 77 |             self.apos.util.warn(e);
 78 |             // never got that far
 79 |           }
 80 |         }
 81 |       },
 82 |       aiHelperImageUrl(req, image) {
 83 |         return image.url || (new URL(self.uploadfs.getUrl() + `/ai-helper-images/${image._id}.png`, req.baseUrl)).toString();
 84 |       },
 85 |       async aiHelperRemoveImage(_id, criteria = {}) {
 86 |         const aiImage = await self.aiHelperImages.findOne({
 87 |           _id,
 88 |           ...criteria
 89 |         });
 90 |         if (!aiImage) {
 91 |           // Already gone, races are normal for this in multiserver
 92 |           return;
 93 |         }
 94 |         if (!aiImage.url) {
 95 |           // Newer image, received as base64, now in uploadfs waiting to be removed
 96 |           try {
 97 |             await util.promisify(self.uploadfs.remove)(`/ai-helper-images/${_id}.png`);
 98 |           } catch (e) {
 99 |             // Probably already deleted
100 |             self.apos.util.warn(e);
101 |           }
102 |         }
103 |         await self.aiHelperImages.deleteOne({ _id });
104 |       }
105 |     };
106 |   },
107 |   apiRoutes(self) {
108 |     return {
109 |       get: {
110 |         async 'ai-helper'(req) {
111 |           const images = await self.aiHelperImages.find({
112 |             userId: req.user._id,
113 |             createdAt: {
114 |               // OpenAI image URLs were originally only good for an hour, and
115 |               // it's not a bad policy: the ones you don't use are
116 |               // kept for an hour
117 |               $gte: new Date(Date.now() - 1000 * 60 * 60)
118 |             }
119 |           }).sort({
120 |             createdAt: -1
121 |           }).toArray();
122 |           for (const image of images) {
123 |             image.url ||= self.aiHelperImageUrl(req, image);
124 |           }
125 |           return {
126 |             images
127 |           };
128 |         }
129 |       },
130 |       delete: {
131 |         async 'ai-helper/:_id'(req) {
132 |           await self.aiHelperRemoveImage(req.params._id, {
133 |             userId: req.user._id
134 |           });
135 |           return {};
136 |         }
137 |       },
138 |       post: {
139 |         async 'ai-helper'(req) {
140 |           const aiHelper = self.apos.modules['@apostrophecms/ai-helper'];
141 |           aiHelper.checkPermissions(req);
142 |           const prompt = self.apos.launder.string(req.body.prompt);
143 |           const variantOf = self.apos.launder.id(req.body.variantOf);
144 |           if (!prompt.length) {
145 |             throw self.apos.error('invalid');
146 |           }
147 |           const body = variantOf ? new FormData() : {};
148 |           if (!variantOf) {
149 |             set('prompt', prompt);
150 |           } else {
151 |             set('prompt', `${prompt} (but a little different this time)`);
152 |           }
153 |           // The modern models are slow and tend to nail it, so don't make things
154 |           // even slower by generating multiples every time
155 |           set('n', 1);
156 |           set('size', '1024x1024');
157 |           set('model', aiHelper.options.imageModel);
158 |           let temp;
159 |           // Fake results for cheap & offline testing
160 |           if (process.env.APOS_AI_HELPER_MOCK) {
161 |             const now = new Date();
162 |             const images = [];
163 |             for (let i = 0; (i < 4); i++) {
164 |               images.push({
165 |                 _id: createId(),
166 |                 userId: req.user._id,
167 |                 createdAt: now,
168 |                 url: self.apos.asset.url('/modules/@apostrophecms/ai-helper-image/placeholder.jpg')
169 |               });
170 |             }
171 |             return {
172 |               images
173 |             };
174 |           }
175 |           try {
176 |             if (variantOf) {
177 |               const existing = await self.aiHelperImages.findOne({
178 |                 _id: variantOf
179 |               });
180 |               if (!existing) {
181 |                 throw self.apos.error('notfound');
182 |               }
183 |               temp = await self.aiHelperFetchImage(req, existing);
184 |               body.append('image', fs.createReadStream(temp));
185 |             }
186 |             const command = variantOf ? 'edits' : 'generations';
187 |             const result = await self.apos.http.post(`https://api.openai.com/v1/images/${command}`, {
188 |               headers: {
189 |                 Authorization: `Bearer ${process.env.APOS_OPENAI_KEY}`
190 |               },
191 |               body
192 |             });
193 |             if (temp) {
194 |               fs.unlinkSync(temp);
195 |             }
196 |             if (!result.data) {
197 |               throw self.apos.error('error');
198 |             }
199 |             const images = [];
200 |             const now = new Date();
201 |             for (const item of result.data) {
202 |               const id = createId();
203 |               const image = {
204 |                 _id: id,
205 |                 userId: req.user._id,
206 |                 createdAt: now,
207 |                 uploadfs: !!item.b64_json,
208 |                 prompt
209 |               };
210 |               if (item.b64_json) {
211 |                 await self.aiHelperWriteImageToUploadfs(id, item.b64_json);
212 |               }
213 |               item.b64_json = true;
214 |               // Do not write url to the database, that's our cue that it has
215 |               // to be loaded on demand
216 |               await self.aiHelperImages.insertOne(image);
217 |               image.url = self.aiHelperImageUrl(req, image);
218 |               images.push(image);
219 |             }
220 |             return {
221 |               images
222 |             };
223 |           } catch (e) {
224 |             if (e.status === 429) {
225 |               self.apos.notify(req, 'aposAiHelper:rateLimitExceeded');
226 |             } else if (e.status === 400) {
227 |               self.apos.util.error(e);
228 |               self.apos.notify(req, 'aposAiHelper:invalidRequest');
229 |             } else {
230 |               self.apos.util.error(e);
231 |             }
232 |             throw e;
233 |           } finally {
234 |             if (temp) {
235 |               try {
236 |                 await unlink(temp);
237 |               } catch (e) {
238 |                 // Don't care if it never got there
239 |               }
240 |             }
241 |           }
242 |           function set(key, value) {
243 |             if (variantOf) {
244 |               body.append(key, value);
245 |             } else {
246 |               body[key] = value;
247 |             }
248 |           }
249 |         }
250 |       },
251 |       patch: {
252 |         async 'ai-helper/:_id'(req) {
253 |           const _id = req.params._id;
254 |           if (!req.body.accepted) {
255 |             // Currently the only property that can be PATCHed is "accepted"
256 |             throw self.apos.error('invalid');
257 |           }
258 |           if (!_id.length) {
259 |             throw self.apos.error('invalid');
260 |           }
261 |           const helperImage = await self.aiHelperImages.findOne({ _id });
262 |           if (!helperImage) {
263 |             throw self.apos.error('notfound');
264 |           }
265 |           const { prompt } = helperImage;
266 |           // apos.http has a bug with binary data, use node-fetch
267 |           let temp;
268 |           let tempJpg;
269 |           try {
270 |             temp = await self.aiHelperFetchImage(req, helperImage);
271 |             const tempJpg = temp.replace(/\.\w+$/, '') + '.jpg';
272 |             await sharp(temp).toFile(tempJpg);
273 |             const attachment = await self.apos.attachment.insert(req, {
274 |               name: self.apos.util.slugify(prompt) + '.jpg',
275 |               path: tempJpg
276 |             });
277 |             const image = await self.apos.image.insert(req, {
278 |               title: prompt,
279 |               attachment
280 |             });
281 |             self.apos.attachment.all(image, { annotate: true });
282 |             const updated = {
283 |               ...image,
284 |               accepted: true,
285 |               imageId: image._id,
286 |               _image: image
287 |             };
288 |             await self.aiHelperImages.updateOne({ _id }, {
289 |               $set: {
290 |                 accepted: true,
291 |                 imageId: image._id
292 |               }
293 |             });
294 |             return updated;
295 |           } finally {
296 |             try {
297 |               if (temp) {
298 |                 await unlink(temp);
299 |               }
300 |               if (tempJpg) {
301 |                 await unlink(tempJpg);
302 |               }
303 |             } catch (e) {
304 |               // We don't care if it never got there
305 |             }
306 |           }
307 |         }
308 |       }
309 |     };
310 |   }
311 | };
312 | 


--------------------------------------------------------------------------------