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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
import { v4 as uuidv4 } from 'uuid';
import { BaseTool } from './BaseTool';
import { Observation } from '../types/types';
import { ParametersType, TypeMap, Parameter } from '../types/tool_types';
import { DocumentOptions, Docs } from '../../../../documents/Documents';
/**
* List of supported document types.
*/
const supportedDocumentTypes = [
'text',
'image',
'pdf',
'video',
'audio',
'web',
'map',
'equation',
'functionPlot',
'dataViz',
'chat',
// Add more document types as needed
];
/**
* Description of document options for each type.
*/
const documentOptionsDescription = {
text: ['title', 'backgroundColor', 'fontColor', 'text_align', 'layout', 'text_content'],
image: ['title', 'backgroundColor', 'width', 'height', 'layout'],
pdf: ['title', 'backgroundColor', 'width', 'height', 'layout'],
video: ['title', 'backgroundColor', 'width', 'height', 'layout'],
audio: ['title', 'backgroundColor', 'layout'],
web: ['title', 'backgroundColor', 'width', 'height', 'layout', 'url'],
// Include descriptions for other document types
};
const createAnyDocumentToolParams = [
{
name: 'document_type',
type: 'string',
description: `The type of the document to create. Supported types are: ${supportedDocumentTypes.join(', ')}`,
required: true,
},
{
name: 'data',
type: 'string',
description: 'The content or data of the document (e.g., text content, URL, etc.).',
required: false,
},
{
name: 'options',
type: 'string',
description: `A JSON string representing the document options. Available options depend on the document type. For example, for 'text' documents, options include: ${documentOptionsDescription['text'].join(', ')}.`,
required: false,
},
] as const;
type CreateAnyDocumentToolParamsType = typeof createAnyDocumentToolParams;
export class CreateAnyDocumentTool extends BaseTool<CreateAnyDocumentToolParamsType> {
private _addLinkedDoc: (doc_type: string, data: string | undefined, options: DocumentOptions, id: string) => void;
constructor(addLinkedDoc: (doc_type: string, data: string | undefined, options: DocumentOptions, id: string) => void) {
super(
'createAnyDocument',
`Creates any type of document with the provided options and data. Supported document types are: ${supportedDocumentTypes.join(', ')}.`,
createAnyDocumentToolParams,
'Provide the document type, data, and options for the document. Options should be a valid JSON string containing the document options specific to the document type.',
'Creates any type of document with the provided options and data.'
);
this._addLinkedDoc = addLinkedDoc;
}
async execute(args: ParametersType<CreateAnyDocumentToolParamsType>): Promise<Observation[]> {
try {
const documentType = args.document_type.toLowerCase();
let options: DocumentOptions = {};
if (!supportedDocumentTypes.includes(documentType)) {
throw new Error(`Unsupported document type: ${documentType}. Supported types are: ${supportedDocumentTypes.join(', ')}.`);
}
if (args.options) {
try {
options = JSON.parse(args.options as string) as DocumentOptions;
} catch (e) {
throw new Error('Options must be a valid JSON string.');
}
}
const data = args.data as string | undefined;
const id = uuidv4();
// Validate and set default options based on document type
switch (documentType) {
case 'text':
if (!data) {
throw new Error('Data is required for text documents.');
}
options.title = options.title || 'New Text Document';
break;
case 'image':
case 'pdf':
case 'video':
case 'audio':
case 'web':
if (!data) {
throw new Error(`Data (e.g., URL) is required for ${documentType} documents.`);
}
options.title = options.title || `New ${documentType.charAt(0).toUpperCase() + documentType.slice(1)} Document`;
break;
// Add cases and default options for other document types as needed
default:
break;
}
this._addLinkedDoc(documentType, data, options, id);
return [{ type: 'text', text: `Created ${documentType} document with ID ${id}.` }];
} catch (error) {
return [{ type: 'text', text: 'Error creating document: ' + (error as Error).message }];
}
}
}
|
