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
|
import { Observation } from '../types/types';
import { ParametersType } from './ToolTypes';
import { BaseTool } from './BaseTool';
import { DocServer } from '../../../../DocServer';
import { Docs } from '../../../../documents/Documents';
import { DocumentView } from '../../DocumentView';
import { OpenWhere } from '../../OpenWhere';
import { DocCast } from '../../../../../fields/Types';
const getDocsToolParams = [
{
name: 'title',
type: 'string',
description: 'Title of the collection being created from retrieved documents',
required: true,
},
{
name: 'document_ids',
type: 'string[]',
description: 'List of document IDs to retrieve',
required: true,
},
] as const;
type GetDocsToolParamsType = typeof getDocsToolParams;
export class GetDocsTool extends BaseTool<GetDocsToolParamsType> {
private _docView: DocumentView;
constructor(docView: DocumentView) {
super(
'retrieveDocs',
'Retrieves the contents of all Documents that the user is interacting with in Dash',
getDocsToolParams,
'No need to provide anything. Just run the tool and it will retrieve the contents of all Documents that the user is interacting with in Dash.',
'Returns the documents in Dash in JSON form.'
);
this._docView = docView;
}
async execute(args: ParametersType<GetDocsToolParamsType>): Promise<Observation[]> {
const docs = args.document_ids.map(doc_id => DocCast(DocServer.GetCachedRefField(doc_id)));
const collection = Docs.Create.FreeformDocument(docs, { title: args.title });
this._docView._props.addDocTab(collection, OpenWhere.addRight);
return [{ type: 'text', text: `Collection created in Dash called ${args.title}` }];
}
}
|
