diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py
index ddf296353..52907b4d6 100644
--- a/rag/llm/chat_model.py
+++ b/rag/llm/chat_model.py
@@ -1659,7 +1659,7 @@ class LiteLLMBase(ABC):
                     "api_base": self.base_url,
                 }
             )
-        elif self.provider == SupportedLiteLLMProvider.GPUStack:
+        elif self.provider == SupportedLiteLLMProvider.Azure_OpenAI:
             completion_args.pop("api_key", None)
             completion_args.pop("api_base", None)
             completion_args.update(