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