浏览代码

fix alpha 限速

chenbiao 2 年之前
父节点
当前提交
43efd3c79a
共有 1 个文件被更改,包括 4 次插入2 次删除
  1. 4 2
      midjourney/src/main/java/com/yhlxj/web/mirror/AlphaMidjourneyController.java

+ 4 - 2
midjourney/src/main/java/com/yhlxj/web/mirror/AlphaMidjourneyController.java

@@ -46,7 +46,7 @@ public class AlphaMidjourneyController {
     private static final String AUTHORIZATION_HEADER = "Authorization";
     private static final String MODE_HEADER = "MJ-Mode";
     private static final String MODE_FAST = "fast";
-    private static final String MODE_RELAX = "relax";
+    private static final String MODE_RELAX = "relaxed";
     private static final String MODE_TURBO = "turbo";
 
     @RequestMapping("/auditLimit")
@@ -72,7 +72,9 @@ public class AlphaMidjourneyController {
             return badRequestResponse("模式错误.请选择fast 或者 relax模式");
         }
         try {
-            midjourneyService.userSubmitLimit(user,"imagine", "1234",false);
+            if (MODE_RELAX.equals(mode)) {
+                midjourneyService.userSubmitLimit(user,"imagine", "1234",false);
+            }
         }catch (Exception e){
             return badRequestResponse(e.getMessage());
         }