From 69f3483ae52d309dac5ef82716b82306e6c29e15 Mon Sep 17 00:00:00 2001 From: Renato Rosa Date: Wed, 19 Aug 2026 22:30:19 -0300 Subject: [PATCH] Atualizar app/tasks.py --- app/tasks.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/app/tasks.py b/app/tasks.py index 7e0930b..f58c2f8 100644 --- a/app/tasks.py +++ b/app/tasks.py @@ -24,6 +24,11 @@ def generate_task(self, conversation_id, model, prompt): output = r.json().get("response", "") 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( id=str(uuid.uuid4()), 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) -def system_admin_generate(self, cid, model, prompt): +def system_admin_generate(self, email, cid, model, prompt): logger.info("system_admin_task_started") + conv = db.query(Conversation).filter_by(id=cid, user_email=email).first() return run_ollama(cid, model, prompt) @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") + conv = db.query(Conversation).filter_by(id=cid, user_email=email).first() return run_ollama(cid, model, prompt) @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") + conv = db.query(Conversation).filter_by(id=cid, user_email=email).first() return run_ollama(cid, model, prompt)