diff --git a/docker-compose.yml b/docker-compose.yml
index 49270f4..e2c0d96 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -80,10 +80,13 @@ services:
     restart: unless-stopped
     environment:
       DATABASE_STRING: postgresql+psycopg://skyvern:${SKYVERN_DB_PASSWORD:-skyvern}@skyvern-db:5432/skyvern
-      OPENAI_API_KEY: ${LITELLM_API_KEY}
-      OPENAI_API_BASE: ${LITELLM_BASE_URL}/v1
-      LLM_KEY: OPENAI_GPT4O
-      SECONDARY_LLM_KEY: OPENAI_GPT4O_MINI
+      ENABLE_OPENAI_COMPATIBLE: "true"
+      OPENAI_COMPATIBLE_API_KEY: ${LITELLM_API_KEY}
+      OPENAI_COMPATIBLE_API_BASE: ${LITELLM_BASE_URL}/v1
+      OPENAI_COMPATIBLE_MODEL_NAME: gpt-4o
+      OPENAI_COMPATIBLE_SUPPORTS_VISION: "true"
+      LLM_KEY: OPENAI_COMPATIBLE
+      SECONDARY_LLM_KEY: OPENAI_COMPATIBLE
       BROWSER_TYPE: chromium-headful
       ENABLE_CODE_BLOCK: "true"
       ENV: local