From d674b44b94e8b123e85e3c3920c2e36005d36168 Mon Sep 17 00:00:00 2001
From: Risto McGehee <risto.mcgehee@gmail.com>
Date: Thu, 18 Jan 2024 20:39:00 -0800
Subject: [PATCH] Fix JS API client

---
 .github/workflows/javascript-tests.yaml |  1 +
 javascript-sdk/src/api.ts               | 18 ++----------------
 2 files changed, 3 insertions(+), 16 deletions(-)

diff --git a/.github/workflows/javascript-tests.yaml b/.github/workflows/javascript-tests.yaml
index fdb5e24..96d73c5 100644
--- a/.github/workflows/javascript-tests.yaml
+++ b/.github/workflows/javascript-tests.yaml
@@ -42,6 +42,7 @@ jobs:
           CHROMA_COLLECTION_NAME: javascript_tests
         run: |
           npm install
+          npm run build
           npm test
 
       - name: Dump docker logs
diff --git a/javascript-sdk/src/api.ts b/javascript-sdk/src/api.ts
index 0f6e719..e0a63af 100644
--- a/javascript-sdk/src/api.ts
+++ b/javascript-sdk/src/api.ts
@@ -44,12 +44,7 @@ export default class RebuffApi implements Rebuff {
 
   async detectInjection({
     userInput = "",
-    maxHeuristicScore = 0.75,
-    maxVectorScore = 0.9,
-    maxModelScore = 0.9,
-    runHeuristicCheck = true,
-    runVectorCheck = true,
-    runLanguageModelCheck = true,
+    tacticOverrides = [],
   }: DetectRequest): Promise<DetectResponse> {
     if (userInput === null) {
       throw new RebuffError("userInput is required");
@@ -57,12 +52,7 @@ export default class RebuffApi implements Rebuff {
     const requestData: DetectRequest = {
       userInput: "",
       userInputBase64: encodeString(userInput),
-      runHeuristicCheck: runHeuristicCheck,
-      runVectorCheck: runVectorCheck,
-      runLanguageModelCheck: runLanguageModelCheck,
-      maxVectorScore,
-      maxModelScore,
-      maxHeuristicScore,
+      tacticOverrides,
     };
 
     const response = await fetch(`${this.apiUrl}/api/detect`, {
@@ -76,10 +66,6 @@ export default class RebuffApi implements Rebuff {
     if (!response.ok) {
       throw new RebuffError((responseData as any)?.message);
     }
-    responseData.injectionDetected =
-      responseData.heuristicScore > maxHeuristicScore ||
-      responseData.modelScore > maxModelScore ||
-      responseData.vectorScore.topScore > maxVectorScore;
     return responseData;
   }