From f0abb1f3e6b8d8c70e6f4880b588a46ebe77f16c Mon Sep 17 00:00:00 2001
From: Alexander Myasoedov <msoedov@gmail.com>
Date: Mon, 30 Dec 2024 14:55:49 +0200
Subject: [PATCH] feat(Add labels):

---
 agentic_security/static/base.js    | 4 ++--
 agentic_security/static/index.html | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/agentic_security/static/base.js b/agentic_security/static/base.js
index b255ff7..98f37ee 100644
--- a/agentic_security/static/base.js
+++ b/agentic_security/static/base.js
@@ -168,8 +168,8 @@ let LLM_CONFIGS = [
   { 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' },
+  { name: 'Custom API Image', prompts: 40000, customInstructions: 'Requires api spec', modality: 'Image' },
+  { name: 'Custom API Files', prompts: 40000, customInstructions: 'Requires api spec', modality: 'Files' },
   { name: 'Gemini', prompts: 40000 },
   { name: 'Claude', prompts: 40000 },
   { name: 'Cohere', prompts: 40000 },
diff --git a/agentic_security/static/index.html b/agentic_security/static/index.html
index 21d7686..133895a 100644
--- a/agentic_security/static/index.html
+++ b/agentic_security/static/index.html
@@ -51,7 +51,8 @@
               <div class="text-sm text-gray-400">
                 {{ config.customInstructions || 'Requires API key' }}
               </div>
-              <div class="mt-2 text-dark-accent-green font-semibold">API</div>
+              <div class="mt-2 text-dark-accent-green font-semibold">
+                {{config.modality || 'API'}}</div>
             </div>
           </div>
         </section>