فهرست منبع

提交任务:fast/relaxed

TanJingyu 6 ماه پیش
والد
کامیت
1b6dcbcbba
2فایلهای تغییر یافته به همراه4 افزوده شده و 3 حذف شده
  1. 3 2
      api/mj_api.py
  2. 1 1
      app.py

+ 3 - 2
api/mj_api.py

@@ -24,8 +24,9 @@ def query_job_queue(cookie, user_id):
     return response_catch(response)
 
 
-# 提交任务
-def submit_job(cookie, prompt, user_id, mode='fast'):
+# 提交任务 mode: relaxed/fast
+def submit_job(cookie, prompt, user_id, mode):
+    mode = mode or "fast"
     request_data = {
         "f": {
             "mode": mode,

+ 1 - 1
app.py

@@ -72,7 +72,7 @@ def submit_job():
         return jsonify({"error": "user_id is required!"}), 400
 
     response_json_str = mj_api.submit_job(cookie=request_data.get('cookie'), prompt=request_data.get('prompt'),
-                                          user_id=request_data.get('user_id'))
+                                          user_id=request_data.get('user_id'), mode=request_data.get('mode'))
     return response_json_str