diff --git a/aidial_adapter_dial/utils/reflection.py b/aidial_adapter_dial/utils/reflection.py
index 6366789..a300604 100644
--- a/aidial_adapter_dial/utils/reflection.py
+++ b/aidial_adapter_dial/utils/reflection.py
@@ -25,6 +25,7 @@ async def call_with_extra_body(
         )
 
     arg["extra_body"] = arg.get("extra_body") or {}
+    arg["model"] = arg.get("model")  # model arg is required in AsyncAzureOpenAI
 
     for extra_arg in extra_args:
         arg["extra_body"][extra_arg] = arg[extra_arg]