File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1515from .vertex_adapter import VertexAdapter
1616from .gemini_openai_adapter import GeminiOpenAIAdapter
1717from .alibaba_adapter import AlibabaAdapter
18+ from .zai_adapter import ZAIAdapter
1819
1920
2021class ProviderAdapterFactory :
@@ -181,6 +182,10 @@ class ProviderAdapterFactory:
181182 "base_url" : "https://open.bigmodel.cn/api/paas/v4" ,
182183 "adapter" : ZhipuAdapter ,
183184 },
185+ "z.ai" : {
186+ "base_url" : "https://api.z.ai/api/paas/v4" ,
187+ "adapter" : ZAIAdapter ,
188+ },
184189 "azure" : {
185190 "adapter" : AzureAdapter ,
186191 },
Original file line number Diff line number Diff line change 1+ from .openai_adapter import OpenAIAdapter
2+
3+ # https://docs.z.ai/api-reference/llm/chat-completion#body-model
4+ ZAI_MODELS = [
5+ "glm-4.5" ,
6+ "glm-4.5-air" ,
7+ "glm-4.5-x" ,
8+ "glm-4.5-airx" ,
9+ "glm-4.5-flash" ,
10+ "glm-4-32b-0414-128k" ,
11+ ]
12+
13+
14+ class ZAIAdapter (OpenAIAdapter ):
15+ """Adapter for Zai API"""
16+
17+ async def list_models (self , api_key : str ) -> list [str ]:
18+ return ZAI_MODELS
You can’t perform that action at this time.
0 commit comments