refactor(ragToolkit): rename llmInstance to llm for consistency; upda…

…te query in tool stories
kaiban-ai · Nov 27, 2024 · 1ccbd4c · 1ccbd4c
1 parent 8dd2e08
commit 1ccbd4c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -100,7 +100,7 @@ export const Default = {
   args: {
     toolInstance: simpleRAGTool,
     callParams: {
-      query: 'What is the capital of France?',
+      query: 'What is Kaiban?',
       content,
     },
   },