From 8abff3da5112d0fa949665f872a9342c2615394d Mon Sep 17 00:00:00 2001
From: George Sproston <gsproston@scottlogic.com>
Date: Tue, 15 Aug 2023 11:17:24 +0100
Subject: [PATCH] Only doing LLM evaluation on phase 2 and sandbox

---
 backend/src/openai.js | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/backend/src/openai.js b/backend/src/openai.js
index 25f25d9bb..698b3272b 100644
--- a/backend/src/openai.js
+++ b/backend/src/openai.js
@@ -234,18 +234,22 @@ async function chatGptSendMessage(message, session, currentPhase) {
   let wonPhase = false;
 
   // evaluate the message for prompt injection
-  const evalPrompt = await queryPromptEvaluationModel(message);
-  if (evalPrompt.isMalicious) {
-    defenceInfo.triggeredDefences.push("LLM_EVALUATION");
-    if (isDefenceActive("LLM_EVALUATION", session.defences)) {
-      console.debug("LLM evalutation defence active.");
-      defenceInfo.blocked = true;
-      const evalResponse =
-        "Message blocked by the malicious prompt evaluator." +
-        evalPrompt.reason;
-      return { reply: evalResponse, defenceInfo: defenceInfo };
+  // to speed up replies, only do this on phases where defences are active
+  if (currentPhase >= 2) {
+    const evalPrompt = await queryPromptEvaluationModel(message);
+    if (evalPrompt.isMalicious) {
+      defenceInfo.triggeredDefences.push("LLM_EVALUATION");
+      if (isDefenceActive("LLM_EVALUATION", session.defences)) {
+        console.debug("LLM evalutation defence active.");
+        defenceInfo.blocked = true;
+        const evalResponse =
+          "Message blocked by the malicious prompt evaluator." +
+          evalPrompt.reason;
+        return { reply: evalResponse, defenceInfo: defenceInfo };
+      }
     }
   }
+
   // add user message to chat
   session.chatHistory.push({ role: "user", content: message });