mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-28 15:00:57 +03:00
Feature/GoogleDocs (#4613)
* add teams, gmail, outlook tools * update docs link * update credentials for oauth2 * add jira tool * add google drive, google calendar, google sheets tools, powerpoint, excel, word doc loader * update jira logo * Refactor Gmail and Outlook tools to remove maxOutputLength parameter and enhance request handling. Update response formatting to include parameters in the output. Adjust Google Drive tools to simplify success messages by removing unnecessary parameter details. * Update pnpm-lock.yaml * add google docs
This commit is contained in:
Henry Heng
@@ -0,0 +1,253 @@
|
||||
import { convertMultiOptionsToStringArray, getCredentialData, getCredentialParam, refreshOAuth2Token } from '../../../src/utils'
|
||||
import { createGoogleDocsTools } from './core'
|
||||
import type { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
|
||||
|
||||
class GoogleDocs_Tools implements INode {
|
||||
label: string
|
||||
name: string
|
||||
version: number
|
||||
type: string
|
||||
icon: string
|
||||
category: string
|
||||
description: string
|
||||
baseClasses: string[]
|
||||
credential: INodeParams
|
||||
inputs: INodeParams[]
|
||||
|
||||
constructor() {
|
||||
this.label = 'Google Docs'
|
||||
this.name = 'googleDocsTool'
|
||||
this.version = 1.0
|
||||
this.type = 'GoogleDocs'
|
||||
this.icon = 'google-docs.svg'
|
||||
this.category = 'Tools'
|
||||
this.description =
|
||||
'Perform Google Docs operations such as creating, reading, updating, and deleting documents, as well as text manipulation'
|
||||
this.baseClasses = ['Tool']
|
||||
this.credential = {
|
||||
label: 'Connect Credential',
|
||||
name: 'credential',
|
||||
type: 'credential',
|
||||
credentialNames: ['googleDocsOAuth2']
|
||||
}
|
||||
this.inputs = [
|
||||
// Document Actions
|
||||
{
|
||||
label: 'Actions',
|
||||
name: 'actions',
|
||||
type: 'multiOptions',
|
||||
description: 'Actions to perform',
|
||||
options: [
|
||||
{
|
||||
label: 'Create Document',
|
||||
name: 'createDocument'
|
||||
},
|
||||
{
|
||||
label: 'Get Document',
|
||||
name: 'getDocument'
|
||||
},
|
||||
{
|
||||
label: 'Update Document',
|
||||
name: 'updateDocument'
|
||||
},
|
||||
{
|
||||
label: 'Insert Text',
|
||||
name: 'insertText'
|
||||
},
|
||||
{
|
||||
label: 'Replace Text',
|
||||
name: 'replaceText'
|
||||
},
|
||||
{
|
||||
label: 'Append Text',
|
||||
name: 'appendText'
|
||||
},
|
||||
{
|
||||
label: 'Get Text Content',
|
||||
name: 'getTextContent'
|
||||
},
|
||||
{
|
||||
label: 'Insert Image',
|
||||
name: 'insertImage'
|
||||
},
|
||||
{
|
||||
label: 'Create Table',
|
||||
name: 'createTable'
|
||||
}
|
||||
]
|
||||
},
|
||||
// Document Parameters
|
||||
{
|
||||
label: 'Document ID',
|
||||
name: 'documentId',
|
||||
type: 'string',
|
||||
description: 'Document ID for operations on specific documents',
|
||||
show: {
|
||||
actions: [
|
||||
'getDocument',
|
||||
'updateDocument',
|
||||
'insertText',
|
||||
'replaceText',
|
||||
'appendText',
|
||||
'getTextContent',
|
||||
'insertImage',
|
||||
'createTable'
|
||||
]
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Title',
|
||||
name: 'title',
|
||||
type: 'string',
|
||||
description: 'Document title',
|
||||
show: {
|
||||
actions: ['createDocument']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
// Text Parameters
|
||||
{
|
||||
label: 'Text',
|
||||
name: 'text',
|
||||
type: 'string',
|
||||
description: 'Text content to insert or append',
|
||||
show: {
|
||||
actions: ['createDocument', 'updateDocument', 'insertText', 'appendText']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Index',
|
||||
name: 'index',
|
||||
type: 'number',
|
||||
description: 'Index where to insert text or media (1-based, default: 1 for beginning)',
|
||||
default: 1,
|
||||
show: {
|
||||
actions: ['createDocument', 'updateDocument', 'insertText', 'insertImage', 'createTable']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Replace Text',
|
||||
name: 'replaceText',
|
||||
type: 'string',
|
||||
description: 'Text to replace',
|
||||
show: {
|
||||
actions: ['updateDocument', 'replaceText']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'New Text',
|
||||
name: 'newText',
|
||||
type: 'string',
|
||||
description: 'New text to replace with',
|
||||
show: {
|
||||
actions: ['updateDocument', 'replaceText']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Match Case',
|
||||
name: 'matchCase',
|
||||
type: 'boolean',
|
||||
description: 'Whether the search should be case-sensitive',
|
||||
default: false,
|
||||
show: {
|
||||
actions: ['updateDocument', 'replaceText']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
|
||||
// Media Parameters
|
||||
{
|
||||
label: 'Image URL',
|
||||
name: 'imageUrl',
|
||||
type: 'string',
|
||||
description: 'URL of the image to insert',
|
||||
show: {
|
||||
actions: ['createDocument', 'updateDocument', 'insertImage']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Table Rows',
|
||||
name: 'rows',
|
||||
type: 'number',
|
||||
description: 'Number of rows in the table',
|
||||
show: {
|
||||
actions: ['createDocument', 'updateDocument', 'createTable']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
},
|
||||
{
|
||||
label: 'Table Columns',
|
||||
name: 'columns',
|
||||
type: 'number',
|
||||
description: 'Number of columns in the table',
|
||||
show: {
|
||||
actions: ['createDocument', 'updateDocument', 'createTable']
|
||||
},
|
||||
additionalParams: true,
|
||||
optional: true
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
|
||||
let credentialData = await getCredentialData(nodeData.credential ?? '', options)
|
||||
credentialData = await refreshOAuth2Token(nodeData.credential ?? '', credentialData, options)
|
||||
const accessToken = getCredentialParam('access_token', credentialData, nodeData)
|
||||
|
||||
if (!accessToken) {
|
||||
throw new Error('No access token found in credential')
|
||||
}
|
||||
|
||||
// Get all actions
|
||||
const actions = convertMultiOptionsToStringArray(nodeData.inputs?.actions)
|
||||
|
||||
const defaultParams = this.transformNodeInputsToToolArgs(nodeData)
|
||||
|
||||
const tools = createGoogleDocsTools({
|
||||
accessToken,
|
||||
actions,
|
||||
defaultParams
|
||||
})
|
||||
|
||||
return tools
|
||||
}
|
||||
|
||||
transformNodeInputsToToolArgs(nodeData: INodeData): Record<string, any> {
|
||||
const nodeInputs: Record<string, any> = {}
|
||||
|
||||
// Document parameters
|
||||
if (nodeData.inputs?.documentId) nodeInputs.documentId = nodeData.inputs.documentId
|
||||
if (nodeData.inputs?.title) nodeInputs.title = nodeData.inputs.title
|
||||
|
||||
// Text parameters
|
||||
if (nodeData.inputs?.text) nodeInputs.text = nodeData.inputs.text
|
||||
if (nodeData.inputs?.index) nodeInputs.index = nodeData.inputs.index
|
||||
if (nodeData.inputs?.replaceText) nodeInputs.replaceText = nodeData.inputs.replaceText
|
||||
if (nodeData.inputs?.newText) nodeInputs.newText = nodeData.inputs.newText
|
||||
if (nodeData.inputs?.matchCase !== undefined) nodeInputs.matchCase = nodeData.inputs.matchCase
|
||||
|
||||
// Media parameters
|
||||
if (nodeData.inputs?.imageUrl) nodeInputs.imageUrl = nodeData.inputs.imageUrl
|
||||
if (nodeData.inputs?.rows) nodeInputs.rows = nodeData.inputs.rows
|
||||
if (nodeData.inputs?.columns) nodeInputs.columns = nodeData.inputs.columns
|
||||
|
||||
return nodeInputs
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { nodeClass: GoogleDocs_Tools }
|
||||
@@ -0,0 +1,729 @@
|
||||
import { z } from 'zod'
|
||||
import fetch from 'node-fetch'
|
||||
import { DynamicStructuredTool } from '../OpenAPIToolkit/core'
|
||||
import { TOOL_ARGS_PREFIX } from '../../../src/agents'
|
||||
|
||||
export const desc = `Use this when you want to access Google Docs API for managing documents`
|
||||
|
||||
export interface Headers {
|
||||
[key: string]: string
|
||||
}
|
||||
|
||||
export interface Body {
|
||||
[key: string]: any
|
||||
}
|
||||
|
||||
export interface RequestParameters {
|
||||
headers?: Headers
|
||||
body?: Body
|
||||
url?: string
|
||||
description?: string
|
||||
name?: string
|
||||
actions?: string[]
|
||||
accessToken?: string
|
||||
defaultParams?: any
|
||||
}
|
||||
|
||||
// Define schemas for different Google Docs operations
|
||||
|
||||
// Document Schemas
|
||||
const CreateDocumentSchema = z.object({
|
||||
title: z.string().describe('Document title'),
|
||||
text: z.string().optional().describe('Text content to insert after creating document'),
|
||||
index: z.number().optional().default(1).describe('Index where to insert text or media (1-based, default: 1 for beginning)'),
|
||||
imageUrl: z.string().optional().describe('URL of the image to insert after creating document'),
|
||||
rows: z.number().optional().describe('Number of rows in the table to create'),
|
||||
columns: z.number().optional().describe('Number of columns in the table to create')
|
||||
})
|
||||
|
||||
const GetDocumentSchema = z.object({
|
||||
documentId: z.string().describe('Document ID to retrieve')
|
||||
})
|
||||
|
||||
const UpdateDocumentSchema = z.object({
|
||||
documentId: z.string().describe('Document ID to update'),
|
||||
text: z.string().optional().describe('Text content to insert'),
|
||||
index: z.number().optional().default(1).describe('Index where to insert text or media (1-based, default: 1 for beginning)'),
|
||||
replaceText: z.string().optional().describe('Text to replace'),
|
||||
newText: z.string().optional().describe('New text to replace with'),
|
||||
matchCase: z.boolean().optional().default(false).describe('Whether the search should be case-sensitive'),
|
||||
imageUrl: z.string().optional().describe('URL of the image to insert'),
|
||||
rows: z.number().optional().describe('Number of rows in the table to create'),
|
||||
columns: z.number().optional().describe('Number of columns in the table to create')
|
||||
})
|
||||
|
||||
const InsertTextSchema = z.object({
|
||||
documentId: z.string().describe('Document ID'),
|
||||
text: z.string().describe('Text to insert'),
|
||||
index: z.number().optional().default(1).describe('Index where to insert text (1-based, default: 1 for beginning)')
|
||||
})
|
||||
|
||||
const ReplaceTextSchema = z.object({
|
||||
documentId: z.string().describe('Document ID'),
|
||||
replaceText: z.string().describe('Text to replace'),
|
||||
newText: z.string().describe('New text to replace with'),
|
||||
matchCase: z.boolean().optional().default(false).describe('Whether the search should be case-sensitive')
|
||||
})
|
||||
|
||||
const AppendTextSchema = z.object({
|
||||
documentId: z.string().describe('Document ID'),
|
||||
text: z.string().describe('Text to append to the document')
|
||||
})
|
||||
|
||||
const GetTextContentSchema = z.object({
|
||||
documentId: z.string().describe('Document ID to get text content from')
|
||||
})
|
||||
|
||||
const InsertImageSchema = z.object({
|
||||
documentId: z.string().describe('Document ID'),
|
||||
imageUrl: z.string().describe('URL of the image to insert'),
|
||||
index: z.number().optional().default(1).describe('Index where to insert image (1-based)')
|
||||
})
|
||||
|
||||
const CreateTableSchema = z.object({
|
||||
documentId: z.string().describe('Document ID'),
|
||||
rows: z.number().describe('Number of rows in the table'),
|
||||
columns: z.number().describe('Number of columns in the table'),
|
||||
index: z.number().optional().default(1).describe('Index where to insert table (1-based)')
|
||||
})
|
||||
|
||||
class BaseGoogleDocsTool extends DynamicStructuredTool {
|
||||
protected accessToken: string = ''
|
||||
|
||||
constructor(args: any) {
|
||||
super(args)
|
||||
this.accessToken = args.accessToken ?? ''
|
||||
}
|
||||
|
||||
async makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method = 'GET',
|
||||
body,
|
||||
params
|
||||
}: {
|
||||
endpoint: string
|
||||
method?: string
|
||||
body?: any
|
||||
params?: any
|
||||
}): Promise<string> {
|
||||
const url = `https://docs.googleapis.com/v1/${endpoint}`
|
||||
|
||||
const headers = {
|
||||
Authorization: `Bearer ${this.accessToken}`,
|
||||
'Content-Type': 'application/json',
|
||||
Accept: 'application/json',
|
||||
...this.headers
|
||||
}
|
||||
|
||||
const response = await fetch(url, {
|
||||
method,
|
||||
headers,
|
||||
body: body ? JSON.stringify(body) : undefined
|
||||
})
|
||||
|
||||
if (!response.ok) {
|
||||
const errorText = await response.text()
|
||||
throw new Error(`Google Docs API Error ${response.status}: ${response.statusText} - ${errorText}`)
|
||||
}
|
||||
|
||||
const data = await response.text()
|
||||
return data + TOOL_ARGS_PREFIX + JSON.stringify(params)
|
||||
}
|
||||
|
||||
async makeDriveRequest({
|
||||
endpoint,
|
||||
method = 'GET',
|
||||
body,
|
||||
params
|
||||
}: {
|
||||
endpoint: string
|
||||
method?: string
|
||||
body?: any
|
||||
params?: any
|
||||
}): Promise<string> {
|
||||
const url = `https://www.googleapis.com/drive/v3/${endpoint}`
|
||||
|
||||
const headers = {
|
||||
Authorization: `Bearer ${this.accessToken}`,
|
||||
'Content-Type': 'application/json',
|
||||
Accept: 'application/json',
|
||||
...this.headers
|
||||
}
|
||||
|
||||
const response = await fetch(url, {
|
||||
method,
|
||||
headers,
|
||||
body: body ? JSON.stringify(body) : undefined
|
||||
})
|
||||
|
||||
if (!response.ok) {
|
||||
const errorText = await response.text()
|
||||
throw new Error(`Google Drive API Error ${response.status}: ${response.statusText} - ${errorText}`)
|
||||
}
|
||||
|
||||
const data = await response.text()
|
||||
return data + TOOL_ARGS_PREFIX + JSON.stringify(params)
|
||||
}
|
||||
}
|
||||
|
||||
// Document Tools
|
||||
class CreateDocumentTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'create_document',
|
||||
description: 'Create a new Google Docs document',
|
||||
schema: CreateDocumentSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const documentData = {
|
||||
title: params.title
|
||||
}
|
||||
|
||||
const endpoint = 'documents'
|
||||
const createResponse = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: documentData,
|
||||
params
|
||||
})
|
||||
|
||||
// Get the document ID from the response
|
||||
const documentResponse = JSON.parse(createResponse.split(TOOL_ARGS_PREFIX)[0])
|
||||
const documentId = documentResponse.documentId
|
||||
|
||||
// Now add content if provided
|
||||
const requests = []
|
||||
|
||||
if (params.text) {
|
||||
requests.push({
|
||||
insertText: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
text: params.text
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
if (params.imageUrl) {
|
||||
requests.push({
|
||||
insertInlineImage: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
uri: params.imageUrl
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
if (params.rows && params.columns) {
|
||||
requests.push({
|
||||
insertTable: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
rows: params.rows,
|
||||
columns: params.columns
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// If we have content to add, make a batch update
|
||||
if (requests.length > 0) {
|
||||
const updateEndpoint = `documents/${encodeURIComponent(documentId)}:batchUpdate`
|
||||
await this.makeGoogleDocsRequest({
|
||||
endpoint: updateEndpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params: {}
|
||||
})
|
||||
}
|
||||
|
||||
return createResponse
|
||||
} catch (error) {
|
||||
return `Error creating document: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class GetDocumentTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'get_document',
|
||||
description: 'Get a Google Docs document by ID',
|
||||
schema: GetDocumentSchema,
|
||||
baseUrl: '',
|
||||
method: 'GET',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}`
|
||||
const response = await this.makeGoogleDocsRequest({ endpoint, params })
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error getting document: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class UpdateDocumentTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'update_document',
|
||||
description: 'Update a Google Docs document with batch requests',
|
||||
schema: UpdateDocumentSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const requests = []
|
||||
|
||||
// Insert text
|
||||
if (params.text) {
|
||||
requests.push({
|
||||
insertText: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
text: params.text
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// Replace text
|
||||
if (params.replaceText && params.newText) {
|
||||
requests.push({
|
||||
replaceAllText: {
|
||||
containsText: {
|
||||
text: params.replaceText,
|
||||
matchCase: params.matchCase || false
|
||||
},
|
||||
replaceText: params.newText
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// Insert image
|
||||
if (params.imageUrl) {
|
||||
requests.push({
|
||||
insertInlineImage: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
uri: params.imageUrl
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// Create table
|
||||
if (params.rows && params.columns) {
|
||||
requests.push({
|
||||
insertTable: {
|
||||
location: {
|
||||
index: params.index || 1
|
||||
},
|
||||
rows: params.rows,
|
||||
columns: params.columns
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
if (requests.length > 0) {
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} else {
|
||||
return `No updates specified` + TOOL_ARGS_PREFIX + JSON.stringify(params)
|
||||
}
|
||||
} catch (error) {
|
||||
return `Error updating document: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class InsertTextTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'insert_text',
|
||||
description: 'Insert text into a Google Docs document',
|
||||
schema: InsertTextSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const requests = [
|
||||
{
|
||||
insertText: {
|
||||
location: {
|
||||
index: params.index
|
||||
},
|
||||
text: params.text
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error inserting text: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class ReplaceTextTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'replace_text',
|
||||
description: 'Replace text in a Google Docs document',
|
||||
schema: ReplaceTextSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const requests = [
|
||||
{
|
||||
replaceAllText: {
|
||||
containsText: {
|
||||
text: params.replaceText,
|
||||
matchCase: params.matchCase
|
||||
},
|
||||
replaceText: params.newText
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error replacing text: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class AppendTextTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'append_text',
|
||||
description: 'Append text to the end of a Google Docs document',
|
||||
schema: AppendTextSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
// First get the document to find the end index
|
||||
const getEndpoint = `documents/${encodeURIComponent(params.documentId)}`
|
||||
const docResponse = await this.makeGoogleDocsRequest({ endpoint: getEndpoint, params: {} })
|
||||
const docData = JSON.parse(docResponse.split(TOOL_ARGS_PREFIX)[0])
|
||||
|
||||
// Get the end index of the document body
|
||||
const endIndex = docData.body.content[docData.body.content.length - 1].endIndex - 1
|
||||
|
||||
const requests = [
|
||||
{
|
||||
insertText: {
|
||||
location: {
|
||||
index: endIndex
|
||||
},
|
||||
text: params.text
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error appending text: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class GetTextContentTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'get_text_content',
|
||||
description: 'Get the text content from a Google Docs document',
|
||||
schema: GetTextContentSchema,
|
||||
baseUrl: '',
|
||||
method: 'GET',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}`
|
||||
const response = await this.makeGoogleDocsRequest({ endpoint, params })
|
||||
|
||||
// Extract and return just the text content
|
||||
const docData = JSON.parse(response.split(TOOL_ARGS_PREFIX)[0])
|
||||
let textContent = ''
|
||||
|
||||
const extractText = (element: any) => {
|
||||
if (element.paragraph) {
|
||||
element.paragraph.elements?.forEach((elem: any) => {
|
||||
if (elem.textRun) {
|
||||
textContent += elem.textRun.content
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
docData.body.content?.forEach(extractText)
|
||||
|
||||
return JSON.stringify({ textContent }) + TOOL_ARGS_PREFIX + JSON.stringify(params)
|
||||
} catch (error) {
|
||||
return `Error getting text content: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class InsertImageTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'insert_image',
|
||||
description: 'Insert an image into a Google Docs document',
|
||||
schema: InsertImageSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const requests = [
|
||||
{
|
||||
insertInlineImage: {
|
||||
location: {
|
||||
index: params.index
|
||||
},
|
||||
uri: params.imageUrl
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error inserting image: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class CreateTableTool extends BaseGoogleDocsTool {
|
||||
defaultParams: any
|
||||
|
||||
constructor(args: any) {
|
||||
const toolInput = {
|
||||
name: 'create_table',
|
||||
description: 'Create a table in a Google Docs document',
|
||||
schema: CreateTableSchema,
|
||||
baseUrl: '',
|
||||
method: 'POST',
|
||||
headers: {}
|
||||
}
|
||||
super({
|
||||
...toolInput,
|
||||
accessToken: args.accessToken
|
||||
})
|
||||
this.defaultParams = args.defaultParams || {}
|
||||
}
|
||||
|
||||
async _call(arg: any): Promise<string> {
|
||||
const params = { ...arg, ...this.defaultParams }
|
||||
|
||||
try {
|
||||
const requests = [
|
||||
{
|
||||
insertTable: {
|
||||
location: {
|
||||
index: params.index
|
||||
},
|
||||
rows: params.rows,
|
||||
columns: params.columns
|
||||
}
|
||||
}
|
||||
]
|
||||
|
||||
const endpoint = `documents/${encodeURIComponent(params.documentId)}:batchUpdate`
|
||||
const response = await this.makeGoogleDocsRequest({
|
||||
endpoint,
|
||||
method: 'POST',
|
||||
body: { requests },
|
||||
params
|
||||
})
|
||||
return response
|
||||
} catch (error) {
|
||||
return `Error creating table: ${error}`
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
export const createGoogleDocsTools = (args?: RequestParameters): DynamicStructuredTool[] => {
|
||||
const actions = args?.actions || []
|
||||
const tools: DynamicStructuredTool[] = []
|
||||
|
||||
if (actions.includes('createDocument') || actions.length === 0) {
|
||||
tools.push(new CreateDocumentTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('getDocument') || actions.length === 0) {
|
||||
tools.push(new GetDocumentTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('updateDocument') || actions.length === 0) {
|
||||
tools.push(new UpdateDocumentTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('insertText') || actions.length === 0) {
|
||||
tools.push(new InsertTextTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('replaceText') || actions.length === 0) {
|
||||
tools.push(new ReplaceTextTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('appendText') || actions.length === 0) {
|
||||
tools.push(new AppendTextTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('getTextContent') || actions.length === 0) {
|
||||
tools.push(new GetTextContentTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('insertImage') || actions.length === 0) {
|
||||
tools.push(new InsertImageTool(args))
|
||||
}
|
||||
|
||||
if (actions.includes('createTable') || actions.length === 0) {
|
||||
tools.push(new CreateTableTool(args))
|
||||
}
|
||||
|
||||
return tools
|
||||
}
|
||||
@@ -0,0 +1 @@
|
||||
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 48 48" width="96px" height="96px" fill-rule="evenodd" clip-rule="evenodd" baseProfile="basic"><linearGradient id="pg10I3OeSC0NOv22QZ6aWa" x1="-209.942" x2="-179.36" y1="-3.055" y2="27.526" gradientTransform="translate(208.979 6.006)" gradientUnits="userSpaceOnUse"><stop offset="0" stop-color="#55adfd"/><stop offset="1" stop-color="#438ffd"/></linearGradient><path fill="url(#pg10I3OeSC0NOv22QZ6aWa)" d="M39.001,13.999v27c0,1.105-0.896,2-2,2h-26 c-1.105,0-2-0.895-2-2v-34c0-1.104,0.895-2,2-2h19l2,7L39.001,13.999z"/><path fill="#fff" fill-rule="evenodd" d="M15.999,18.001v2.999 h17.002v-2.999H15.999z" clip-rule="evenodd"/><path fill="#fff" fill-rule="evenodd" d="M16.001,24.001v2.999 h17.002v-2.999H16.001z" clip-rule="evenodd"/><path fill="#fff" fill-rule="evenodd" d="M15.999,30.001v2.999 h12.001v-2.999H15.999z" clip-rule="evenodd"/><linearGradient id="pg10I3OeSC0NOv22QZ6aWb" x1="-197.862" x2="-203.384" y1="-4.632" y2=".89" gradientTransform="translate(234.385 12.109)" gradientUnits="userSpaceOnUse"><stop offset="0" stop-color="#427fdb"/><stop offset="1" stop-color="#0c52bb"/></linearGradient><path fill="url(#pg10I3OeSC0NOv22QZ6aWb)" d="M30.001,13.999l0.001-9l8.999,8.999L30.001,13.999z"/></svg>
|
||||
|
After Width: | Height: | Size: 1.2 KiB |
Reference in New Issue
Block a user