Syntax fix in ImageConversationChain.ts

vgees · Oct 23, 2023 · 752fb13 · 752fb13
1 parent 3c0d784
commit 752fb13
Showing 1 changed file with 27 additions and 27 deletions.
diff --git a/packages/components/nodes/chains/ImageConversationChain/ImageConversationChain.ts b/packages/components/nodes/chains/ImageConversationChain/ImageConversationChain.ts
@@ -1,35 +1,35 @@
-import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface';
-import { RetrievalQAChain } from 'langchain/chains';
-import { getBaseClasses } from '../../../src/utils';
-import { ConsoleCallbackHandler, additionalCallbacks } from '../../../src/handler';
+import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
+import { RetrievalQAChain } from 'langchain/chains'
+import { getBaseClasses } from '../../../src/utils'
+import { ConsoleCallbackHandler, additionalCallbacks } from '../../../src/handler'
 
 class ImageConversationChain_Chains implements INode {
-    label: string;
-    name: string;
-    version: number;
-    type: string;
-    icon: string;
-    category: string;
-    baseClasses: string[];
-    description: string;
-    inputs: INodeParams[];
+    label: string
+    name: string
+    version: number
+    type: string
+    icon: string
+    category: string
+    baseClasses: string[]
+    description: string
+    inputs: INodeParams[]
 
     constructor() {
-        this.label = 'Image Conversation Chain';
-        this.name = 'imageConversationChain';
-        this.version = 1.0;
-        this.type = 'ImageConversationChain';
-        this.icon = 'image.png';
-        this.category = 'Chains';
-        this.description = 'QA chain to answer a question based on the retrieved images';
-        this.baseClasses = [this.type, ...getBaseClasses(RetrievalQAChain)];
+        this.label = 'Image Conversation Chain'
+        this.name = 'imageConversationChain'
+        this.version = 1.0
+        this.type = 'ImageConversationChain'
+        this.icon = 'image.png'
+        this.category = 'Chains'
+        this.description = 'QA chain to answer a question based on the retrieved images'
+        this.baseClasses = [this.type, ...getBaseClasses(RetrievalQAChain)]
         this.inputs = [
             {
                 label: 'Image Description',
                 name: 'imageDescription',
                 type: 'description'
             }
-        ];
+        ]
     }
 
     async init(nodeData: INodeData): Promise<any> {
@@ -38,15 +38,15 @@ class ImageConversationChain_Chains implements INode {
     }
 
     async run(nodeData: INodeData, input: string, options: ICommonObject): Promise<string> {
-        const imageDescription = nodeData.inputs?.imageDescription as string;
-        const loggerHandler = new ConsoleCallbackHandler(options.logger);
-        const callbacks = await additionalCallbacks(nodeData, options);
+        const imageDescription = nodeData.inputs?.imageDescription as string
+        const loggerHandler = new ConsoleCallbackHandler(options.logger)
+        const callbacks = await additionalCallbacks(nodeData, options)
 
         // TODO: Any logic or operations you want to do with the image description can go here
         // For now, it simply returns the description
 
-        return imageDescription;
+        return imageDescription
     }
 }
 
-module.exports = { nodeClass: ImageConversationChain_Chains };
+module.exports = { nodeClass: ImageConversationChain_Chains }