From 45aa3086e2e7c58e35b24a473654a8dd6c60559f Mon Sep 17 00:00:00 2001
From: Renato Rosa <renatoxsr@gmail.com>
Date: Wed, 19 Aug 2026 21:33:26 -0300
Subject: [PATCH] Adicionar app/tasks.py

---
 app/tasks.py | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 app/tasks.py

diff --git a/app/tasks.py b/app/tasks.py
new file mode 100644
index 0000000..32528ad
--- /dev/null
+++ b/app/tasks.py
@@ -0,0 +1,40 @@
+# All comments are in English.
+
+from celery_app import celery
+import requests
+import json
+import uuid
+from models import Message, Conversation
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy import create_engine
+from config import Config
+
+engine = create_engine(Config.DATABASE_URL)
+SessionLocal = sessionmaker(bind=engine)
+
+@celery.task(bind=True, max_retries=3)
+def generate_task(self, conversation_id, model, prompt):
+    """Runs an Ollama generation asynchronously."""
+    try:
+        url = f"{Config.OLLAMA_BASE_URL}/api/generate"
+        payload = {"model": model, "prompt": prompt}
+
+        r = requests.post(url, json=payload)
+        r.raise_for_status()
+        output = r.json().get("response", "")
+
+        db = SessionLocal()
+        msg = Message(
+            id=str(uuid.uuid4()),
+            conversation_id=conversation_id,
+            role="assistant",
+            content=output,
+        )
+        db.add(msg)
+        db.commit()
+        db.close()
+
+        return output
+
+    except Exception as e:
+        raise self.retry(exc=e, countdown=2)