agsamantha/node_modules/langchain/dist/tools/requests.cjs

114 lines
3.9 KiB
JavaScript
Raw Permalink Normal View History

2024-10-02 15:15:21 -05:00
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.RequestsPostTool = exports.RequestsGetTool = void 0;
const tools_1 = require("@langchain/core/tools");
/**
* Class for making GET requests. Extends the Tool class and implements
* the RequestTool interface. The input should be a URL string, and the
* output will be the text response of the GET request.
*/
class RequestsGetTool extends tools_1.Tool {
static lc_name() {
return "RequestsGetTool";
}
constructor(headers = {}, { maxOutputLength } = {}) {
super(...arguments);
Object.defineProperty(this, "headers", {
enumerable: true,
configurable: true,
writable: true,
value: headers
});
Object.defineProperty(this, "name", {
enumerable: true,
configurable: true,
writable: true,
value: "requests_get"
});
Object.defineProperty(this, "maxOutputLength", {
enumerable: true,
configurable: true,
writable: true,
value: 2000
});
Object.defineProperty(this, "description", {
enumerable: true,
configurable: true,
writable: true,
value: `A portal to the internet. Use this when you need to get specific content from a website.
Input should be a url string (i.e. "https://www.google.com"). The output will be the text response of the GET request.`
});
this.maxOutputLength = maxOutputLength ?? this.maxOutputLength;
}
/** @ignore */
async _call(input) {
const res = await fetch(input, {
headers: this.headers,
});
const text = await res.text();
return text.slice(0, this.maxOutputLength);
}
}
exports.RequestsGetTool = RequestsGetTool;
/**
* Class for making POST requests. Extends the Tool class and implements
* the RequestTool interface. The input should be a JSON string with two
* keys: 'url' and 'data'. The output will be the text response of the
* POST request.
*/
class RequestsPostTool extends tools_1.Tool {
static lc_name() {
return "RequestsPostTool";
}
constructor(headers = {}, { maxOutputLength } = {}) {
super(...arguments);
Object.defineProperty(this, "headers", {
enumerable: true,
configurable: true,
writable: true,
value: headers
});
Object.defineProperty(this, "name", {
enumerable: true,
configurable: true,
writable: true,
value: "requests_post"
});
Object.defineProperty(this, "maxOutputLength", {
enumerable: true,
configurable: true,
writable: true,
value: Infinity
});
Object.defineProperty(this, "description", {
enumerable: true,
configurable: true,
writable: true,
value: `Use this when you want to POST to a website.
Input should be a json string with two keys: "url" and "data".
The value of "url" should be a string, and the value of "data" should be a dictionary of
key-value pairs you want to POST to the url as a JSON body.
Be careful to always use double quotes for strings in the json string
The output will be the text response of the POST request.`
});
this.maxOutputLength = maxOutputLength ?? this.maxOutputLength;
}
/** @ignore */
async _call(input) {
try {
const { url, data } = JSON.parse(input);
const res = await fetch(url, {
method: "POST",
headers: this.headers,
body: JSON.stringify(data),
});
const text = await res.text();
return text.slice(0, this.maxOutputLength);
}
catch (error) {
return `${error}`;
}
}
}
exports.RequestsPostTool = RequestsPostTool;