add requests tools

This commit is contained in:
Henry
2023-05-23 00:01:16 +01:00
parent d3de0d33d7
commit aac9160a9f
7 changed files with 199 additions and 29 deletions
@@ -1,6 +1,6 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { getBaseClasses } from '../../../src/utils'
import { RequestsPostTool } from 'langchain/tools'
import { RequestParameters, desc, RequestsPostTool } from './core'
class RequestsPost_Tools implements INode {
label: string
@@ -22,9 +22,31 @@ class RequestsPost_Tools implements INode {
this.baseClasses = [this.type, ...getBaseClasses(RequestsPostTool)]
this.inputs = [
{
label: 'Max Output Length',
name: 'maxOutputLength',
type: 'number',
label: 'URL',
name: 'url',
type: 'string',
description:
'Agent will make call to this exact URL. If not specified, agent will try to figure out itself from AIPlugin if provided',
additionalParams: true,
optional: true
},
{
label: 'Body',
name: 'body',
type: 'json',
description:
'JSON body for the POST request. If not specified, agent will try to figure out itself from AIPlugin if provided',
additionalParams: true,
optional: true
},
{
label: 'Description',
name: 'description',
type: 'string',
rows: 4,
default: desc,
description: 'Acts like a prompt to tell agent when it should use this tool',
additionalParams: true,
optional: true
},
{
@@ -39,18 +61,23 @@ class RequestsPost_Tools implements INode {
async init(nodeData: INodeData): Promise<any> {
const headers = nodeData.inputs?.headers as string
const maxOutputLength = nodeData.inputs?.maxOutputLength as string
const obj: any = {}
if (maxOutputLength) {
obj.maxOutputLength = parseInt(maxOutputLength, 10)
}
const url = nodeData.inputs?.url as string
const description = nodeData.inputs?.description as string
const body = nodeData.inputs?.body as string
const obj: RequestParameters = {}
if (url) obj.url = url
if (description) obj.description = description
if (headers) {
const parsedHeaders = typeof headers === 'object' ? headers : JSON.parse(headers)
return Object.keys(obj).length ? new RequestsPostTool(parsedHeaders, obj) : new RequestsPostTool(parsedHeaders)
obj.headers = parsedHeaders
}
return Object.keys(obj).length ? new RequestsPostTool(undefined, obj) : new RequestsPostTool()
if (body) {
const parsedBody = typeof body === 'object' ? body : JSON.parse(body)
obj.body = parsedBody
}
return new RequestsPostTool(obj)
}
}