mirror of
https://github.com/farcasclaudiu/Flowise.git
synced 2026-06-28 21:00:58 +03:00
add requests tools
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import { INode, INodeData, INodeParams } from '../../../src/Interface'
|
||||
import { getBaseClasses } from '../../../src/utils'
|
||||
import { RequestsGetTool } from 'langchain/tools'
|
||||
import { desc, RequestParameters, RequestsGetTool } from './core'
|
||||
|
||||
class RequestsGet_Tools implements INode {
|
||||
label: string
|
||||
@@ -22,9 +22,22 @@ class RequestsGet_Tools implements INode {
|
||||
this.baseClasses = [this.type, ...getBaseClasses(RequestsGetTool)]
|
||||
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: '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 +52,18 @@ class RequestsGet_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 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 RequestsGetTool(parsedHeaders, obj) : new RequestsGetTool(parsedHeaders)
|
||||
obj.headers = parsedHeaders
|
||||
}
|
||||
return Object.keys(obj).length ? new RequestsGetTool(undefined, obj) : new RequestsGetTool()
|
||||
|
||||
return new RequestsGetTool(obj)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user