From f9e886f7923342157a9d3ea42d25806bc325221c Mon Sep 17 00:00:00 2001
From: UnaiAlias <52742669+UnaiAlias@users.noreply.github.com>
Date: Fri, 31 Jul 2026 19:10:50 +0000
Subject: [PATCH] fix: add z-ai to NIM model providers

z-ai/glm-5.2 is hosted on NVIDIA NIM (build.nvidia.com/z-ai/glm-5.2).
The NIM rotation was incorrectly excluding it. Add "z-ai" to the
known NIM provider list in both LiteLLM and direct httpx paths.
---
 src/cai/sdk/agents/models/openai_chatcompletions.py | 2 ++
 1 file changed, 2 insertions(+)

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 (