From 6dbc1d2dbd955aa5d3b6b4fc9ef013f895aef6eb Mon Sep 17 00:00:00 2001
From: LawyZheng <lawyzheng1106@gmail.com>
Date: Mon, 27 May 2024 18:13:10 +0800
Subject: [PATCH] add skyvern container health check (#374)

---
 docker-compose.yml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 45877aad..a2e8c8e3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -37,6 +37,7 @@ services:
       - OPENAI_API_KEY=<your_openai_key>
       # If you want to use other LLM provider, like azure and anthropic:
       # - ENABLE_ANTHROPIC=true
+      # - LLM_KEY=ANTHROPIC_CLAUDE3_OPUS
       # - ANTHROPIC_API_KEY=<your_anthropic_key>
       # - ENABLE_AZURE=true
       # - LLM_KEY=AZURE_OPENAI_GPT4V
@@ -47,6 +48,11 @@ services:
     depends_on:
       postgres:
         condition: service_healthy
+    healthcheck:
+      test: ["CMD", "test", "-f", "/app/.streamlit/secrets.toml"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
   
   streamlit:
     image: public.ecr.aws/t6d4b5t4/skyvern:latest
@@ -60,5 +66,6 @@ services:
       - ./.streamlit:/app/.streamlit
     command: ["/bin/bash", "entrypoint-streamlit.sh"]
     depends_on:
-      - skyvern
+      skyvern:
+        condition: service_healthy