update langchain version

packages/components/nodes/moderation/OpenAIModeration/OpenAIModeration.ts

@@ -1,5 +1,5 @@
-import { INode, INodeData, INodeParams } from '../../../src/Interface'
-import { getBaseClasses } from '../../../src'
+import { ICommonObject, INode, INodeData, INodeParams } from '../../../src/Interface'
+import { getBaseClasses, getCredentialData, getCredentialParam } from '../../../src'
 import { Moderation } from '../Moderation'
 import { OpenAIModerationRunner } from './OpenAIModerationRunner'
 
@@ -12,6 +12,7 @@ class OpenAIModeration implements INode {
     icon: string
     category: string
     baseClasses: string[]
+    credential: INodeParams
     inputs: INodeParams[]
 
     constructor() {
@@ -23,6 +24,12 @@ class OpenAIModeration implements INode {
         this.category = 'Moderation'
         this.description = 'Check whether content complies with OpenAI usage policies.'
         this.baseClasses = [this.type, ...getBaseClasses(Moderation)]
+        this.credential = {
+            label: 'Connect Credential',
+            name: 'credential',
+            type: 'credential',
+            credentialNames: ['openAIApi']
+        }
         this.inputs = [
             {
                 label: 'Error Message',
@@ -35,8 +42,11 @@ class OpenAIModeration implements INode {
         ]
     }
 
-    async init(nodeData: INodeData): Promise<any> {
-        const runner = new OpenAIModerationRunner()
+    async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
+        const credentialData = await getCredentialData(nodeData.credential ?? '', options)
+        const openAIApiKey = getCredentialParam('openAIApiKey', credentialData, nodeData)
+
+        const runner = new OpenAIModerationRunner(openAIApiKey)
         const moderationErrorMessage = nodeData.inputs?.moderationErrorMessage as string
         if (moderationErrorMessage) runner.setErrorMessage(moderationErrorMessage)
         return runner

packages/components/nodes/moderation/OpenAIModeration/OpenAIModerationRunner.ts

@@ -1,18 +1,21 @@
 import { Moderation } from '../Moderation'
-import { BaseLanguageModel } from 'langchain/base_language'
 import { OpenAIModerationChain } from 'langchain/chains'
 
 export class OpenAIModerationRunner implements Moderation {
+    private openAIApiKey = ''
     private moderationErrorMessage: string = "Text was found that violates OpenAI's content policy."
 
-    async checkForViolations(llm: BaseLanguageModel, input: string): Promise<string> {
-        const openAIApiKey = (llm as any).openAIApiKey
-        if (!openAIApiKey) {
+    constructor(openAIApiKey: string) {
+        this.openAIApiKey = openAIApiKey
+    }
+
+    async checkForViolations(input: string): Promise<string> {
+        if (!this.openAIApiKey) {
             throw Error('OpenAI API key not found')
         }
         // Create a new instance of the OpenAIModerationChain
         const moderation = new OpenAIModerationChain({
-            openAIApiKey: openAIApiKey,
+            openAIApiKey: this.openAIApiKey,
             throwError: false // If set to true, the call will throw an error when the moderation chain detects violating content. If set to false, violating content will return "Text was found that violates OpenAI's content policy.".
         })
         // Send the user's input to the moderation chain and wait for the result