|
import fetch from 'node-fetch' |
|
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 |
|
} |
|
|
|
|
|
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) |
|
} |
|
} |
|
|