Bugfix/Restore Requests Tool (#2513)

restore requests tool
2026-06-29 05:01:10 +03:00 · 2024-05-29 23:39:11 +01:00
parent 2878af69e4
commit 48ac815f8e
6 changed files with 289 additions and 0 deletions
@@ -0,0 +1,72 @@
+import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses } from '../../../src/utils'
+import { desc, RequestParameters, RequestsGetTool } from './core'
+
+class RequestsGet_Tools implements INode {
+    label: string
+    name: string
+    version: number
+    description: string
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    inputs: INodeParams[]
+
+    constructor() {
+        this.label = 'Requests Get'
+        this.name = 'requestsGet'
+        this.version = 1.0
+        this.type = 'RequestsGet'
+        this.icon = 'requestsget.svg'
+        this.category = 'Tools'
+        this.description = 'Execute HTTP GET requests'
+        this.baseClasses = [this.type, ...getBaseClasses(RequestsGetTool)]
+        this.inputs = [
+            {
+                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
+            },
+            {
+                label: 'Headers',
+                name: 'headers',
+                type: 'json',
+                additionalParams: true,
+                optional: true
+            }
+        ]
+    }
+
+    async init(nodeData: INodeData): Promise<any> {
+        const headers = nodeData.inputs?.headers as string
+        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)
+            obj.headers = parsedHeaders
+        }
+
+        return new RequestsGetTool(obj)
+    }
+}
+
+module.exports = { nodeClass: RequestsGet_Tools }
@@ -0,0 +1,46 @@
+import fetch from 'node-fetch'
+import { Tool } from '@langchain/core/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 : this.url
+
+        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)
+    }
+}
@@ -0,0 +1,6 @@
+<svg width="32" height="32" viewBox="0 0 32 32" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M10.5 20.5C10.5 20.5 10 20 9 20C7.067 20 6 21.567 6 23.5C6 25.433 7.067 27 9 27C10 27 10.7037 26.4812 11 26V24H10" stroke="#110000" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+<path d="M18.5 20H14V27H18.5M14 23.5H17.5" stroke="black" stroke-width="1.8" stroke-linecap="round" stroke-linejoin="round"/>
+<path d="M23.5 27V20M21 20H26" stroke="black" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+<path d="M19.1112 15.2076L17.482 13.3556C15.4506 14.3228 13.0464 14.0464 11.477 12.477C10.1962 11.1962 9.77656 9.35939 10.1913 7.62299C10.3492 6.9619 11.1601 6.82676 11.6407 7.30737L13.5196 9.18628C14.1962 9.86283 15.3416 9.81433 16.078 9.07795C16.8143 8.34157 16.8628 7.19616 16.1863 6.51961L14.3074 4.64071C13.8268 4.16009 13.9619 3.34916 14.623 3.19127C16.3594 2.77656 18.1962 3.19622 19.477 4.477C21.0464 6.04639 21.3228 8.45065 20.3556 10.482L22.2076 12.1112" stroke="black" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+</svg>