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:30:19 -03:00
parent 3ed8e69f5c
commit 69f3483ae5
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/tasks.py
+11 -3
View File
@@ -24,6 +24,11 @@ def generate_task(self, conversation_id, model, prompt):
output = r.json().get("response", "") output = r.json().get("response", "")
db = SessionLocal() db = SessionLocal()
conv = db.query(Conversation).filter_by(id=cid, user_email=email).first()
if not conv:
logger.error("conversation_access_denied")
return None
msg = Message( msg = Message(
id=str(uuid.uuid4()), id=str(uuid.uuid4()),
conversation_id=conversation_id, conversation_id=conversation_id,
@@ -42,16 +47,19 @@ def generate_task(self, conversation_id, model, prompt):
@celery.task(name="tasks.system_admin_generate", bind=True) @celery.task(name="tasks.system_admin_generate", bind=True)
def system_admin_generate(self, cid, model, prompt): def system_admin_generate(self, email, cid, model, prompt):
logger.info("system_admin_task_started") logger.info("system_admin_task_started")
conv = db.query(Conversation).filter_by(id=cid, user_email=email).first()
return run_ollama(cid, model, prompt) return run_ollama(cid, model, prompt)
@celery.task(name="tasks.corp_admin_generate", bind=True) @celery.task(name="tasks.corp_admin_generate", bind=True)
def corp_admin_generate(self, cid, model, prompt): def corp_admin_generate(self, email, cid, model, prompt):
logger.info("corp_admin_task_started") logger.info("corp_admin_task_started")
conv = db.query(Conversation).filter_by(id=cid, user_email=email).first()
return run_ollama(cid, model, prompt) return run_ollama(cid, model, prompt)
@celery.task(name="tasks.user_generate", bind=True) @celery.task(name="tasks.user_generate", bind=True)
def user_generate(self, cid, model, prompt): def user_generate(self, email, cid, model, prompt):
logger.info("user_task_started") logger.info("user_task_started")
conv = db.query(Conversation).filter_by(id=cid, user_email=email).first()
return run_ollama(cid, model, prompt) return run_ollama(cid, model, prompt)