diff --git a/rlm/clients/azure_openai.py b/rlm/clients/azure_openai.py
index e65d9e1e2..90e04da3f 100644
--- a/rlm/clients/azure_openai.py
+++ b/rlm/clients/azure_openai.py
@@ -39,6 +39,9 @@ def __init__(
         if api_version is None:
             api_version = os.getenv("AZURE_OPENAI_API_VERSION", "2024-02-01")
 
+        if azure_deployment is None:
+            azure_deployment = os.getenv("AZURE_OPENAI_DEPLOYMENT")
+
         if azure_endpoint is None:
             raise ValueError(
                 "azure_endpoint is required for Azure OpenAI client. "