8th

.env

@@ -17,8 +17,9 @@ QDRANT_API_KEY="eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJhY2Nlc3MiOiJtIn0.NRbT0QP
 
 
 # ---------- REDIS ----------
-REDIS_HOST="rediss://default:gQAAAAAAAS-BAAIncDFiM2E3OGQ1MmU5Zjk0OGM5ODU2ZmMzYzc4NjZjYzdjMHAxNzc2OTc@steady-clam-77697.upstash.io"
+REDIS_HOST="https://steady-clam-77697.upstash.io"
 REDIS_PORT=6379
+REDIS_PASSWORD="gQAAAAAAAS-BAAIncDFiM2E3OGQ1MmU5Zjk0OGM5ODU2ZmMzYzc4NjZjYzdjMHAxNzc2OTc"
 
 # ---------- WEBHOOKS ----------
 CALLBACK_URL="https://webhooksite.net/c93aac48-5237-4078-9511-14d778acba2f"

celery_app.py

@@ -3,11 +3,26 @@ from celery import Celery
 import redis
 from config import get_settings
 
+settings = get_settings()
+
+REDIS_HOST = settings.REDIS_HOST
+REDIS_PORT = settings.REDIS_PORT
+REDIS_PASSWORD = settings.REDIS_PASSWORD
+
+# Upstash (and most managed Redis) uses rediss:// (TLS)
+# Local/Docker Redis uses redis:// (plain)
+if REDIS_PASSWORD:
+    BROKER_URL   = f"rediss://:{REDIS_PASSWORD}@{REDIS_HOST}:{REDIS_PORT}/0?ssl_cert_reqs=CERT_NONE"
+    BACKEND_URL  = f"rediss://:{REDIS_PASSWORD}@{REDIS_HOST}:{REDIS_PORT}/1?ssl_cert_reqs=CERT_NONE"
+else:
+    BROKER_URL   = f"redis://{REDIS_HOST}:{REDIS_PORT}/0"
+    BACKEND_URL  = f"redis://{REDIS_HOST}:{REDIS_PORT}/1"
+
 celery_app = Celery(
     "assistant_worker",
-    broker=f"{get_settings().REDIS_HOST}:{get_settings().REDIS_PORT}/0",
-    backend=f"{get_settings().REDIS_HOST}:{get_settings().REDIS_PORT}/1",
-    include=['generation.ExamAnswer']
+    broker=BROKER_URL,
+    backend=BACKEND_URL,
+    include=["generation.ExamAnswer"]
 )
 
 celery_app.conf.update(
@@ -15,16 +30,31 @@ celery_app.conf.update(
     accept_content=["json"],
     result_serializer="json",
     task_track_started=True,
-    task_time_limit=60*60,
+    task_time_limit=60 * 60,
+    broker_use_ssl={"ssl_cert_reqs": "CERT_NONE"} if REDIS_PASSWORD else None,
+    redis_backend_use_ssl={"ssl_cert_reqs": "CERT_NONE"} if REDIS_PASSWORD else None,
 )
 
-import worker.tasks  
-from generation.ExamAnswer import grade_exam_task 
+import worker.tasks
+from generation.ExamAnswer import grade_exam_task
+
+
 def clear_redis_backend():
-    r = redis.Redis(host=get_settings().REDIS_HOST, port=get_settings().REDIS_PORT, db=1)
-    r.flushdb()  
+    if REDIS_PASSWORD:
+        r = redis.Redis(
+            host=REDIS_HOST,
+            port=REDIS_PORT,
+            password=REDIS_PASSWORD,
+            ssl=True,
+            ssl_cert_reqs=None,
+            db=1,
+        )
+    else:
+        r = redis.Redis(host=REDIS_HOST, port=REDIS_PORT, db=1)
+    r.flushdb()
     print("Redis result backend cleared!")
 
+
 @celery_app.on_after_configure.connect
 def setup(sender, **kwargs):
     clear_redis_backend()

config.py

@@ -18,6 +18,7 @@ class Settings(BaseSettings):
     GRADE_WEBHOOK_URL: Optional[str] = None
     REDIS_HOST: str = "localhost"
     REDIS_PORT: int = 6379
+    REDIS_PASSWORD: Optional[str] = None
     CALLBACK_URL: Optional[str] = None
 
     # ---------- BACKENDS ----------