mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-28 17:01:00 +03:00
Henry Heng
@@ -0,0 +1,271 @@
|
||||
import { StripeAgentToolkit } from '@stripe/agent-toolkit/langchain'
|
||||
import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
|
||||
import { convertMultiOptionsToStringArray, getCredentialData, getCredentialParam } from '../../../src/utils'
|
||||
|
||||
class StripeTool_Tools implements INode {
|
||||
label: string
|
||||
name: string
|
||||
version: number
|
||||
description: string
|
||||
type: string
|
||||
icon: string
|
||||
category: string
|
||||
baseClasses: string[]
|
||||
credential: INodeParams
|
||||
inputs: INodeParams[]
|
||||
badge?: string
|
||||
|
||||
constructor() {
|
||||
this.label = 'StripeAgentTool'
|
||||
this.name = 'stripeAgentTool'
|
||||
this.version = 1.0
|
||||
this.type = 'stripeAgentTool'
|
||||
this.icon = 'stripe.png'
|
||||
this.category = 'Tools'
|
||||
this.description = 'Use Stripe Agent function calling for financial transactions'
|
||||
this.badge = 'BETA'
|
||||
this.inputs = [
|
||||
{
|
||||
label: 'Payment Links',
|
||||
name: 'paymentLinks',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Products',
|
||||
name: 'products',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Prices',
|
||||
name: 'prices',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Balance',
|
||||
name: 'balance',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Invoice Items',
|
||||
name: 'invoiceItems',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Invoices',
|
||||
name: 'invoices',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
},
|
||||
{
|
||||
label: 'Customers',
|
||||
name: 'customers',
|
||||
type: 'multiOptions',
|
||||
options: [
|
||||
{
|
||||
label: 'Create',
|
||||
name: 'create'
|
||||
},
|
||||
{
|
||||
label: 'Update',
|
||||
name: 'update'
|
||||
},
|
||||
{
|
||||
label: 'Read',
|
||||
name: 'read'
|
||||
}
|
||||
],
|
||||
optional: true,
|
||||
additionalParams: true
|
||||
}
|
||||
]
|
||||
this.credential = {
|
||||
label: 'Connect Credential',
|
||||
name: 'credential',
|
||||
type: 'credential',
|
||||
credentialNames: ['stripeApi']
|
||||
}
|
||||
this.baseClasses = [this.type, 'Tool']
|
||||
}
|
||||
|
||||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
|
||||
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
|
||||
const stripeApiToken = getCredentialParam('stripeApiToken', credentialData, nodeData)
|
||||
|
||||
const _paymentLinks = nodeData.inputs?.paymentLinks as string
|
||||
let paymentLinks: string[] = convertMultiOptionsToStringArray(_paymentLinks)
|
||||
|
||||
const _products = nodeData.inputs?.products as string
|
||||
let products: string[] = convertMultiOptionsToStringArray(_products)
|
||||
|
||||
const _prices = nodeData.inputs?.prices as string
|
||||
let prices: string[] = convertMultiOptionsToStringArray(_prices)
|
||||
|
||||
const _balance = nodeData.inputs?.balance as string
|
||||
let balance: string[] = convertMultiOptionsToStringArray(_balance)
|
||||
|
||||
const _invoiceItems = nodeData.inputs?.invoiceItems as string
|
||||
let invoiceItems: string[] = convertMultiOptionsToStringArray(_invoiceItems)
|
||||
|
||||
const _invoices = nodeData.inputs?.invoices as string
|
||||
let invoices: string[] = convertMultiOptionsToStringArray(_invoices)
|
||||
|
||||
const _customers = nodeData.inputs?.customers as string
|
||||
let customers: string[] = convertMultiOptionsToStringArray(_customers)
|
||||
|
||||
const actionObj: any = {}
|
||||
if (paymentLinks.length > 0) {
|
||||
actionObj['paymentLinks'] = {}
|
||||
if (paymentLinks.includes('create')) actionObj['paymentLinks'].create = true
|
||||
if (paymentLinks.includes('read')) actionObj['paymentLinks'].read = true
|
||||
if (paymentLinks.includes('update')) actionObj['paymentLinks'].update = true
|
||||
}
|
||||
if (products.length > 0) {
|
||||
actionObj['products'] = {}
|
||||
if (products.includes('create')) actionObj['products'].create = true
|
||||
if (products.includes('read')) actionObj['products'].read = true
|
||||
if (products.includes('update')) actionObj['products'].update = true
|
||||
}
|
||||
if (prices.length > 0) {
|
||||
actionObj['prices'] = {}
|
||||
if (prices.includes('create')) actionObj['prices'].create = true
|
||||
if (prices.includes('read')) actionObj['prices'].read = true
|
||||
if (prices.includes('update')) actionObj['prices'].update = true
|
||||
}
|
||||
if (balance.length > 0) {
|
||||
actionObj['balance'] = {}
|
||||
if (balance.includes('create')) actionObj['balance'].create = true
|
||||
if (balance.includes('read')) actionObj['balance'].read = true
|
||||
if (balance.includes('update')) actionObj['balance'].update = true
|
||||
}
|
||||
if (invoiceItems.length > 0) {
|
||||
actionObj['invoiceItems'] = {}
|
||||
if (invoiceItems.includes('create')) actionObj['invoiceItems'].create = true
|
||||
if (invoiceItems.includes('read')) actionObj['invoiceItems'].read = true
|
||||
if (invoiceItems.includes('update')) actionObj['invoiceItems'].update = true
|
||||
}
|
||||
if (invoices.length > 0) {
|
||||
actionObj['invoices'] = {}
|
||||
if (invoices.includes('create')) actionObj['invoices'].create = true
|
||||
if (invoices.includes('read')) actionObj['invoices'].read = true
|
||||
if (invoices.includes('update')) actionObj['invoices'].update = true
|
||||
}
|
||||
if (customers.length > 0) {
|
||||
actionObj['customers'] = {}
|
||||
if (customers.includes('create')) actionObj['customers'].create = true
|
||||
if (customers.includes('read')) actionObj['customers'].read = true
|
||||
if (customers.includes('update')) actionObj['customers'].update = true
|
||||
}
|
||||
|
||||
const stripeAgentToolkit = new StripeAgentToolkit({
|
||||
secretKey: stripeApiToken,
|
||||
configuration: {
|
||||
actions: actionObj
|
||||
}
|
||||
})
|
||||
|
||||
const stripeTool = stripeAgentToolkit.getTools()
|
||||
for (const tool of stripeTool) {
|
||||
// convert tool name into small letter, and space to underscore, ex: Create Payment Link => create_payment_link
|
||||
tool.name = tool.name.split(' ').join('_').toLowerCase()
|
||||
}
|
||||
|
||||
return stripeTool
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { nodeClass: StripeTool_Tools }
|
||||
Binary file not shown.
|
After Width: | Height: | Size: 68 KiB |
Reference in New Issue
Block a user