Skip to content

Commit 07e0ffb

Browse files
refactor : formatting
1 parent 7ee1975 commit 07e0ffb

File tree

4 files changed

+229
-208
lines changed

4 files changed

+229
-208
lines changed

plugins/index.ts

Lines changed: 95 additions & 95 deletions
Original file line numberDiff line numberDiff line change
@@ -1,107 +1,107 @@
1-
import { handler as defaultregexMatch } from "./default/regexMatch"
2-
import { handler as defaultsentenceCount } from "./default/sentenceCount"
3-
import { handler as defaultwordCount } from "./default/wordCount"
4-
import { handler as defaultcharacterCount } from "./default/characterCount"
5-
import { handler as defaultjsonSchema } from "./default/jsonSchema"
6-
import { handler as defaultjsonKeys } from "./default/jsonKeys"
7-
import { handler as defaultcontains } from "./default/contains"
8-
import { handler as defaultvalidUrls } from "./default/validUrls"
9-
import { handler as defaultwebhook } from "./default/webhook"
10-
import { handler as defaultlog } from "./default/log"
11-
import { handler as defaultcontainsCode } from "./default/containsCode"
12-
import { handler as defaultalluppercase } from "./default/alluppercase"
13-
import { handler as defaultendsWith } from "./default/endsWith"
14-
import { handler as defaultalllowercase } from "./default/alllowercase"
15-
import { handler as defaultmodelwhitelist } from "./default/modelwhitelist"
16-
import { handler as defaultjwt } from "./default/jwt"
17-
import { handler as defaultrequiredMetadataKeys } from "./default/requiredMetadataKeys"
18-
import { handler as portkeymoderateContent } from "./portkey/moderateContent"
19-
import { handler as portkeylanguage } from "./portkey/language"
20-
import { handler as portkeypii } from "./portkey/pii"
21-
import { handler as portkeygibberish } from "./portkey/gibberish"
22-
import { handler as aporiavalidateProject } from "./aporia/validateProject"
23-
import { handler as sydelabssydeguard } from "./sydelabs/sydeguard"
24-
import { handler as pillarscanPrompt } from "./pillar/scanPrompt"
25-
import { handler as pillarscanResponse } from "./pillar/scanResponse"
26-
import { handler as patronusphi } from "./patronus/phi"
27-
import { handler as patronuspii } from "./patronus/pii"
28-
import { handler as patronusisConcise } from "./patronus/isConcise"
29-
import { handler as patronusisHelpful } from "./patronus/isHelpful"
30-
import { handler as patronusisPolite } from "./patronus/isPolite"
31-
import { handler as patronusnoApologies } from "./patronus/noApologies"
32-
import { handler as patronusnoGenderBias } from "./patronus/noGenderBias"
33-
import { handler as patronusnoRacialBias } from "./patronus/noRacialBias"
34-
import { handler as patronusretrievalAnswerRelevance } from "./patronus/retrievalAnswerRelevance"
35-
import { handler as patronustoxicity } from "./patronus/toxicity"
36-
import { handler as patronuscustom } from "./patronus/custom"
37-
import { handler as pangeatextGuard } from "./pangea/textGuard"
38-
import { handler as pangeapii } from "./pangea/pii"
39-
import { handler as promptsecurityprotectPrompt } from "./promptsecurity/protectPrompt"
40-
import { handler as promptsecurityprotectResponse } from "./promptsecurity/protectResponse"
41-
import { handler as panwPrismaAirsintercept } from "./panw-prisma-airs/intercept"
42-
import { handler as walledaiguardrails } from "./walledai/guardrails"
1+
import { handler as defaultregexMatch } from './default/regexMatch';
2+
import { handler as defaultsentenceCount } from './default/sentenceCount';
3+
import { handler as defaultwordCount } from './default/wordCount';
4+
import { handler as defaultcharacterCount } from './default/characterCount';
5+
import { handler as defaultjsonSchema } from './default/jsonSchema';
6+
import { handler as defaultjsonKeys } from './default/jsonKeys';
7+
import { handler as defaultcontains } from './default/contains';
8+
import { handler as defaultvalidUrls } from './default/validUrls';
9+
import { handler as defaultwebhook } from './default/webhook';
10+
import { handler as defaultlog } from './default/log';
11+
import { handler as defaultcontainsCode } from './default/containsCode';
12+
import { handler as defaultalluppercase } from './default/alluppercase';
13+
import { handler as defaultendsWith } from './default/endsWith';
14+
import { handler as defaultalllowercase } from './default/alllowercase';
15+
import { handler as defaultmodelwhitelist } from './default/modelwhitelist';
16+
import { handler as defaultjwt } from './default/jwt';
17+
import { handler as defaultrequiredMetadataKeys } from './default/requiredMetadataKeys';
18+
import { handler as portkeymoderateContent } from './portkey/moderateContent';
19+
import { handler as portkeylanguage } from './portkey/language';
20+
import { handler as portkeypii } from './portkey/pii';
21+
import { handler as portkeygibberish } from './portkey/gibberish';
22+
import { handler as aporiavalidateProject } from './aporia/validateProject';
23+
import { handler as sydelabssydeguard } from './sydelabs/sydeguard';
24+
import { handler as pillarscanPrompt } from './pillar/scanPrompt';
25+
import { handler as pillarscanResponse } from './pillar/scanResponse';
26+
import { handler as patronusphi } from './patronus/phi';
27+
import { handler as patronuspii } from './patronus/pii';
28+
import { handler as patronusisConcise } from './patronus/isConcise';
29+
import { handler as patronusisHelpful } from './patronus/isHelpful';
30+
import { handler as patronusisPolite } from './patronus/isPolite';
31+
import { handler as patronusnoApologies } from './patronus/noApologies';
32+
import { handler as patronusnoGenderBias } from './patronus/noGenderBias';
33+
import { handler as patronusnoRacialBias } from './patronus/noRacialBias';
34+
import { handler as patronusretrievalAnswerRelevance } from './patronus/retrievalAnswerRelevance';
35+
import { handler as patronustoxicity } from './patronus/toxicity';
36+
import { handler as patronuscustom } from './patronus/custom';
37+
import { handler as pangeatextGuard } from './pangea/textGuard';
38+
import { handler as pangeapii } from './pangea/pii';
39+
import { handler as promptsecurityprotectPrompt } from './promptsecurity/protectPrompt';
40+
import { handler as promptsecurityprotectResponse } from './promptsecurity/protectResponse';
41+
import { handler as panwPrismaAirsintercept } from './panw-prisma-airs/intercept';
42+
import { handler as walledaiguardrails } from './walledai/guardrails';
4343

