add requests tools

packages/components/nodes/tools/RequestsGet/RequestsGet.ts

@@ -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)
     }
 }
 

packages/components/nodes/tools/RequestsGet/core.ts

@@ -0,0 +1,42 @@
+import { Tool } from 'langchain/tools'
+
+export const desc = `A portal to the internet. Use this when you need to get specific content from a website. 
+Input should be a  url (i.e. https://www.google.com). The output will be the text response of the GET request.`
+
+export interface Headers {
+    [key: string]: string
+}
+
+export interface RequestParameters {
+    headers?: Headers
+    url?: string
+    description?: string
+    maxOutputLength?: number
+}
+
+export class RequestsGetTool extends Tool {
+    name = 'requests_get'
+    url = ''
+    description = desc
+    maxOutputLength = 2000
+    headers = {}
+
+    constructor(args?: RequestParameters) {
+        super()
+        this.url = args?.url ?? this.url
+        this.headers = args?.headers ?? this.headers
+        this.description = args?.description ?? this.description
+        this.maxOutputLength = args?.maxOutputLength ?? this.maxOutputLength
+    }
+
+    /** @ignore */
+    async _call(input: string) {
+        const inputUrl = this.url ?? input
+        if (process.env.DEBUG === 'true') console.info(`Making GET API call to ${inputUrl}`)
+        const res = await fetch(inputUrl, {
+            headers: this.headers
+        })
+        const text = await res.text()
+        return text.slice(0, this.maxOutputLength)
+    }
+}