Using easy QA prompt by default

ScottLogic · Aug 22, 2023 · 50837bf · 50837bf
1 parent e1b1e1b
commit 50837bf
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 18 deletions.
diff --git a/backend/src/app.ts b/backend/src/app.ts
@@ -10,6 +10,7 @@ import { ChatCompletionRequestMessage } from "openai";
 import { EmailInfo } from "./models/email";
 import { DefenceInfo } from "./models/defence";
 import { CHAT_MODELS } from "./models/chat";
+import { retrievalQAPrePrompt } from "./promptTemplates";
 
 dotenv.config();
 
@@ -91,7 +92,7 @@ app.listen(port, () => {
 
   // for dev purposes only - set the API key from the environment variable
   const envOpenAiKey = process.env.OPENAI_API_KEY;
-  const prePrompt = getQALLMprePrompt(getInitialDefences());
+  const prePrompt = retrievalQAPrePrompt;
   if (envOpenAiKey && prePrompt) {
     console.debug("Initializing models with API key from environment variable");
     setOpenAiApiKey(envOpenAiKey, defaultModel, prePrompt);

diff --git a/prompt-injection.code-workspace b/prompt-injection.code-workspace
@@ -1,18 +1,21 @@
 {
-  "folders": [
-    {
-      "path": "."
-    },
-    {
-      "name": "frontend",
-      "path": "frontend"
-    },
-    {
-      "name": "backend",
-      "path": "backend"
-    }
-  ],
-  "settings": {
-    "jest.disabledWorkspaceFolders": ["prompt-injection"]
-  }
-}
+	"folders": [
+		{
+			"path": "."
+		},
+		{
+			"name": "backend",
+			"path": "backend"
+		},
+		{
+			"name": "frontend",
+			"path": "frontend"
+		}
+	],
+	"settings": {
+		"jest.disabledWorkspaceFolders": [
+			"prompt-injection",
+			"frontend"
+		]
+	}
+}