Handle sql.ErrNoRows in CheckModelAccess and respond with error m…

…essage. (#524)
swuecho · Sep 8, 2024 · 85a74e8 · 85a74e8
1 parent 1dde723
commit 85a74e8
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go
@@ -437,6 +437,16 @@ func (h *ChatHandler) CheckModelAccess(w http.ResponseWriter, chatSessionUuid st
 			Model:  rate.ChatModelName,
 		})
 
+	if err != nil {
+		// no rows
+		if errors.Is(err, sql.ErrNoRows) {
+			RespondWithError(w, http.StatusInternalServerError, "error.fail_to_get_rate_limit", err)
+			return true 
+		}
+		RespondWithError(w, http.StatusInternalServerError, "error.fail_to_get_rate_limit", err)
+		return true
+	}
+
 	log.Printf("%+v", usage10Min)
 
 	if int32(usage10Min) > rate.RateLimit {