1 files changed, 8 insertions, 4 deletions
diff --git a/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts b/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
index a4871f7fd..4c059177b 100644
--- a/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
+++ b/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
@@ -7,8 +7,8 @@ import { DocumentOptions, Docs } from '../../../../documents/Documents';
 /**
  * List of supported document types that can be created via text LLM.
  */
-type supportedDocumentTypesType = 'text' | 'html' | 'equation' | 'functionPlot' | 'dataviz' | 'noteTaking' | 'rtf' | 'message';
-const supportedDocumentTypes: supportedDocumentTypesType[] = ['text', 'html', 'equation', 'functionPlot', 'dataviz', 'noteTaking', 'rtf', 'message'];
+type supportedDocumentTypesType = 'text' | 'html' | 'equation' | 'function_plot' | 'dataviz' | 'note_taking' | 'rtf' | 'message' | 'mermaid_diagram';
+const supportedDocumentTypes: supportedDocumentTypesType[] = ['text', 'html', 'equation', 'function_plot', 'dataviz', 'note_taking', 'rtf', 'message', 'mermaid_diagram'];
 
 /**
  * Description of document options and data field for each type.
@@ -26,7 +26,7 @@ const documentTypesInfo = {
         options: ['title', 'backgroundColor', 'fontColor', 'layout'],
         dataDescription: 'The equation content as a string.',
     },
-    functionPlot: {
+    function_plot: {
         options: ['title', 'backgroundColor', 'layout', 'function_definition'],
         dataDescription: 'The function definition(s) for plotting. Provide as a string or array of function definitions.',
     },
@@ -34,7 +34,7 @@ const documentTypesInfo = {
         options: ['title', 'backgroundColor', 'layout', 'chartType'],
         dataDescription: 'A string of comma-separated values representing the CSV data.',
     },
-    noteTaking: {
+    note_taking: {
         options: ['title', 'backgroundColor', 'layout'],
         dataDescription: 'The initial content or structure for note-taking.',
     },
@@ -46,6 +46,10 @@ const documentTypesInfo = {
         options: ['title', 'backgroundColor', 'layout'],
         dataDescription: 'The message content of the document.',
     },
+    mermaid_diagram: {
+        options: ['title', 'backgroundColor', 'layout'],
+        dataDescription: 'The Mermaid diagram content.',
+    },
 };
 
 const createAnyDocumentToolParams = [