fix: respect payment_required setting

edgeandnode · Oct 30, 2024 · f0cbb9a · f0cbb9a
1 parent 136ad69
commit f0cbb9a
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/auth.rs b/src/auth.rs
@@ -88,13 +88,15 @@ impl AuthContext {
             .get(token)
             .ok_or_else(|| anyhow::anyhow!("API key not found"))?;
 
-        match api_key.query_status {
-            QueryStatus::Active => (),
-            QueryStatus::ServiceShutoff => {
-                bail!("payment required for subsequent requests for this API key");
-            }
-            QueryStatus::MonthlyCapReached => {
-                bail!("spend limit exceeded for this API key");
+        if self.payment_required {
+            match api_key.query_status {
+                QueryStatus::Active => (),
+                QueryStatus::ServiceShutoff => {
+                    bail!("payment required for subsequent requests for this API key");
+                }
+                QueryStatus::MonthlyCapReached => {
+                    bail!("spend limit exceeded for this API key");
+                }
             }
         }