1 files changed, 22 insertions, 21 deletions
diff --git a/src/client/views/nodes/chatbot/tools/CreateTextDocumentTool.ts b/src/client/views/nodes/chatbot/tools/CreateTextDocumentTool.ts
index fae78aa49..487fc951d 100644
--- a/src/client/views/nodes/chatbot/tools/CreateTextDocumentTool.ts
+++ b/src/client/views/nodes/chatbot/tools/CreateTextDocumentTool.ts
@@ -2,7 +2,7 @@ import { v4 as uuidv4 } from 'uuid';
 import { Networking } from '../../../../Network';
 import { BaseTool } from './BaseTool';
 import { Observation } from '../types/types';
-import { ParametersType } from '../types/tool_types';
+import { ParametersType, ToolInfo } from '../types/tool_types';
 import { DocumentOptions } from '../../../../documents/Documents';
 import { RTFCast, StrCast } from '../../../../../fields/Types';
 
@@ -19,40 +19,41 @@ const createTextDocToolParams = [
         description: 'The title of the document',
         required: true,
     },
-    {
-        name: 'background_color',
-        type: 'string',
-        description: 'The background color of the document as a hex string',
-        required: false,
-    },
-    {
-        name: 'font_color',
-        type: 'string',
-        description: 'The font color of the document as a hex string',
-        required: false,
-    },
+    // {
+    //     name: 'background_color',
+    //     type: 'string',
+    //     description: 'The background color of the document as a hex string',
+    //     required: false,
+    // },
+    // {
+    //     name: 'font_color',
+    //     type: 'string',
+    //     description: 'The font color of the document as a hex string',
+    //     required: false,
+    // },
 ] as const;
 
 type CreateTextDocToolParamsType = typeof createTextDocToolParams;
 
+const createTextDocToolInfo: ToolInfo<CreateTextDocToolParamsType> = {
+    name: 'createTextDoc',
+    description: 'Creates a text document with the provided content and title. Use if the user wants to create a textbox or text document of some sort. Can use after a search or other tool to save information.',
+    citationRules: 'No citation needed.',
+    parameterRules: createTextDocToolParams,
+};
+
 export class CreateTextDocTool extends BaseTool<CreateTextDocToolParamsType> {
     private _addLinkedDoc: (doc_type: string, data: string, options: DocumentOptions, id: string) => void;
 
     constructor(addLinkedDoc: (text_content: string, data: string, options: DocumentOptions, id: string) => void) {
-        super(
-            'createTextDoc',
-            'Creates a text document with the provided content and title (and of specified other options if wanted)',
-            createTextDocToolParams,
-            'Provide the text content and title (and optionally color) for the document.',
-            'Creates a text document with the provided content and title (and of specified other options if wanted). Use if the user wants to create a textbox or text document of some sort. Can use after a search or other tool to save information.'
-        );
+        super(createTextDocToolInfo);
         this._addLinkedDoc = addLinkedDoc;
     }
 
     async execute(args: ParametersType<CreateTextDocToolParamsType>): Promise<Observation[]> {
         try {
             console.log(RTFCast(args.text_content));
-            this._addLinkedDoc('text', args.text_content, { title: args.title, backgroundColor: args.background_color, text_fontColor: args.font_color }, uuidv4());
+            this._addLinkedDoc('text', args.text_content, { title: args.title }, uuidv4());
             return [{ type: 'text', text: 'Created text document.' }];
         } catch (error) {
             return [{ type: 'text', text: 'Error creating text document, ' + error }];