CalvinBackup/agentic_security
1
0
Fork 0
mirror of https://github.com/msoedov/agentic_security.git synced 2026-06-24 06:09:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(add has_files / has_image functionality):

Browse Source
This commit is contained in:
Alexander Myasoedov
2024-12-19 21:51:20 +02:00
parent 59c45d2281
commit 0100306da7
2 changed files with 32 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agentic_security/static/base.js
+31 -8
View File
@@ -1,13 +1,13 @@
let URL = window.location.href;
if (URL.endsWith('/')) {
URL = URL.slice(0, -1);
URL = URL.slice(0, -1);
}
URL = URL.replace('/#', '');
// Vue application
let LLM_SPECS = [
`POST ${URL}/v1/self-probe
`POST ${URL}/v1/self-probe
Authorization: Bearer XXXXX
Content-Type: application/json
@@ -16,7 +16,7 @@ Content-Type: application/json
}
`,
`POST https://api.openai.com/v1/chat/completions
`POST https://api.openai.com/v1/chat/completions
Authorization: Bearer sk-xxxxxxxxx
Content-Type: application/json
@@ -26,7 +26,7 @@ Content-Type: application/json
"temperature": 0.7
}
`,
`POST https://api.replicate.com/v1/models/mistralai/mixtral-8x7b-instruct-v0.1/predictions
`POST https://api.replicate.com/v1/models/mistralai/mixtral-8x7b-instruct-v0.1/predictions
Authorization: Bearer $APIKEY
Content-Type: application/json
@@ -43,7 +43,7 @@ Content-Type: application/json
}
}
`,
`POST https://api.groq.com/v1/request_manager/text_completion
`POST https://api.groq.com/v1/request_manager/text_completion
Authorization: Bearer $APIKEY
Content-Type: application/json
@@ -53,7 +53,7 @@ Content-Type: application/json
"user_prompt": "<<PROMPT>>"
}
`,
`POST https://api.together.xyz/v1/chat/completions
`POST https://api.together.xyz/v1/chat/completions
Authorization: Bearer $TOGETHER_API_KEY
Content-Type: application/json
@@ -65,7 +65,7 @@ Content-Type: application/json
]
}
`,
`POST ${URL}/v1/self-probe-image
`POST ${URL}/v1/self-probe-image
Authorization: Bearer XXXXX
Content-Type: application/json
@@ -87,7 +87,7 @@ Content-Type: application/json
}
]
`,
`POST ${URL}/v1/self-probe-file
`POST ${URL}/v1/self-probe-file
Authorization: Bearer $GROQ_API_KEY
Content-Type: multipart/form-data
@@ -97,3 +97,26 @@ Content-Type: multipart/form-data
}
`,
]
let LLM_CONFIGS = [
{ name: 'Custom API', prompts: 40000, customInstructions: 'Requires api spec' },
{ name: 'Open AI', prompts: 24000 },
{ name: 'Replicate', prompts: 40000 },
{ name: 'Groq', prompts: 40000 },
{ name: 'Together.ai', prompts: 40000 },
{ name: 'Custom API Image', prompts: 40000, customInstructions: 'Requires api spec' },
{ name: 'Custom API Files', prompts: 40000, customInstructions: 'Requires api spec' },
]
// todo: add filters based on the type of API spec for pre built datasets
function has_image(spec) {
return spec.includes('<<BASE64_IMAGE>>');
}
function has_files(spec) {
return spec.includes('multipart/form-data');
}
agentic_security/static/main.js
+1 -10
View File
@@ -29,16 +29,7 @@ var app = new Vue({
showLLMSpec: true, // Default to showing the LLM Spec Input
logs: [], // This will store all the logs
maxDisplayedLogs: 50, // Maximum number of logs to display
configs: [
{ name: 'Custom API', prompts: 40000, customInstructions: 'Requires api spec' },
{ name: 'Open AI', prompts: 24000 },
{ name: 'Replicate', prompts: 40000 },
{ name: 'Groq', prompts: 40000 },
{ name: 'Together.ai', prompts: 40000 },
{ name: 'Custom API Image', prompts: 40000, customInstructions: 'Requires api spec' },
{ name: 'Custom API Files', prompts: 40000, customInstructions: 'Requires api spec' },
],
configs: LLM_CONFIGS,
dataConfig: [],
},
created() {