diff --git a/api/core/model_runtime/model_providers/azure_openai/llm/llm.py b/api/core/model_runtime/model_providers/azure_openai/llm/llm.py
index d12e21a6e8..b52b19f417 100644
--- a/api/core/model_runtime/model_providers/azure_openai/llm/llm.py
+++ b/api/core/model_runtime/model_providers/azure_openai/llm/llm.py
@@ -309,7 +309,7 @@ class AzureOpenAILargeLanguageModel(_CommonAzureOpenAI, LargeLanguageModel):
 
         # transform response
         response = LLMResult(
-            model=response.model,
+            model=response.model or model,
             prompt_messages=prompt_messages,
             message=assistant_prompt_message,
             usage=usage,