2 files changed, 17 insertions, 16 deletions

diff --git a/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts b/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
index ef4bbbc47..754d230c8 100644
--- a/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
+++ b/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
@@ -13,63 +13,63 @@ const standardOptions = ['title', 'backgroundColor'];
  * Description of document options and data field for each type.
  */
 const documentTypesInfo: { [key in supportedDocTypes]: { options: string[]; dataDescription: string } } = {
-    [supportedDocumentTypes.flashcard]: {
+    [supportedDocTypes.flashcard]: {
         options: [...standardOptions, 'fontColor', 'text_align'],
         dataDescription: 'an array of two strings.  the first string contains a question, and the second string contains an answer',
     },
-    [supportedDocumentTypes.text]: {
+    [supportedDocTypes.text]: {
         options: [...standardOptions, 'fontColor', 'text_align'],
         dataDescription: 'The text content of the document.',
     },
-    [supportedDocumentTypes.html]: {
+    [supportedDocTypes.html]: {
         options: [],
         dataDescription: 'The HTML-formatted text content of the document.',
     },
-    [supportedDocumentTypes.equation]: {
+    [supportedDocTypes.equation]: {
         options: [...standardOptions, 'fontColor'],
         dataDescription: 'The equation content as a string.',
     },
-    [supportedDocumentTypes.functionplot]: {
+    [supportedDocTypes.functionplot]: {
         options: [...standardOptions, 'function_definition'],
         dataDescription: 'The function definition(s) for plotting. Provide as a string or array of function definitions.',
     },
-    [supportedDocumentTypes.dataviz]: {
+    [supportedDocTypes.dataviz]: {
         options: [...standardOptions, 'chartType'],
         dataDescription: 'A string of comma-separated values representing the CSV data.',
     },
-    [supportedDocumentTypes.notetaking]: {
+    [supportedDocTypes.notetaking]: {
         options: standardOptions,
         dataDescription: 'The initial content or structure for note-taking.',
     },
-    [supportedDocumentTypes.rtf]: {
+    [supportedDocTypes.rtf]: {
         options: standardOptions,
         dataDescription: 'The rich text content in RTF format.',
     },
-    [supportedDocumentTypes.image]: {
+    [supportedDocTypes.image]: {
         options: standardOptions,
         dataDescription: 'The image content as an image file URL.',
     },
-    [supportedDocumentTypes.pdf]: {
+    [supportedDocTypes.pdf]: {
         options: standardOptions,
         dataDescription: 'the pdf content as a PDF file url.',
     },
-    [supportedDocumentTypes.audio]: {
+    [supportedDocTypes.audio]: {
         options: standardOptions,
         dataDescription: 'The audio content as a file url.',
     },
-    [supportedDocumentTypes.video]: {
+    [supportedDocTypes.video]: {
         options: standardOptions,
         dataDescription: 'The video content as a file url.',
     },
-    [supportedDocumentTypes.message]: {
+    [supportedDocTypes.message]: {
         options: standardOptions,
         dataDescription: 'The message content of the document.',
     },
-    [supportedDocumentTypes.diagram]: {
+    [supportedDocTypes.diagram]: {
         options: ['title', 'backgroundColor'],
         dataDescription: 'diagram content as a text string in Mermaid format.',
     },
-    [supportedDocumentTypes.script]: {
+    [supportedDocTypes.script]: {
         options: ['title', 'backgroundColor'],
         dataDescription: 'The compilable JavaScript code. Use this for creating scripts.',
     },
diff --git a/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts b/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
index e92d87dfd..37907fd4f 100644
--- a/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
+++ b/src/client/views/nodes/chatbot/tools/ImageCreationTool.ts
@@ -5,6 +5,7 @@ import { ParametersType, ToolInfo } from '../types/tool_types';
 import { Observation } from '../types/types';
 import { BaseTool } from './BaseTool';
 import { Upload } from '../../../../../server/SharedMediaTypes';
+import { List } from '../../../../../fields/List';
 
 const imageCreationToolParams = [
     {
@@ -41,7 +42,7 @@ export class ImageCreationTool extends BaseTool<ImageCreationToolParamsType> {
                 image_prompt,
             })) as { result: Upload.FileInformation & Upload.InspectionResults; url: string };
             console.log('Image generation result:', result);
-            this._createImage(result, { text: RTFCast(image_prompt) });
+            this._createImage(result, { text: RTFCast(image_prompt), ai: 'dall-e-3', tags: new List<string>(['@ai']) });
             return url
                 ? [
                       {