From 4f77706d0c2ce5a2e835b2cb7c8765c570537425 Mon Sep 17 00:00:00 2001
From: Mick Grove <mick.grove@mongodb.com>
Date: Wed, 27 Aug 2025 15:43:31 -0700
Subject: [PATCH] changes in response to code review

---
 data/rules/cerebras.yml         | 2 +-
 data/rules/fireworksai.yml      | 2 +-
 data/rules/nvidia.yml           | 1 -
 data/rules/weightsandbiases.yml | 6 ++----
 4 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/data/rules/cerebras.yml b/data/rules/cerebras.yml
index af0f49c..73760a6 100644
--- a/data/rules/cerebras.yml
+++ b/data/rules/cerebras.yml
@@ -17,7 +17,7 @@ rules:
           method: GET
           url: "https://api.cerebras.ai/v1/models"
           headers:
-            Authorization: "Bearer {{TOKEN}}"
+            Authorization: "Bearer {{ TOKEN }}"
           response_matcher:
             - report_response: true
             - type: StatusMatch
diff --git a/data/rules/fireworksai.yml b/data/rules/fireworksai.yml
index 9ca431f..0933441 100644
--- a/data/rules/fireworksai.yml
+++ b/data/rules/fireworksai.yml
@@ -17,7 +17,7 @@ rules:
           method: GET
           url: "https://api.fireworks.ai/inference/v1/models"
           headers:
-            Authorization: "Bearer {{TOKEN}}"
+            Authorization: "Bearer {{ TOKEN }}"
           response_matcher:
             - report_response: true
             - type: StatusMatch
diff --git a/data/rules/nvidia.yml b/data/rules/nvidia.yml
index 25a90e1..1dc7b31 100644
--- a/data/rules/nvidia.yml
+++ b/data/rules/nvidia.yml
@@ -26,6 +26,5 @@ rules:
             - report_response: true
             - type: StatusMatch
               status: [200]
-              expected: ["application/json"]
             - type: WordMatch
               words: ["id", "versionId"] 
diff --git a/data/rules/weightsandbiases.yml b/data/rules/weightsandbiases.yml
index 6661f53..bed9ca5 100644
--- a/data/rules/weightsandbiases.yml
+++ b/data/rules/weightsandbiases.yml
@@ -5,8 +5,6 @@ rules:
       (?xi)
       \b
       (?:wandb|weightsandbiases)
-      (?:.|[\n\r]){0,32}?
-      (?:SECRET|PRIVATE|ACCESS|KEY|TOKEN)
       (?:.|[\n\r]){0,16}?
       (
         [a-f0-9]{40}
@@ -28,10 +26,10 @@ rules:
             Authorization: "Basic {{ 'api:' | append: TOKEN | b64enc }}"
             Content-Type: "application/json"
           body: |
-            {"query":"query { viewer { id } }"}
+            {"query":"query { viewer { email username } }"}
           response_matcher:
             - report_response: true
             - type: JsonValid
             - type: WordMatch
               words:
-                - '"id"'
+                - '"username"'