add additional llm_gateway error types (#1177)

BloopAI · Dec 15, 2023 · 7f44f51 · 7f44f51
1 parent 6d4ea7d
commit 7f44f51
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/server/bleep/src/llm_gateway.rs b/server/bleep/src/llm_gateway.rs
@@ -112,6 +112,21 @@ pub mod api {
 
         #[error("incorrect configuration")]
         BadConfiguration,
+
+        #[error("failed to call quota API")]
+        QuotaCallFailed,
+
+        #[error("exceeded quota")]
+        ExceededQuota,
+
+        #[error("quota API was not set")]
+        QuotaApiNotSet,
+
+        #[error("called quota API with no authorization")]
+        UnauthorizedQuotaCall,
+
+        #[error("waiting for the next token took longer than allowed")]
+        TokenDelayTooLarge,
     }
 
     pub type Result = std::result::Result<String, Error>;