From b9d1d75d6a043770cf3ca2f3db7781746423ea8f Mon Sep 17 00:00:00 2001
From: Keith Kacsh <keith@kacsh.com>
Date: Sat, 13 Jan 2024 19:14:45 -0700
Subject: [PATCH] Fixing naming, handling embeddings for LocalAI also

---
 ....credential.ts => LocalAIApi.credential.ts} |  4 ++--
 .../chatmodels/ChatLocalAI/ChatLocalAI.ts      |  6 +++---
 .../LocalAIEmbedding/LocalAIEmbedding.ts       | 18 ++++++++++++++++--
 3 files changed, 21 insertions(+), 7 deletions(-)
 rename packages/components/credentials/{LcoalAIApi.credential.ts => LocalAIApi.credential.ts} (86%)

diff --git a/packages/components/credentials/LcoalAIApi.credential.ts b/packages/components/credentials/LocalAIApi.credential.ts
similarity index 86%
rename from packages/components/credentials/LcoalAIApi.credential.ts
rename to packages/components/credentials/LocalAIApi.credential.ts
index 624e07fa4..4aafe040d 100644
--- a/packages/components/credentials/LcoalAIApi.credential.ts
+++ b/packages/components/credentials/LocalAIApi.credential.ts
@@ -8,12 +8,12 @@ class LocalAIApi implements INodeCredential {
 
     constructor() {
         this.label = 'LocalAI API'
-        this.name = 'LocalAIApi'
+        this.name = 'localAIApi'
         this.version = 1.0
         this.inputs = [
             {
                 label: 'LocalAI Api Key',
-                name: 'LocalAIApiKey',
+                name: 'localAIApiKey',
                 type: 'password'
             }
         ]
diff --git a/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts b/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts
index c44f03ce1..f2825d0d3 100644
--- a/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts
+++ b/packages/components/nodes/chatmodels/ChatLocalAI/ChatLocalAI.ts
@@ -30,7 +30,7 @@ class ChatLocalAI_ChatModels implements INode {
             label: 'Connect Credential',
             name: 'credential',
             type: 'credential',
-            credentialNames: ['LocalAIApi'],
+            credentialNames: ['localAIApi'],
             optional: true
         }
         this.inputs = [
@@ -95,11 +95,11 @@ class ChatLocalAI_ChatModels implements INode {
         const timeout = nodeData.inputs?.timeout as string
         const basePath = nodeData.inputs?.basePath as string
         const credentialData = await getCredentialData(nodeData.credential ?? '', options)
-        const localAIApiKey = getCredentialParam('LocalAIApiKey', credentialData, nodeData)
+        const localAIApiKey = getCredentialParam('localAIApiKey', credentialData, nodeData)
 
         const cache = nodeData.inputs?.cache as BaseCache
 
-        const obj: Partial<OpenAIChatInput> & BaseLLMParams & { localAIApiKey?: string } = {
+        const obj: Partial<OpenAIChatInput> & BaseLLMParams & { openAIApiKey?: string } = {
             temperature: parseFloat(temperature),
             modelName,
             openAIApiKey: 'sk-'
diff --git a/packages/components/nodes/embeddings/LocalAIEmbedding/LocalAIEmbedding.ts b/packages/components/nodes/embeddings/LocalAIEmbedding/LocalAIEmbedding.ts
index 557e35d68..24efaf8c6 100644
--- a/packages/components/nodes/embeddings/LocalAIEmbedding/LocalAIEmbedding.ts
+++ b/packages/components/nodes/embeddings/LocalAIEmbedding/LocalAIEmbedding.ts
@@ -1,4 +1,5 @@
-import { INode, INodeData, INodeParams } from '../../../src/Interface'
+import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getCredentialData, getCredentialParam } from '../../../src/utils'
 import { OpenAIEmbeddings, OpenAIEmbeddingsParams } from 'langchain/embeddings/openai'
 
 class LocalAIEmbedding_Embeddings implements INode {
@@ -10,6 +11,7 @@ class LocalAIEmbedding_Embeddings implements INode {
     category: string
     description: string
     baseClasses: string[]
+    credential: INodeParams
     inputs: INodeParams[]
 
     constructor() {
@@ -21,6 +23,13 @@ class LocalAIEmbedding_Embeddings implements INode {
         this.category = 'Embeddings'
         this.description = 'Use local embeddings models like llama.cpp'
         this.baseClasses = [this.type, 'Embeddings']
+        this.credential = {
+            label: 'Connect Credential',
+            name: 'credential',
+            type: 'credential',
+            credentialNames: ['localAIApi'],
+            optional: true
+        }
         this.inputs = [
             {
                 label: 'Base Path',
@@ -37,15 +46,20 @@ class LocalAIEmbedding_Embeddings implements INode {
         ]
     }
 
-    async init(nodeData: INodeData): Promise<any> {
+    async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
         const modelName = nodeData.inputs?.modelName as string
         const basePath = nodeData.inputs?.basePath as string
 
+        const credentialData = await getCredentialData(nodeData.credential ?? '', options)
+        const localAIApiKey = getCredentialParam('localAIApiKey', credentialData, nodeData)
+
         const obj: Partial<OpenAIEmbeddingsParams> & { openAIApiKey?: string } = {
             modelName,
             openAIApiKey: 'sk-'
         }
 
+        if (localAIApiKey) obj.openAIApiKey = localAIApiKey
+
         const model = new OpenAIEmbeddings(obj, { basePath })
 
         return model