diff --git a/py-genai/docker-compose.yml b/py-genai/docker-compose.yml
index 764077b9..dfa9bfb8 100644
--- a/py-genai/docker-compose.yml
+++ b/py-genai/docker-compose.yml
@@ -10,6 +10,10 @@ services:
       - LOG_LEVEL=INFO
     x-genai-models:
       - llama
+    models:
+      llama:
+        endpoint_var: LLAMA_URL
+        model_var: LLAMA_MODEL
     healthcheck:
       test: ["CMD", "curl", "-f", "http://localhost:8080/health"]
       interval: 30s