This commit is contained in:
Chaitanya Laxman 2026-09-12 13:49:14 +02:00 committed by GitHub
commit f477390888
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 59 additions and 1 deletions

View file

@ -3421,6 +3421,34 @@
],
"default_model_placeholder": "gpt-3.5-turbo"
},
{
"provider": "ZAI",
"provider_display_name": "Z.AI (Zhipu AI)",
"litellm_provider": "zai",
"credential_fields": [
{
"key": "api_base",
"label": "API Base",
"placeholder": "https://api.z.ai/api/paas/v4",
"tooltip": null,
"required": false,
"field_type": "text",
"options": null,
"default_value": null
},
{
"key": "api_key",
"label": "API Key",
"placeholder": null,
"tooltip": null,
"required": true,
"field_type": "password",
"options": null,
"default_value": null
}
],
"default_model_placeholder": "zai/glm-4.5"
},
{
"provider": "CURSOR",
"provider_display_name": "Cursor",

View file

@ -328,6 +328,37 @@ def test_cognition_provider_fields():
assert fields_by_key["api_base"]["required"] is False
def test_zai_provider_fields():
"""Z.AI must be selectable in the Add Credential flow (#39310).
The form is driven by GET /public/providers/fields. Without a ZAI row the
UI shows only Credential Name and Provider, so the key cannot be saved.
"""
app_instance = FastAPI()
app_instance.include_router(router)
test_client = TestClient(app_instance)
response = test_client.get("/public/providers/fields")
assert response.status_code == 200
providers = response.json()
zai = next((p for p in providers if p["provider"] == "ZAI"), None)
assert zai is not None, "Z.AI provider entry not found"
assert zai["provider_display_name"] == "Z.AI (Zhipu AI)"
assert zai["litellm_provider"] == "zai"
assert zai["default_model_placeholder"].startswith("zai/")
fields_by_key = {f["key"]: f for f in zai["credential_fields"]}
assert fields_by_key["api_key"]["required"] is True
assert fields_by_key["api_key"]["field_type"] == "password"
assert fields_by_key["api_base"]["required"] is False
assert fields_by_key["api_base"]["field_type"] == "text"
assert fields_by_key["api_base"]["placeholder"] == "https://api.z.ai/api/paas/v4"
def test_chatgpt_provider_fields():
app_instance = FastAPI()
app_instance.include_router(router)
@ -386,7 +417,6 @@ ADD_MODEL_UNLISTED_PROVIDERS: Final = frozenset(
"text-completion-inception",
"valkey",
"xiaomi_mimo",
"zai",
}
)