From 9269aff7a6927e6d6b13c301094ab7ab328ffc5c Mon Sep 17 00:00:00 2001
From: ShadowVR <shadowvr@shadowhosting.xyz>
Date: Mon, 10 Jun 2024 01:51:01 -0400
Subject: [PATCH] Added new flags

---
 utils/functions.js | 50 ++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 46 insertions(+), 4 deletions(-)

diff --git a/utils/functions.js b/utils/functions.js
index 9c3634b..caac3eb 100644
--- a/utils/functions.js
+++ b/utils/functions.js
@@ -32,19 +32,37 @@ module.exports = {
         let Harassment = false;
         let SelfHarm = false;
         let Violence = false;
+        let Ignorance = false;
+        let Homophobia = false;
+        let Transphobic = false;
+        let Disrespect = false;
+        let Politics = false;
+        let Slurs = false;
 
         if (object['sexual'] || object['sexual/minors']) Sexual = true;
         if (object['hate'] || object['hate/threatening']) Hate = true;
         if (object['harassment'] || object['harassment/threatening']) Harassment = true;
         if (object['self-harm'] || object['self-harm/intent'] || object['self-harm/instructions']) SelfHarm = true;
         if (object['violence'] || object['violence/graphic']) Violence = true;
+        if (object['ignorance']) Ignorance = true;
+        if (object['homophobia']) Homophobia = true;
+        if (object['transphobic']) Transphobic = true;
+        if (object['disrespect']) Disrespect = true;
+        if (object['politics']) Politics = true;
+        if (object['slurs']) Slurs = true;
 
         const flags = {
             "Sexual": Sexual,
             "Hate": Hate,
             "Harassment": Harassment,
             "Self-Harm": SelfHarm,
-            "Violence": Violence
+            "Violence": Violence,
+            "Ignorance": Ignorance,
+            "Homophobia": Homophobia,
+            "Transphobic": Transphobic,
+            "Disrespect": Disrespect,
+            "Politics": Politics,
+            "Slurs": Slurs
         };
 
         const allFlags = Object.keys(flags).map(key => flags[key] ? `${key}: ✅` : `${key}: ❌`).join("\n");
@@ -86,8 +104,8 @@ module.exports = {
         const messagesTokens = messagesTokenCounts.reduce((a, b) => a + b, 0) + 2;
 
         let maxTokens;
-        if (model === 'gpt-3.5') maxTokens = 4097
-        else if (model === 'gpt-4') maxTokens = 8192
+        if (model === 'gpt-4') maxTokens = 4097;
+        else if (model === 'gpt-3.5') maxTokens = 8192;
 
         return {
             tokens: messagesTokens,
@@ -114,4 +132,28 @@ module.exports = {
 
     },
 
-};
\ No newline at end of file
+};
+
+// Test the flagCheck function
+const testCases = [
+    { "sexual": true },
+    { "hate": true },
+    { "harassment": true },
+    { "self-harm": true },
+    { "violence": true },
+    { "ignorance": true },
+    { "homophobia": true },
+    { "transphobic": true },
+    { "disrespect": true },
+    { "politics": true },
+    { "slurs": true },
+];
+
+// Run test cases
+testCases.forEach(testCase => {
+    const result = module.exports.flagCheck(testCase);
+    console.log(`Testing: ${JSON.stringify(testCase)}`);
+    console.log(result.allFlags);
+    console.log(result.trueFlags);
+    console.log('\n');
+});