diff --git a/ai.yaml b/ai.yaml
index 98a23f0..b1c3437 100644
--- a/ai.yaml
+++ b/ai.yaml
@@ -57,7 +57,7 @@ services:
       - traefik.http.services.open-webui.loadbalancer.sticky.cookie.httponly=true
       - traefik.swarm.network=traefik_backend
   litellm:
-    image: docker.litellm.ai/berriai/litellm:1.90.0
+    image: docker.litellm.ai/berriai/litellm-database:main-latest
     command: --config /app/config.yaml --port 4000
     environment:
     - AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}