aboutsummaryrefslogtreecommitdiff
path: root/src/client/views/nodes/chatbot/tools/CreateTextDocumentTool.ts
blob: fa978bdc35d8a1ca85f58a2cbfa66efe2279e5cd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import { v4 as uuidv4 } from 'uuid';
import { Networking } from '../../../../Network';
import { BaseTool } from './BaseTool';
import { Observation } from '../types/types';
import { ParametersType } from './ToolTypes';
import { DocumentOptions } from '../../../../documents/Documents';

const createTextDocToolParams = [
    {
        name: 'text_content',
        type: 'string',
        description: 'The text content that the document will display',
        required: true,
    },
    {
        name: 'title',
        type: 'string',
        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,
    },
] as const;

type CreateTextDocToolParamsType = typeof createTextDocToolParams;

export class CreateTextDocTool extends BaseTool<CreateTextDocToolParamsType> {
    private _addLinkedTextDoc: (text_content: string, options: DocumentOptions, id: string) => void;

    constructor(addLinkedTextDoc: (text_content: 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.'
        );
        this._addLinkedTextDoc = addLinkedTextDoc;
    }

    async execute(args: ParametersType<CreateTextDocToolParamsType>): Promise<Observation[]> {
        try {
            this._addLinkedTextDoc(args.text_content, { title: args.title, backgroundColor: args.background_color, text_fontColor: args.font_color }, uuidv4());
            return [{ type: 'text', text: 'Created text document.' }];
        } catch (error) {
            return [{ type: 'text', text: 'Error creating text document, ' + error }];
        }
    }
}