diff --git a/src/cai/sdk/agents/models/openai_chatcompletions.py b/src/cai/sdk/agents/models/openai_chatcompletions.py
index 48d15683..60602da0 100644
--- a/src/cai/sdk/agents/models/openai_chatcompletions.py
+++ b/src/cai/sdk/agents/models/openai_chatcompletions.py
@@ -3677,6 +3677,7 @@ class OpenAIChatCompletionsModel(Model):
                 "llama", "nemotron", "codellama", "mixtral", "phi",
                 "qwen", "yi", "deepseek", "gemma", "stable-diffusion",
                 "sdxl", "cosxl", "proteus", "realistic-vision",
+                "z-ai",
             }
             return provider in nim_providers
 
@@ -3804,6 +3805,7 @@ class OpenAIChatCompletionsModel(Model):
                         "llama", "nemotron", "codellama", "mixtral", "phi",
                         "qwen", "yi", "deepseek", "gemma", "stable-diffusion",
                         "sdxl", "cosxl", "proteus", "realistic-vision",
+                        "z-ai",
                     }
                     return provider in nim_providers
                 if (