4444
export const plugins = {
45-
"default": {
46-
"regexMatch": defaultregexMatch,
47-
"sentenceCount": defaultsentenceCount,
48-
"wordCount": defaultwordCount,
49-
"characterCount": defaultcharacterCount,
50-
"jsonSchema": defaultjsonSchema,
51-
"jsonKeys": defaultjsonKeys,
52-
"contains": defaultcontains,
53-
"validUrls": defaultvalidUrls,
54-
"webhook": defaultwebhook,
55-
"log": defaultlog,
56-
"containsCode": defaultcontainsCode,
57-
"alluppercase": defaultalluppercase,
58-
"endsWith": defaultendsWith,
59-
"alllowercase": defaultalllowercase,
60-
"modelwhitelist": defaultmodelwhitelist,
61-
"jwt": defaultjwt,
62-
"requiredMetadataKeys": defaultrequiredMetadataKeys
45+
default: {
46+
regexMatch: defaultregexMatch,
47+
sentenceCount: defaultsentenceCount,
48+
wordCount: defaultwordCount,
49+
characterCount: defaultcharacterCount,
50+
jsonSchema: defaultjsonSchema,
51+
jsonKeys: defaultjsonKeys,
52+
contains: defaultcontains,
53+
validUrls: defaultvalidUrls,
54+
webhook: defaultwebhook,
55+
log: defaultlog,
56+
containsCode: defaultcontainsCode,
57+
alluppercase: defaultalluppercase,
58+
endsWith: defaultendsWith,
59+
alllowercase: defaultalllowercase,
60+
modelwhitelist: defaultmodelwhitelist,
61+
jwt: defaultjwt,
62+
requiredMetadataKeys: defaultrequiredMetadataKeys,
6363
},
64-
"portkey": {
65-
"moderateContent": portkeymoderateContent,
66-
"language": portkeylanguage,
67-
"pii": portkeypii,
68-
"gibberish": portkeygibberish
64+
portkey: {
65+
moderateContent: portkeymoderateContent,
66+
language: portkeylanguage,
67+
pii: portkeypii,
68+
gibberish: portkeygibberish,
6969
},
70-
"aporia": {
71-
"validateProject": aporiavalidateProject
70+
aporia: {
71+
validateProject: aporiavalidateProject,
7272
},
73-
"sydelabs": {
74-
"sydeguard": sydelabssydeguard
73+
sydelabs: {
74+
sydeguard: sydelabssydeguard,
7575
},
76-
"pillar": {
77-
"scanPrompt": pillarscanPrompt,
78-
"scanResponse": pillarscanResponse
76+
pillar: {
77+
scanPrompt: pillarscanPrompt,
78+
scanResponse: pillarscanResponse,
7979
},
80-
"patronus": {
81-
"phi": patronusphi,
82-
"pii": patronuspii,
83-
"isConcise": patronusisConcise,
84-
"isHelpful": patronusisHelpful,
85-
"isPolite": patronusisPolite,
86-
"noApologies": patronusnoApologies,
87-
"noGenderBias": patronusnoGenderBias,
88-
"noRacialBias": patronusnoRacialBias,
89-
"retrievalAnswerRelevance": patronusretrievalAnswerRelevance,
90-
"toxicity": patronustoxicity,
91-
"custom": patronuscustom
80+
patronus: {
81+
phi: patronusphi,
82+
pii: patronuspii,
83+
isConcise: patronusisConcise,
84+
isHelpful: patronusisHelpful,
85+
isPolite: patronusisPolite,
86+
noApologies: patronusnoApologies,
87+
noGenderBias: patronusnoGenderBias,
88+
noRacialBias: patronusnoRacialBias,
89+
retrievalAnswerRelevance: patronusretrievalAnswerRelevance,
90+
toxicity: patronustoxicity,
91+
custom: patronuscustom,
9292
},
93-
"pangea": {
94-
"textGuard": pangeatextGuard,
95-
"pii": pangeapii
93+
pangea: {
94+
textGuard: pangeatextGuard,
95+
pii: pangeapii,
9696
},
97-
"promptsecurity": {
98-
"protectPrompt": promptsecurityprotectPrompt,
99-
"protectResponse": promptsecurityprotectResponse
97+
promptsecurity: {
98+
protectPrompt: promptsecurityprotectPrompt,
99+
protectResponse: promptsecurityprotectResponse,
100100
},
101-
"panw-prisma-airs": {
102-
"intercept": panwPrismaAirsintercept
101+
'panw-prisma-airs': {
102+
intercept: panwPrismaAirsintercept,
103+
},
104+
walledai: {
105+
guardrails: walledaiguardrails,
103106
},
104-
"walledai": {
105-
"guardrails": walledaiguardrails
106-
}
107107
};

plugins/walledai/guardrails.ts

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,7 @@ import {
66
} from '../types';
77
import { post, getText } from '../utils';
88

9-
const API_URL =
10-
'https://services.walled.ai/v1/guardrail/moderate';
9+
const API_URL = 'https://services.walled.ai/v1/guardrail/moderate';
1110

1211
export const handler: PluginHandler = async (
1312
context: PluginContext,
@@ -47,20 +46,23 @@ export const handler: PluginHandler = async (
4746
const requestOptions = {
4847
headers: {
4948
'Content-Type': 'application/json',
50-
'Authorization': `Bearer ${parameters.credentials.apiKey}`, // Uncomment if API key is required
49+
Authorization: `Bearer ${parameters.credentials.apiKey}`, // Uncomment if API key is required
5150
},
5251
};
5352

5453
try {
55-
const response = await post(API_URL, requestBody, requestOptions, parameters.timeout);
54+
const response = await post(
55+
API_URL,
56+
requestBody,
57+
requestOptions,
58+
parameters.timeout
59+
);
5660
data = response.data;
57-
if (
58-
data.safety[0]?.isSafe==false
59-
) {
61+
if (data.safety[0]?.isSafe == false) {
6062
verdict = false;
6163
}
6264
} catch (e) {
63-
console.log(e)
65+
console.log(e);
6466
error = e instanceof Error ? e.message : String(e);
6567
verdict = true;
6668
data = null;

0 commit comments

Comments
 (0)