Atualizar app/app.py
This commit is contained in:
+3
-2
@@ -113,7 +113,7 @@ def require_subscription(f):
|
|||||||
import hashlib
|
import hashlib
|
||||||
|
|
||||||
def make_cache_key(model, prompt):
|
def make_cache_key(model, prompt):
|
||||||
raw = f"{model}:{prompt}"
|
raw = f"{email}:{model}:{prompt}"
|
||||||
return hashlib.sha256(raw.encode()).hexdigest()
|
return hashlib.sha256(raw.encode()).hexdigest()
|
||||||
|
|
||||||
def get_cache(model, prompt):
|
def get_cache(model, prompt):
|
||||||
@@ -1176,7 +1176,8 @@ def generate():
|
|||||||
user = db.query(User).filter_by(email=email).first()
|
user = db.query(User).filter_by(email=email).first()
|
||||||
db.close()
|
db.close()
|
||||||
|
|
||||||
task = enqueue_by_priority(user, cid, model, prompt)
|
#task = enqueue_by_priority(user, cid, model, prompt)
|
||||||
|
task = enqueue_by_priority(user.email, cid, model, prompt)
|
||||||
|
|
||||||
return jsonify({"task_id": task.id})
|
return jsonify({"task_id": task.id})
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user