Error messages that to be sent from embedding gateway (#1118)

stargate · May 28, 2024 · 5104a23 · 5104a23
1 parent 36a2b63
commit 5104a23
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/java/io/stargate/sgv2/jsonapi/exception/ErrorCode.java b/src/main/java/io/stargate/sgv2/jsonapi/exception/ErrorCode.java
@@ -183,7 +183,9 @@ public enum ErrorCode {
   SERVER_FAILURE("Driver failed"),
   SERVER_UNHANDLED_ERROR("Server failed"),
   INVALID_PARAMETER_VALIDATION_TYPE("Invalid Parameter Validation Type"),
-  SERVER_EMBEDDING_GATEWAY_NOT_AVAILABLE("Embedding Gateway is not available");
+  SERVER_EMBEDDING_GATEWAY_NOT_AVAILABLE("Embedding Gateway is not available"),
+  EMBEDDING_GATEWAY_ERROR_RATE_LIMIT("Embedding Gateway error rate limit reached for the tenant"),
+  EMBEDDING_GATEWAY_PROCESSING_ERROR("Embedding Gateway failed to process request");
 
   private final String message;
 

diff --git a/src/main/java/io/stargate/sgv2/jsonapi/service/resolver/model/impl/ValidateCredentials.java b/src/main/java/io/stargate/sgv2/jsonapi/service/resolver/model/impl/ValidateCredentials.java
@@ -46,6 +46,6 @@ public boolean validate(String provider, String value) {
   }
 
   void onStop(@Observes ShutdownEvent ev) {
-    channel.shutdown();
+    if (channel != null) channel.shutdown();
   }
 }