renatoxsr/jusfutura
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Atualizar app/tasks.py

Browse Source
This commit is contained in:
renatoxsr
2026-08-19 22:11:03 -03:00
parent 63beb7627a
commit ec61283b6b
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/tasks.py
+12 -10
View File
@@ -40,16 +40,18 @@ def generate_task(self, conversation_id, model, prompt):
raise self.retry(exc=e, countdown=2) raise self.retry(exc=e, countdown=2)
def enqueue_task(model, prompt, cid, user):
db = SessionLocal()
plan = db.query(Plan).filter_by(id=user.plan_id).first()
db.close()
queue = plan.priority_queue # default, premium, admin @celery.task(name="tasks.system_admin_generate", bind=True)
def system_admin_generate(self, cid, model, prompt):
logger.info("system_admin_task_started")
return run_ollama(cid, model, prompt)
if queue == "admin": @celery.task(name="tasks.corp_admin_generate", bind=True)
return admin_generate.delay(cid, model, prompt) def corp_admin_generate(self, cid, model, prompt):
if queue == "premium": logger.info("corp_admin_task_started")
return premium_generate.delay(cid, model, prompt) return run_ollama(cid, model, prompt)
return generate_task.delay(cid, model, prompt)
@celery.task(name="tasks.user_generate", bind=True)
def user_generate(self, cid, model, prompt):
logger.info("user_task_started")
return run_ollama(cid, model, prompt)