fix(rag-ai): update model type

RoadieHQ · Nov 13, 2024 · 14057d9 · 14057d9
1 parent 1cca13c
commit 14057d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plugins/backend/rag-ai-backend/src/plugin.ts b/plugins/backend/rag-ai-backend/src/plugin.ts
@@ -29,6 +29,7 @@ import {
   modelExtensionPoint,
 } from '@roadiehq/rag-ai-node';
 import { BaseLLM } from '@langchain/core/language_models/llms';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 
 /**
  * Rag AI backend plugin
@@ -40,7 +41,7 @@ export const ragAiPlugin = createBackendPlugin({
   register(env) {
     let augmentationIndexer: AugmentationIndexer | undefined;
     let retrievalPipeline: RetrievalPipeline | undefined;
-    let model: BaseLLM | undefined;
+    let model: BaseLLM | BaseChatModel | undefined;
 
     env.registerExtensionPoint(augmentationIndexerExtensionPoint, {
       setAugmentationIndexer(indexer) {