-
Notifications
You must be signed in to change notification settings - Fork 521
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
395 additions
and
419 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,7 +1,7 @@ | ||
| import { PluginContext, PluginParameters } from '../types'; | ||
| import { BedrockParameters, pluginHandler } from './index'; | ||
| import { bedrockPIIHandler } from './redactPii'; | ||
| import { pluginHandler } from './index'; | ||
| import creds from './.creds.json'; | ||
| import { BedrockParameters } from './type'; | ||
|
|
||
| /** | ||
| * @example Parameters object | ||
|
|
@@ -90,7 +90,7 @@ describe('Credentials check', () => { | |
| expect(result).toBeDefined(); | ||
| expect(result.verdict).toBe(false); | ||
| expect(result.error).toBe(null); | ||
| expect(result.data.customWords).toHaveLength(1); | ||
| expect(result.data).toBeDefined(); | ||
| }); | ||
|
|
||
| test('Should be working with content_filter', async () => { | ||
|
|
@@ -113,77 +113,77 @@ describe('Credentials check', () => { | |
| expect(result).toBeDefined(); | ||
| expect(result.verdict).toBe(false); | ||
| expect(result.error).toBe(null); | ||
| expect(result.data.filters).toHaveLength(1); | ||
| expect(result.data).toBeDefined(); | ||
| }); | ||
|
|
||
| test('Should work fine with redaction for sensitive info', async () => { | ||
| const context = { | ||
| response: { | ||
| json: { | ||
| choices: [ | ||
| { | ||
| message: { | ||
| content: | ||
| 'Hello, John doe. How are you doing?. I see your email is [email protected]', | ||
| }, | ||
| }, | ||
| ], | ||
| }, | ||
| }, | ||
| requestType: 'chatComplete', | ||
| }; | ||
|
|
||
| const parameters: PluginParameters<BedrockParameters['credentials']> = { | ||
| ...creds, | ||
| }; | ||
|
|
||
| const result = await bedrockPIIHandler( | ||
| context as unknown as PluginContext, | ||
| parameters, | ||
| 'afterRequestHook', | ||
| { env: {} } | ||
| ); | ||
|
|
||
| const outputMessage = | ||
| result.transformedData?.response.json.choices[0].message.content; | ||
| expect(result).toBeDefined(); | ||
| expect(result.verdict).toBe(true); | ||
| expect(outputMessage).toEqual( | ||
| 'Hello, {NAME}. How are you doing?. I see your email is {EMAIL}\n' | ||
| ); | ||
| }); | ||
|
|
||
| test('Should work fine with regex redaction for sensitive info', async () => { | ||
| const context = { | ||
| response: { | ||
| json: { | ||
| choices: [ | ||
| { | ||
| message: { | ||
| content: 'bedrock-12121, bedrock-12121', | ||
| }, | ||
| }, | ||
| ], | ||
| }, | ||
| }, | ||
| requestType: 'chatComplete', | ||
| }; | ||
|
|
||
| const parameters: PluginParameters<BedrockParameters['credentials']> = { | ||
| ...creds, | ||
| }; | ||
|
|
||
| const result = await bedrockPIIHandler( | ||
| context as unknown as PluginContext, | ||
| parameters, | ||
| 'afterRequestHook', | ||
| { env: {} } | ||
| ); | ||
|
|
||
| const outputMessage = | ||
| result.transformedData?.response.json.choices[0].message.content; | ||
| expect(result).toBeDefined(); | ||
| expect(result.verdict).toBe(true); | ||
| expect(outputMessage).toBe('{bedrock-id}, {bedrock-id}\n'); | ||
| }); | ||
| // test('Should work fine with redaction for sensitive info', async () => { | ||
| // const context = { | ||
| // response: { | ||
| // json: { | ||
| // choices: [ | ||
| // { | ||
| // message: { | ||
| // content: | ||
| // 'Hello, John doe. How are you doing?. I see your email is [email protected]', | ||
| // }, | ||
| // }, | ||
| // ], | ||
| // }, | ||
| // }, | ||
| // requestType: 'chatComplete', | ||
| // }; | ||
|
|
||
| // const parameters: PluginParameters<BedrockParameters['credentials']> = { | ||
| // ...creds, | ||
| // }; | ||
|
|
||
| // const result = await bedrockPIIHandler( | ||
| // context as unknown as PluginContext, | ||
| // parameters, | ||
| // 'afterRequestHook', | ||
| // { env: {} } | ||
| // ); | ||
|
|
||
| // const outputMessage = | ||
| // result.transformedData?.response.json.choices[0].message.content; | ||
| // expect(result).toBeDefined(); | ||
| // expect(result.verdict).toBe(true); | ||
| // expect(outputMessage).toEqual( | ||
| // 'Hello, {NAME}. How are you doing?. I see your email is {EMAIL}\n' | ||
| // ); | ||
| // }); | ||
|
|
||
| // test('Should work fine with regex redaction for sensitive info', async () => { | ||
| // const context = { | ||
| // response: { | ||
| // json: { | ||
| // choices: [ | ||
| // { | ||
| // message: { | ||
| // content: 'bedrock-12121, bedrock-12121', | ||
| // }, | ||
| // }, | ||
| // ], | ||
| // }, | ||
| // }, | ||
| // requestType: 'chatComplete', | ||
| // }; | ||
|
|
||
| // const parameters: PluginParameters<BedrockParameters['credentials']> = { | ||
| // ...creds, | ||
| // }; | ||
|
|
||
| // const result = await bedrockPIIHandler( | ||
| // context as unknown as PluginContext, | ||
| // parameters, | ||
| // 'afterRequestHook', | ||
| // { env: {} } | ||
| // ); | ||
|
|
||
| // const outputMessage = | ||
| // result.transformedData?.response.json.choices[0].message.content; | ||
| // expect(result).toBeDefined(); | ||
| // expect(result.verdict).toBe(true); | ||
| // expect(outputMessage).toBe('{bedrock-id}, {bedrock-id}\n'); | ||
| // }); | ||
| }); | ||
Oops, something went wrong.