Skip to content

Commit

Permalink
ResponsibleAI - Input Moderation - Renaming of files for better clari…
Browse files Browse the repository at this point in the history
…ty and alignment to functionality
  • Loading branch information
vinodkiran committed Nov 23, 2023
1 parent c274085 commit b3f44e0
Show file tree
Hide file tree
Showing 9 changed files with 14 additions and 16 deletions.
2 changes: 1 addition & 1 deletion packages/components/nodes/chains/LLMChain/LLMChain.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ import { BaseOutputParser } from 'langchain/schema/output_parser'
import { formatResponse, injectOutputParser } from '../../outputparsers/OutputParserHelpers'
import { BaseLLMOutputParser } from 'langchain/schema/output_parser'
import { OutputFixingParser } from 'langchain/output_parsers'
import { checkInputs, Moderation, streamResponse } from '../../responsibleAI/ResponsibleAI'
import { checkInputs, Moderation, streamResponse } from '../../moderation/Moderation'

class LLMChain_Chains implements INode {
label: string
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
import { BaseLanguageModel } from 'langchain/base_language'
import { Server } from 'socket.io'

export abstract class ResponsibleAI {}

export abstract class Moderation extends ResponsibleAI {
export abstract class Moderation {
abstract checkForViolations(llm: BaseLanguageModel, input: string): Promise<string>
}

Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { getBaseClasses } from '../../../src'
import { Moderation } from '../ResponsibleAI'
import { Moderation } from '../Moderation'
import { OpenAIModerationRunner } from './OpenAIModerationRunner'

class OpenAIModeration implements INode {
Expand All @@ -20,7 +20,7 @@ class OpenAIModeration implements INode {
this.version = 1.0
this.type = 'Moderation'
this.icon = 'openai-moderation.png'
this.category = 'Responsible AI'
this.category = 'Moderation'
this.description = 'Check whether content complies with OpenAI usage policies.'
this.baseClasses = [this.type, ...getBaseClasses(Moderation)]
this.inputs = [
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { Moderation } from '../ResponsibleAI'
import { Moderation } from '../Moderation'
import { BaseLanguageModel } from 'langchain/base_language'
import { OpenAIModerationChain } from 'langchain/chains'

Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import { INode, INodeData, INodeParams } from '../../../src/Interface'
import { getBaseClasses } from '../../../src'
import { Moderation } from '../ResponsibleAI'
import { Moderation } from '../Moderation'
import { SimplePromptModerationRunner } from './SimplePromptModerationRunner'

class SimplePromptModeration implements INode {
Expand All @@ -20,7 +20,7 @@ class SimplePromptModeration implements INode {
this.version = 1.0
this.type = 'Moderation'
this.icon = 'simple_moderation.png'
this.category = 'Responsible AI'
this.category = 'Moderation'
this.description = 'Check whether input consists of any text from Deny list, and prevent being sent to LLM'
this.baseClasses = [this.type, ...getBaseClasses(Moderation)]
this.inputs = [
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import { Moderation } from '../ResponsibleAI'
import { Moderation } from '../Moderation'
import { BaseLanguageModel } from 'langchain/base_language'

export class SimplePromptModerationRunner implements Moderation {
Expand Down
12 changes: 6 additions & 6 deletions packages/server/marketplaces/chatflows/Input Moderation.json
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@
"version": 1,
"name": "inputModerationOpenAI",
"type": "Moderation",
"baseClasses": ["Moderation", "ResponsibleAI"],
"category": "Responsible AI",
"baseClasses": ["Moderation"],
"category": "Moderation",
"description": "Check whether content complies with OpenAI usage policies.",
"inputParams": [
{
Expand All @@ -37,10 +37,10 @@
},
"outputAnchors": [
{
"id": "inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|ResponsibleAI",
"id": "inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|Moderation",
"name": "inputModerationOpenAI",
"label": "Moderation",
"type": "Moderation | ResponsibleAI"
"type": "Moderation"
}
],
"outputs": {},
Expand Down Expand Up @@ -406,11 +406,11 @@
"edges": [
{
"source": "inputModerationOpenAI_0",
"sourceHandle": "inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|ResponsibleAI",
"sourceHandle": "inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|Moderation",
"target": "llmChain_0",
"targetHandle": "llmChain_0-input-inputModeration-Moderation",
"type": "buttonedge",
"id": "inputModerationOpenAI_0-inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|ResponsibleAI-llmChain_0-llmChain_0-input-inputModeration-Moderation",
"id": "inputModerationOpenAI_0-inputModerationOpenAI_0-output-inputModerationOpenAI-Moderation|Moderation-llmChain_0-llmChain_0-input-inputModeration-Moderation",
"data": {
"label": ""
}
Expand Down

0 comments on commit b3f44e0

Please sign in to comment.