diff --git a/docker-compose.yml b/docker-compose.yml
index d6912f8..94860e9 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,3 +1,5 @@
+version: '3.8'
+
 services:
   codegenerator:
     build: ./CodeGenerator
@@ -6,6 +8,8 @@ services:
       - "8000:8000"
     environment:
       - OLLAMA_URL=http://host.docker.internal:11434
+    volumes:
+      - /var/run/docker.sock:/var/run/docker.sock
     restart: unless-stopped
 
   open-webui: