first commit

packages/@n8n/nodes-langchain/nodes/Guardrails/test/helpers/configureNodeInputs.test.ts

@@ -0,0 +1,75 @@
+import type { GuardrailsOptions } from '../../actions/types';
+import { configureNodeInputsV2, hasLLMGuardrails } from '../../helpers/configureNodeInputs';
+
+describe('configureNodeInputs', () => {
+	describe('hasLLMGuardrails+configureNodeInputs', () => {
+		it.each([
+			{
+				guardrails: { nsfw: { value: { threshold: 0.5 } } },
+				expected: true,
+				expectedInputs: 2,
+				name: 'nsfw',
+			},
+			{
+				guardrails: { topicalAlignment: { value: { threshold: 0.7, prompt: 'test' } } },
+				expected: true,
+				expectedInputs: 2,
+				name: 'topicalAlignment',
+			},
+			{
+				guardrails: {
+					custom: { guardrail: [{ name: 'custom', prompt: 'test prompt', threshold: 0.6 }] },
+				},
+				expected: true,
+				expectedInputs: 2,
+				name: 'custom',
+			},
+			{
+				guardrails: { jailbreak: { value: { threshold: 0.8 } } },
+				expected: true,
+				expectedInputs: 2,
+				name: 'jailbreak',
+			},
+			{
+				guardrails: {
+					nsfw: { value: { threshold: 0.5 } },
+					topicalAlignment: { value: { threshold: 0.7, prompt: 'test' } },
+					custom: { guardrail: [{ name: 'custom1', prompt: 'test prompt', threshold: 0.6 }] },
+					jailbreak: { value: { threshold: 0.8 } },
+				},
+				expectedInputs: 2,
+				name: 'multiple LLM checks',
+				expected: true,
+			},
+			{
+				guardrails: {
+					keywords: 'test, keywords',
+					pii: { value: { type: 'all' } },
+				},
+				expected: false,
+				expectedInputs: 1,
+				name: 'only non-LLM checks',
+			},
+			{
+				guardrails: {},
+				expected: false,
+				expectedInputs: 1,
+				name: 'empty guardrails',
+			},
+			{
+				guardrails: undefined,
+				expected: false,
+				expectedInputs: 1,
+				name: 'undefined guardrails',
+			},
+		])(
+			'should return $expected when guardrails contain $name',
+			({ guardrails, expected, expectedInputs }) => {
+				expect(hasLLMGuardrails(guardrails as GuardrailsOptions)).toBe(expected);
+				expect(configureNodeInputsV2({ guardrails: guardrails as GuardrailsOptions })).toHaveLength(
+					expectedInputs,
+				);
+			},
+		);
+	});
+});