27 lines
828 B
JavaScript
27 lines
828 B
JavaScript
|
import { DynamicTool } from "@langchain/core/tools";
|
||
|
|
/**
|
||
|
|
* @deprecated Wrap in a DynamicTool instead.
|
||
|
|
* Class that extends DynamicTool for creating tools that can run chains.
|
||
|
|
* Takes an instance of a class that extends BaseChain as a parameter in
|
||
|
|
* its constructor and uses it to run the chain when its 'func' method is
|
||
|
|
* called.
|
||
|
|
*/
|
||
|
|
export class ChainTool extends DynamicTool {
|
||
|
|
static lc_name() {
|
||
|
|
return "ChainTool";
|
||
|
|
}
|
||
|
|
constructor({ chain, ...rest }) {
|
||
|
|
super({
|
||
|
|
...rest,
|
||
|
|
func: async (input, runManager) => chain.run(input, runManager?.getChild()),
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "chain", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: void 0
|
||
|
|
});
|
||
|
|
this.chain = chain;
|
||
|
|
}
|
||
|
|
}
|