From 1ccbd4cff3f0de5cc7cfb2ba7ce226f79b179f1a Mon Sep 17 00:00:00 2001 From: AntonyDevs Date: Wed, 27 Nov 2024 10:13:48 -0500 Subject: [PATCH] refactor(ragToolkit): rename llmInstance to llm for consistency; update query in tool stories --- packages/tools/src/_utils/rag/ragToolkit.js | 2 +- packages/tools/src/simple-rag/tool.stories.jsx | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/tools/src/_utils/rag/ragToolkit.js b/packages/tools/src/_utils/rag/ragToolkit.js index 339e730..cd7c9c5 100644 --- a/packages/tools/src/_utils/rag/ragToolkit.js +++ b/packages/tools/src/_utils/rag/ragToolkit.js @@ -86,7 +86,7 @@ class RAGToolkit { ); const chain = await createStuffDocumentsChain({ - llmInstance: this.llmInstance, + llm: this.llmInstance, prompt: promptTemplate, outputParser: new StringOutputParser(), }); diff --git a/packages/tools/src/simple-rag/tool.stories.jsx b/packages/tools/src/simple-rag/tool.stories.jsx index 54ff10c..39e1bfc 100644 --- a/packages/tools/src/simple-rag/tool.stories.jsx +++ b/packages/tools/src/simple-rag/tool.stories.jsx @@ -100,7 +100,7 @@ export const Default = { args: { toolInstance: simpleRAGTool, callParams: { - query: 'What is the capital of France?', + query: 'What is Kaiban?', content, }, },