Merge pull request #141 from FlowiseAI/feature/LocalAI

Bugfix/Add BaseChatModel class to LocalAI
2023-05-21 12:07:57 +01:00 · 2023-05-21 12:07:57 +01:00 · 2ffd26b147
parent 37f8d1f409 5b27339bc2
commit 2ffd26b147
1 changed files with 1 additions and 1 deletions
--- a/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts
+++ b/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts
@ -20,7 +20,7 @@ class ChatLocalAI_ChatModels implements INode {
        this.icon = 'localai.png'
        this.category = 'Chat Models'
        this.description = 'Use local LLMs like llama.cpp, gpt4all using LocalAI'
-        this.baseClasses = [this.type, ...getBaseClasses(OpenAIChat)]
+        this.baseClasses = [this.type, 'BaseChatModel', ...getBaseClasses(OpenAIChat)]
        this.inputs = [
            {
                label: 'Base Path',