From d28991fc60258601b8e7549de7a2abc87dbc7b1a Mon Sep 17 00:00:00 2001
From: Sabiha Khan <sabihak89@gmail.com>
Date: Tue, 9 Sep 2025 16:37:05 +0530
Subject: [PATCH] fix storage_backend value and docker image versions

---
 api/db/workflow_run_client.py |  5 +++++
 docker-compose.yaml           | 15 +++++++++++----
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/api/db/workflow_run_client.py b/api/db/workflow_run_client.py
index d71984f..db44e04 100644
--- a/api/db/workflow_run_client.py
+++ b/api/db/workflow_run_client.py
@@ -14,6 +14,7 @@ from api.db.models import (
     WorkflowModel,
     WorkflowRunModel,
 )
+from api.enums import StorageBackend
 from api.schemas.workflow import WorkflowRunResponseSchema
 
 
@@ -67,6 +68,9 @@ class WorkflowRunClient(BaseDBClient):
             )
             current_def = current_def_result.scalars().first()
 
+            # Get the current storage backend based on ENABLE_AWS_S3 flag
+            current_backend = StorageBackend.get_current_backend()
+            
             new_run = WorkflowRunModel(
                 name=name,
                 workflow=workflow,
@@ -75,6 +79,7 @@ class WorkflowRunClient(BaseDBClient):
                 initial_context=initial_context or workflow.template_context_variables,
                 campaign_id=campaign_id,
                 queued_run_id=queued_run_id,
+                storage_backend=current_backend.value,
             )
             session.add(new_run)
             try:
diff --git a/docker-compose.yaml b/docker-compose.yaml
index e66c522..624ddf5 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -59,7 +59,7 @@ services:
       - app-network
 
   api:
-    image: chewieee/dograh-api:v7
+    image: dograhai/dograh-api:v1
     volumes:
       - shared-tmp:/tmp
     environment:
@@ -124,7 +124,7 @@ services:
       - app-network
 
   arq-worker:
-    image: chewieee/dograh-api:v7
+    image: dograhai/dograh-api:v1
     volumes:
       - shared-tmp:/tmp
     environment:
@@ -151,22 +151,29 @@ services:
       # Sentry
       ENABLE_SETRY: "false"
       SENTRY_DSN: ""
+    command: >
+      bash -c "
+        cd /app/api &&
+        python -m arq api.tasks.arq.WorkerSettings
+      "
     depends_on:
       postgres:
         condition: service_healthy
       redis:
         condition: service_healthy
+      minio:
+        condition: service_healthy
     networks:
       - app-network
 
   ui:
-    image: chewieee/dograh-ui:v10
+    image: dograhai/dograh-ui:v1
     environment:
       NEXT_PUBLIC_NODE_ENV: "local"
       NEXT_PUBLIC_AUTH_PROVIDER: "local"
       # Client-side URL (from browser)
       NEXT_PUBLIC_BACKEND_URL: "http://localhost:8000"
-      # Server-side URL (SSR, internal Docker network)
+      # Server-side URL (SSR, internal Docker network)p
       BACKEND_URL: "http://api:8000"
 
       # Posthog