From 8263b8ae7af1825395a07901f5b76e05d30610ba Mon Sep 17 00:00:00 2001
From: pablodanswer <pablo@danswer.ai>
Date: Sun, 10 Nov 2024 11:19:09 -0800
Subject: [PATCH] ensure all pushed

---
 backend/danswer/db/engine.py                             | 7 +++++++
 deployment/cloud_kubernetes/workers/beat.yaml            | 2 +-
 deployment/cloud_kubernetes/workers/heavy_worker.yaml    | 2 +-
 deployment/cloud_kubernetes/workers/indexing_worker.yaml | 2 +-
 deployment/cloud_kubernetes/workers/light_worker.yaml    | 2 +-
 deployment/cloud_kubernetes/workers/primary.yaml         | 2 +-
 6 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py
index 639f6addc75..fbe8020dddc 100644
--- a/backend/danswer/db/engine.py
+++ b/backend/danswer/db/engine.py
@@ -189,6 +189,13 @@ def get_app_name(cls) -> str:
             return ""
         return cls._app_name
 
+    @classmethod
+    def reset_engine(cls) -> None:
+        with cls._lock:
+            if cls._engine:
+                cls._engine.dispose()
+                cls._engine = None
+
 
 def get_all_tenant_ids() -> list[str] | list[None]:
     if not MULTI_TENANT:
diff --git a/deployment/cloud_kubernetes/workers/beat.yaml b/deployment/cloud_kubernetes/workers/beat.yaml
index 196c2612214..7e131d45208 100644
--- a/deployment/cloud_kubernetes/workers/beat.yaml
+++ b/deployment/cloud_kubernetes/workers/beat.yaml
@@ -14,7 +14,7 @@ spec:
     spec:
       containers:
         - name: celery-beat
-          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.3
+          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.5
           imagePullPolicy: Always
           command:
             [
diff --git a/deployment/cloud_kubernetes/workers/heavy_worker.yaml b/deployment/cloud_kubernetes/workers/heavy_worker.yaml
index 5a22ba260fa..f13214729a7 100644
--- a/deployment/cloud_kubernetes/workers/heavy_worker.yaml
+++ b/deployment/cloud_kubernetes/workers/heavy_worker.yaml
@@ -14,7 +14,7 @@ spec:
     spec:
       containers:
         - name: celery-worker-heavy
-          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.3
+          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.5
           imagePullPolicy: Always
           command:
             [
diff --git a/deployment/cloud_kubernetes/workers/indexing_worker.yaml b/deployment/cloud_kubernetes/workers/indexing_worker.yaml
index 65406cc0744..33df49d77b0 100644
--- a/deployment/cloud_kubernetes/workers/indexing_worker.yaml
+++ b/deployment/cloud_kubernetes/workers/indexing_worker.yaml
@@ -14,7 +14,7 @@ spec:
     spec:
       containers:
         - name: celery-worker-indexing
-          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.3
+          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.5
           imagePullPolicy: Always
           command:
             [
diff --git a/deployment/cloud_kubernetes/workers/light_worker.yaml b/deployment/cloud_kubernetes/workers/light_worker.yaml
index a4ceb27a6b1..d1fd5b2ed69 100644
--- a/deployment/cloud_kubernetes/workers/light_worker.yaml
+++ b/deployment/cloud_kubernetes/workers/light_worker.yaml
@@ -14,7 +14,7 @@ spec:
     spec:
       containers:
         - name: celery-worker-light
-          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.3
+          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.5
           imagePullPolicy: Always
           command:
             [
diff --git a/deployment/cloud_kubernetes/workers/primary.yaml b/deployment/cloud_kubernetes/workers/primary.yaml
index 09c33e21e65..c9a6bbe6a48 100644
--- a/deployment/cloud_kubernetes/workers/primary.yaml
+++ b/deployment/cloud_kubernetes/workers/primary.yaml
@@ -14,7 +14,7 @@ spec:
     spec:
       containers:
         - name: celery-worker-primary
-          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.3
+          image: danswer/danswer-backend-cloud:v0.12.0-cloud.beta.5
           imagePullPolicy: Always
           command:
             [