image generation works better

1 files changed, 6 insertions, 2 deletions

diff --git a/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts b/src/client/views/nodes/chatbot/tools/CreateAnyDocTool.ts
index 4c059177b..36f133503 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' | 'function_plot' | 'dataviz' | 'note_taking' | 'rtf' | 'message' | 'mermaid_diagram';
-const supportedDocumentTypes: supportedDocumentTypesType[] = ['text', 'html', 'equation', 'function_plot', 'dataviz', 'note_taking', 'rtf', 'message', 'mermaid_diagram'];
+type supportedDocumentTypesType = 'text' | 'html' | 'equation' | 'function_plot' | 'dataviz' | 'note_taking' | 'rtf' | 'message' | 'mermaid_diagram' | 'script';
+const supportedDocumentTypes: supportedDocumentTypesType[] = ['text', 'html', 'equation', 'function_plot', 'dataviz', 'note_taking', 'rtf', 'message', 'mermaid_diagram', 'script'];
 
 /**
  * Description of document options and data field for each type.
@@ -50,6 +50,10 @@ const documentTypesInfo = {
         options: ['title', 'backgroundColor', 'layout'],
         dataDescription: 'The Mermaid diagram content.',
     },
+    script: {
+        options: ['title', 'backgroundColor', 'layout'],
+        dataDescription: 'The compilable JavaScript code. Use this for creating scripts.',
+    },
 };
 
 const createAnyDocumentToolParams = [