remove

Author: ajgeiss0702

src/main/java/us/ajg0702/bots/ajsupport/CommandListener.java

@@ -39,16 +39,35 @@ public void onMessageContextInteraction(MessageContextInteractionEvent e) {
         final String name = e.getName();
 
         if(name.equals("Add to Vectorize")) {
-            StringSelectMenu.Builder selectMenuBuilder = StringSelectMenu.create("add-vectorize-message");
+            if(e.getMember().getId().equals("171160105155297282")) {
+                StringSelectMenu.Builder selectMenuBuilder = StringSelectMenu.create("add-vectorize-message");
 
-            for (Map.Entry<String, JsonElement> entry : bot.getJson().entrySet()) {
-                String key = entry.getKey();
-                String value = entry.getValue().getAsString();
+                for (Map.Entry<String, JsonElement> entry : bot.getJson().entrySet()) {
+                    String key = entry.getKey();
+                    String value = entry.getValue().getAsString();
 
-                selectMenuBuilder.addOption(key, key, SupportBot.cutString(value, 100));
+                    selectMenuBuilder.addOption(key, key, SupportBot.cutString(value, 100));
+                }
+
+                e.reply("Pick a message to attach to this vector | "+e.getTarget().getId()).addActionRow(selectMenuBuilder.build()).setEphemeral(true).queue();
+            } else {
+                e.reply("You can't do this!").setEphemeral(true).queue();
+                return;
             }
+        }
+        if(name.equals("Remove from Vectorize")) {
+            if(e.getMember().getId().equals("171160105155297282")) {
+                e.deferReply(true).queue();
+                InteractionHook hook = e.getHook().setEphemeral(true);
 
-            e.reply("Pick a message to attach to this vector | "+e.getTarget().getId()).addActionRow(selectMenuBuilder.build()).setEphemeral(true).queue();
+                try {
+                    EmbeddingUtils.deleteFromVectorize(e.getTarget().getId());
+                    hook.sendMessage("Removed from vectorize!").queue();
+                } catch (IOException ex) {
+                    hook.sendMessage("Failed to remove from vectorize: " + ex.getMessage()).queue();
+                    bot.getLogger().warn("Failed to remove from vectorize:", ex);
+                }
+            }
         }
     }
 

src/main/java/us/ajg0702/bots/ajsupport/autorespond/EmbeddingUtils.java

@@ -139,20 +139,49 @@ public static void insertIntoVectorize(String id, BigDecimal[] vector, String ch
             outputStream.flush();
         }
 
-        System.out.println("(Vectorize) Upsert request returned status " + con.getResponseCode());
+        System.out.println("(Vectorize: Upsert) Request returned status " + con.getResponseCode());
         try (var inputStream = con.getResponseCode() < HttpURLConnection.HTTP_BAD_REQUEST ? con.getInputStream() : con.getErrorStream();
              var reader = new InputStreamReader(inputStream)) {
             StringBuilder responseBuilder = new StringBuilder();
             int c;
             while ((c = reader.read()) != -1) {
                 responseBuilder.append((char) c);
             }
-            System.out.println("(Vectorize) Response body: " + responseBuilder);
+            System.out.println("(Vectorize: Upsert) Response body: " + responseBuilder);
         }
 
-        
+    }
 
-        
+    public static void deleteFromVectorize(String id) throws IOException {
+        final String token = System.getenv("CF_TOKEN");
+        if(token == null || token.isEmpty()) throw new IOException("Missing CF Token!");
+
+        URL url = new URL("https://api.cloudflare.com/client/v4/accounts/f55b85c8a963663b11036975203c63c0/vectorize/v2/indexes/support-autoresponse/delete_by_ids");
+        HttpURLConnection con = (HttpURLConnection) url.openConnection();
+        con.setRequestMethod("POST");
+        con.setRequestProperty("Content-Type", "application/json");
+        con.setRequestProperty("Authorization", "Bearer " + token); // Replace YOUR_BEARER_TOKEN with your actual token
+        con.setDoOutput(true);
+
+        String payload = String.format(
+                "{\"ids\": [\"%s\"]}",
+                id
+        );
+        try (var outputStream = con.getOutputStream()) {
+            outputStream.write(payload.getBytes());
+            outputStream.flush();
+        }
+
+        System.out.println("(Vectorize: Delete) Request returned status " + con.getResponseCode());
+        try (var inputStream = con.getResponseCode() < HttpURLConnection.HTTP_BAD_REQUEST ? con.getInputStream() : con.getErrorStream();
+             var reader = new InputStreamReader(inputStream)) {
+            StringBuilder responseBuilder = new StringBuilder();
+            int c;
+            while ((c = reader.read()) != -1) {
+                responseBuilder.append((char) c);
+            }
+            System.out.println("(Vectorize: Delete) Response body: " + responseBuilder);
+        }
     }
 
     public static class VectorizeResponse {