add fix to allow chaintool with llmchain

2026-06-28 17:01:00 +03:00 · 2023-07-07 16:54:19 +01:00
parent 4dd43fb2c4
commit c54cdd1940
2 changed files with 26 additions and 1 deletions
@@ -0,0 +1,25 @@
+import { DynamicTool, DynamicToolInput } from 'langchain/tools'
+import { BaseChain } from 'langchain/chains'
+
+export interface ChainToolInput extends Omit<DynamicToolInput, 'func'> {
+    chain: BaseChain
+}
+
+export class ChainTool extends DynamicTool {
+    chain: BaseChain
+
+    constructor({ chain, ...rest }: ChainToolInput) {
+        super({
+            ...rest,
+            func: async (input, runManager) => {
+                // To enable LLM Chain which has promptValues
+                if ((chain as any).prompt && (chain as any).prompt.promptValues) {
+                    const values = await chain.call((chain as any).prompt.promptValues, runManager?.getChild())
+                    return values?.text
+                }
+                return chain.run(input, runManager?.getChild())
+            }
+        })
+        this.chain = chain
+    }
+}