mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-29 03:01:10 +03:00
Feature/add composio tool (#3722)
* feat: add composio tool * fix: improve error handling & field description * update composio tools for refresh and sorting --------- Co-authored-by: Henry <hzj94@hotmail.com>
This commit is contained in:
Abhishek Patil
@@ -0,0 +1,225 @@
|
||||
import { Tool } from '@langchain/core/tools'
|
||||
import { ICommonObject, INode, INodeData, INodeOptionsValue, INodeParams } from '../../../src/Interface'
|
||||
import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src/utils'
|
||||
import { LangchainToolSet } from 'composio-core'
|
||||
|
||||
class ComposioTool extends Tool {
|
||||
name = 'composio'
|
||||
description = 'Tool for interacting with Composio applications and performing actions'
|
||||
toolset: any
|
||||
appName: string
|
||||
actions: string[]
|
||||
|
||||
constructor(toolset: any, appName: string, actions: string[]) {
|
||||
super()
|
||||
this.toolset = toolset
|
||||
this.appName = appName
|
||||
this.actions = actions
|
||||
}
|
||||
|
||||
async _call(input: string): Promise<string> {
|
||||
try {
|
||||
return `Executed action on ${this.appName} with input: ${input}`
|
||||
} catch (error) {
|
||||
return 'Failed to execute action'
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
class Composio_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 = 'Composio'
|
||||
this.name = 'composio'
|
||||
this.version = 1.0
|
||||
this.type = 'Composio'
|
||||
this.icon = 'composio.svg'
|
||||
this.category = 'Tools'
|
||||
this.description = 'Toolset with over 250+ Apps for building AI-powered applications'
|
||||
this.baseClasses = [this.type, ...getBaseClasses(ComposioTool)]
|
||||
this.credential = {
|
||||
label: 'Connect Credential',
|
||||
name: 'credential',
|
||||
type: 'credential',
|
||||
credentialNames: ['composioApi']
|
||||
}
|
||||
this.inputs = [
|
||||
{
|
||||
label: 'App Name',
|
||||
name: 'appName',
|
||||
type: 'asyncOptions',
|
||||
loadMethod: 'listApps',
|
||||
description: 'Select the app to connect with',
|
||||
refresh: true
|
||||
},
|
||||
{
|
||||
label: 'Auth Status',
|
||||
name: 'authStatus',
|
||||
type: 'asyncOptions',
|
||||
loadMethod: 'authStatus',
|
||||
placeholder: 'Connection status will appear here',
|
||||
refresh: true
|
||||
},
|
||||
{
|
||||
label: 'Actions to Use',
|
||||
name: 'actions',
|
||||
type: 'asyncOptions',
|
||||
loadMethod: 'listActions',
|
||||
description: 'Select the actions you want to use',
|
||||
refresh: true
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
//@ts-ignore
|
||||
loadMethods = {
|
||||
listApps: async (nodeData: INodeData, options?: ICommonObject): Promise<INodeOptionsValue[]> => {
|
||||
try {
|
||||
const credentialData = await getCredentialData(nodeData.credential ?? '', options ?? {})
|
||||
const composioApiKey = getCredentialParam('composioApi', credentialData, nodeData)
|
||||
|
||||
if (!composioApiKey) {
|
||||
return [
|
||||
{
|
||||
label: 'API Key Required',
|
||||
name: 'placeholder',
|
||||
description: 'Enter Composio API key in the credential field'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
const toolset = new LangchainToolSet({ apiKey: composioApiKey })
|
||||
const apps = await toolset.client.apps.list()
|
||||
apps.sort((a: any, b: any) => a.name.localeCompare(b.name))
|
||||
|
||||
return apps.map(({ name, ...rest }) => ({
|
||||
label: name.toUpperCase(),
|
||||
name: name,
|
||||
description: rest.description || name
|
||||
}))
|
||||
} catch (error) {
|
||||
console.error('Error loading apps:', error)
|
||||
return [
|
||||
{
|
||||
label: 'Error Loading Apps',
|
||||
name: 'error',
|
||||
description: 'Failed to load apps. Please check your API key and try again'
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
listActions: async (nodeData: INodeData, options?: ICommonObject): Promise<INodeOptionsValue[]> => {
|
||||
try {
|
||||
const credentialData = await getCredentialData(nodeData.credential ?? '', options ?? {})
|
||||
const composioApiKey = getCredentialParam('composioApi', credentialData, nodeData)
|
||||
const appName = nodeData.inputs?.appName as string
|
||||
|
||||
if (!composioApiKey) {
|
||||
return [
|
||||
{
|
||||
label: 'API Key Required',
|
||||
name: 'placeholder',
|
||||
description: 'Enter Composio API key in the credential field'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
if (!appName) {
|
||||
return [
|
||||
{
|
||||
label: 'Select an App first',
|
||||
name: 'placeholder',
|
||||
description: 'Select an app from the dropdown to view available actions'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
const toolset = new LangchainToolSet({ apiKey: composioApiKey })
|
||||
const actions = await toolset.getTools({ apps: [appName] })
|
||||
actions.sort((a: any, b: any) => a.name.localeCompare(b.name))
|
||||
|
||||
return actions.map(({ name, ...rest }) => ({
|
||||
label: name.toUpperCase(),
|
||||
name: name,
|
||||
description: rest.description || name
|
||||
}))
|
||||
} catch (error) {
|
||||
console.error('Error loading actions:', error)
|
||||
return [
|
||||
{
|
||||
label: 'Error Loading Actions',
|
||||
name: 'error',
|
||||
description: 'Failed to load actions. Please check your API key and try again'
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
authStatus: async (nodeData: INodeData, options?: ICommonObject): Promise<INodeOptionsValue[]> => {
|
||||
const credentialData = await getCredentialData(nodeData.credential ?? '', options ?? {})
|
||||
const composioApiKey = getCredentialParam('composioApi', credentialData, nodeData)
|
||||
const appName = nodeData.inputs?.appName as string
|
||||
|
||||
if (!composioApiKey) {
|
||||
return [
|
||||
{
|
||||
label: 'API Key Required',
|
||||
name: 'placeholder',
|
||||
description: 'Enter Composio API key in the credential field'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
if (!appName) {
|
||||
return [
|
||||
{
|
||||
label: 'Select an App first',
|
||||
name: 'placeholder',
|
||||
description: 'Select an app from the dropdown to view available actions'
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
const toolset = new LangchainToolSet({ apiKey: composioApiKey })
|
||||
const authStatus = await toolset.client.getEntity('default').getConnection({ app: appName.toLowerCase() })
|
||||
return [
|
||||
{
|
||||
label: authStatus ? 'Connected' : 'Not Connected',
|
||||
name: authStatus ? 'Connected' : 'Not Connected',
|
||||
description: authStatus ? 'Selected app has an active connection' : 'Please connect the app on app.composio.dev'
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
|
||||
if (!nodeData.inputs) nodeData.inputs = {}
|
||||
|
||||
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
|
||||
const composioApiKey = getCredentialParam('composioApi', credentialData, nodeData)
|
||||
|
||||
if (!composioApiKey) {
|
||||
nodeData.inputs = {
|
||||
appName: undefined,
|
||||
authStatus: '',
|
||||
actions: []
|
||||
}
|
||||
throw new Error('API Key Required')
|
||||
}
|
||||
|
||||
const toolset = new LangchainToolSet({ apiKey: composioApiKey })
|
||||
const tools = await toolset.getTools({ actions: [nodeData.inputs?.actions as string] })
|
||||
return tools
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { nodeClass: Composio_Tools }
|
||||
Reference in New Issue
Block a user