-
Notifications
You must be signed in to change notification settings - Fork 3.7k
Expand file tree
/
Copy pathinsert-table.ts
More file actions
120 lines (116 loc) · 3.52 KB
/
Copy pathinsert-table.ts
File metadata and controls
120 lines (116 loc) · 3.52 KB
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
import type { GoogleDocsInsertTableResponse, GoogleDocsToolParams } from '@/tools/google_docs/types'
import {
buildBatchUpdateMetadata,
buildInsertLocation,
resolveDocumentId,
} from '@/tools/google_docs/utils'
import type { ToolConfig } from '@/tools/types'
export const insertTableTool: ToolConfig<GoogleDocsToolParams, GoogleDocsInsertTableResponse> = {
id: 'google_docs_insert_table',
name: 'Insert Table into Google Docs Document',
description:
'Insert an empty table with the given number of rows and columns into a Google Docs document. When no index is provided, the table is appended to the end of the document.',
version: '1.0',
oauth: {
required: true,
provider: 'google-docs',
},
params: {
accessToken: {
type: 'string',
required: true,
visibility: 'hidden',
description: 'The access token for the Google Docs API',
},
documentId: {
type: 'string',
required: true,
visibility: 'user-or-llm',
description: 'The ID of the document to insert the table into',
},
rows: {
type: 'number',
required: true,
visibility: 'user-or-llm',
description: 'The number of rows in the table',
},
columns: {
type: 'number',
required: true,
visibility: 'user-or-llm',
description: 'The number of columns in the table',
},
index: {
type: 'number',
required: false,
visibility: 'user-or-llm',
description:
'The character index (the document body starts at index 1) at which to insert the table. When omitted, the table is appended to the end of the document.',
},
},
request: {
url: (params) => {
const documentId = resolveDocumentId(params)
return `https://docs.googleapis.com/v1/documents/${documentId}:batchUpdate`
},
method: 'POST',
headers: (params) => {
if (!params.accessToken) {
throw new Error('Access token is required')
}
return {
Authorization: `Bearer ${params.accessToken}`,
'Content-Type': 'application/json',
}
},
body: (params) => {
const rows = Number(params.rows)
const columns = Number(params.columns)
if (!Number.isFinite(rows) || rows < 1) {
throw new Error('Rows must be a positive number')
}
if (!Number.isFinite(columns) || columns < 1) {
throw new Error('Columns must be a positive number')
}
return {
requests: [
{
insertTable: {
...buildInsertLocation(params.index),
rows,
columns,
},
},
],
}
},
},
transformResponse: async (response: Response) => {
const responseText = await response.text()
const data = responseText.trim() ? JSON.parse(responseText) : {}
const metadata = buildBatchUpdateMetadata(data, response.url)
return {
success: true,
output: {
updatedContent: true,
metadata,
},
}
},
outputs: {
updatedContent: {
type: 'boolean',
description: 'Indicates if the table was inserted successfully',
},
metadata: {
type: 'json',
description: 'Updated document metadata including ID, title, and URL',
properties: {
documentId: { type: 'string', description: 'Google Docs document ID' },
title: { type: 'string', description: 'Document title' },
mimeType: { type: 'string', description: 'Document MIME type' },
url: { type: 'string', description: 'Document URL' },
},
},
},
}