add authentication support
This commit is contained in:
68
dist/auth/JLINCAuthTool.js
vendored
Normal file
68
dist/auth/JLINCAuthTool.js
vendored
Normal file
@@ -0,0 +1,68 @@
|
||||
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,
|
||||
};
|
||||
Reference in New Issue
Block a user