69 lines
1.9 KiB
JavaScript
69 lines
1.9 KiB
JavaScript
const { Tool } = require("@langchain/core/tools");
|
|
const { JLINCTracer } = require("../tracer/index.js");
|
|
const { authDecide, authInvoke, getDescription, getName } = require("./common.js");
|
|
|
|
/**
|
|
* @typedef {import('./common').JLINCConfig} JLINCConfig
|
|
*/
|
|
|
|
/**
|
|
* @typedef {Object} JLINCAuthToolFields
|
|
* @property {JLINCConfig} config
|
|
* @property {Tool} jlincAuthDecision
|
|
* @property {Tool} targetAuthorized
|
|
* @property {Tool|null} targetNotAuthorized
|
|
*/
|
|
|
|
class JLINCAuthTool extends Tool {
|
|
/**
|
|
* @param {JLINCAuthToolFields} fields
|
|
*/
|
|
constructor({ config, jlincAuthDecision, targetAuthorized, targetNotAuthorized }) {
|
|
super({
|
|
name: "jlinc_auth_tool",
|
|
description: "", // overridden dynamically
|
|
});
|
|
Object.defineProperty(this, "name", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: getName(targetAuthorized, targetNotAuthorized)
|
|
});
|
|
Object.defineProperty(this, "description", {
|
|
enumerable: true,
|
|
configurable: true,
|
|
writable: true,
|
|
value: getDescription(targetAuthorized, targetNotAuthorized, 'Tool')
|
|
});
|
|
/** @type {JLINCConfig} */
|
|
this.config = config;
|
|
/** @type {Tool} */
|
|
this.jlincAuthDecision = jlincAuthDecision;
|
|
/** @type {Tool} */
|
|
this.targetAuthorized = targetAuthorized;
|
|
/** @type {Tool|null} */
|
|
this.targetNotAuthorized = targetNotAuthorized;
|
|
/** @type {JLINCTracer} */
|
|
this.tracer = new JLINCTracer(config);
|
|
/** @type {string} */
|
|
this.authType = 'Tool';
|
|
}
|
|
|
|
/**
|
|
* @template TInput
|
|
* @template {undefined | ToolRunnableConfig} TConfig
|
|
* @template TOutput
|
|
*
|
|
* @param {TInput} input
|
|
* @param {TConfig} [config]
|
|
* @returns {Promise<ToolReturnType<TInput, TConfig, TOutput>>}
|
|
*/
|
|
async invoke(input, runManager) {
|
|
return await authInvoke(this, input);
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
JLINCAuthTool,
|
|
};
|