feat: support Z.ai coding platform (#35)

jiakeboge · web-flow · commit eb969e72a611 · 2026-04-15T21:19:11.000+08:00
* feat: support Z.ai coding platform (#34) * fix: address PR feedback (unit tests, docs cleanup, news revert)
diff --git a/README.md b/README.md
@@ -226,7 +226,7 @@ vibe-trading-mcp               # start MCP server (stdio)
 - **Python 3.11+** for Path B
 - **Docker** for Path A
 
-> **Supported LLM providers:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Ollama (local). See `.env.example` for config.
+> **Supported LLM providers:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Z.ai, Ollama (local). See `.env.example` for config.
 
 > **Tip:** All markets work without any API keys thanks to automatic fallback. yfinance (HK/US), OKX (crypto), and AKShare (A-shares, US, HK, futures, forex) are all free. Tushare token is optional — AKShare covers A-shares as a free fallback.
 
diff --git a/README_ar.md b/README_ar.md
@@ -226,7 +226,7 @@ vibe-trading-mcp               # بدء خادم MCP (stdio)
 - **Python 3.11+** للمسار B
 - **Docker** للمسار A
 
-> **مزودو نماذج اللغة المدعومون:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Ollama (محلي). راجع `.env.example` للإعدادات.
+> **مزودو نماذج اللغة المدعومون:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Z.ai, Ollama (محلي). راجع `.env.example` للإعدادات.
 
 > **نصيحة:** جميع الأسواق تعمل بدون أي مفاتيح API بفضل البديل التلقائي. yfinance (HK/US) و OKX (العملات المشفرة) و AKShare (أسهم A، US، HK، العقود الآجلة، الفوركس) جميعها مجانية. رمز Tushare اختياري — AKShare يغطي أسهم A كبديل مجاني.
 
@@ -300,7 +300,7 @@ npx clawhub@latest install vibe-trading --force
 
 | المتغير | مطلوب | الوصف |
 |----------|:--------:|-------------|
-| `LANGCHAIN_PROVIDER` | نعم | اسم المزود (`openrouter`, `deepseek`, `groq`, `ollama`, إلخ) |
+| `LANGCHAIN_PROVIDER` | نعم | اسم المزود (`openrouter`, `deepseek`, `groq`, `z.ai`, `ollama`, إلخ) |
 | `<PROVIDER>_API_KEY` | نعم* | مفتاح API (`OPENROUTER_API_KEY`, `DEEPSEEK_API_KEY`, إلخ) |
 | `<PROVIDER>_BASE_URL` | نعم | رابط نقطة نهاية API |
 | `LANGCHAIN_MODEL_NAME` | نعم | اسم النموذج (مثلاً `deepseek/deepseek-v3.2`) |
diff --git a/README_ja.md b/README_ja.md
@@ -224,7 +224,7 @@ vibe-trading-mcp               # MCPサーバー（stdio）起動
 - Path Bでは**Python 3.11+**
 - Path Aでは**Docker**
 
-> **サポートされるLLMプロバイダー:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Ollama（ローカル）。設定は`.env.example`を参照。
+> **サポートされるLLMプロバイダー:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Z.ai, Ollama（ローカル）。設定は`.env.example`を参照。
 
 > **Tip:** すべての市場でAPIキーなしでも動作（自動フォールバック）。yfinance（HK/US）、OKX（暗号）、AKShare（A株・US・HK・先物・FX）は無料。Tushareトークンは任意 — A株はAKShareで無料フォールバック。
 
diff --git a/README_ko.md b/README_ko.md
@@ -224,7 +224,7 @@ vibe-trading-mcp               # MCP 서버 시작(stdio)
 - 경로 B용 **Python 3.11+**
 - 경로 A용 **Docker**
 
-> **지원 LLM 제공자:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Ollama(로컬). 설정은 `.env.example` 참고.
+> **지원 LLM 제공자:** OpenRouter, OpenAI, DeepSeek, Gemini, Groq, DashScope/Qwen, Zhipu, Moonshot/Kimi, MiniMax, Xiaomi MIMO, Z.ai, Ollama(로컬). 설정은 `.env.example` 참고.
 
 > **팁:** 모든 시장은 자동 폴백 덕분에 API 키 없이도 작동합니다. yfinance(HK/US), OKX(크립토), AKShare(A주, 미국, HK, 선물, FX)는 모두 무료입니다. Tushare 토큰은 선택 사항 — AKShare가 A주 무료 폴백을 제공합니다.
 
@@ -298,7 +298,7 @@ npx clawhub@latest install vibe-trading --force
 
 | Variable | Required | Description |
 |----------|:--------:|-------------|
-| `LANGCHAIN_PROVIDER` | Yes | 제공자 이름(`openrouter`, `deepseek`, `groq`, `ollama` 등) |
+| `LANGCHAIN_PROVIDER` | Yes | 제공자 이름(`openrouter`, `deepseek`, `groq`, `z.ai`, `ollama` 등) |
 | `<PROVIDER>_API_KEY` | Yes* | API 키(`OPENROUTER_API_KEY`, `DEEPSEEK_API_KEY` 등) |
 | `<PROVIDER>_BASE_URL` | Yes | API 엔드포인트 URL |
 | `LANGCHAIN_MODEL_NAME` | Yes | 모델 이름(예: `deepseek/deepseek-v3.2`) |
diff --git a/README_zh.md b/README_zh.md
@@ -224,7 +224,7 @@ vibe-trading-mcp               # 启动 MCP 服务器（stdio）
 - Path B 需 **Python 3.11+**
 - Path A 需 **Docker**
 
-> **支持的 LLM 提供商：** OpenRouter、OpenAI、DeepSeek、Gemini、Groq、DashScope/Qwen、智谱、Moonshot/Kimi、MiniMax、小米 MIMO、Ollama（本地）。参见 `.env.example` 配置。
+> **支持的 LLM 提供商：** OpenRouter、OpenAI、DeepSeek、Gemini、Groq、DashScope/Qwen、智谱、Moonshot/Kimi、MiniMax、小米 MIMO、Z.ai、Ollama（本地）。参见 `.env.example` 配置。
 
 > **提示：** 所有市场都可在无 API key 情况下运行，因自动回退。yfinance（港美股）、OKX（加密）、AKShare（A 股、美股、港股、期货、外汇）均免费。Tushare token 可选——A 股可回退到 AKShare 免费获取。
 
@@ -298,7 +298,7 @@ npx clawhub@latest install vibe-trading --force
 
 | Variable | Required | Description |
 |----------|:--------:|-------------|
-| `LANGCHAIN_PROVIDER` | Yes | 提供商名称（`openrouter`、`deepseek`、`groq`、`ollama` 等） |
+| `LANGCHAIN_PROVIDER` | Yes | 提供商名称（`openrouter`、`deepseek`、`groq`、`z.ai`、`ollama` 等） |
 | `<PROVIDER>_API_KEY` | Yes* | API key（`OPENROUTER_API_KEY`、`DEEPSEEK_API_KEY` 等） |
 | `<PROVIDER>_BASE_URL` | Yes | API 端点 URL |
 | `LANGCHAIN_MODEL_NAME` | Yes | 模型名（如 `deepseek/deepseek-v3.2`） |
diff --git a/agent/.env.example b/agent/.env.example
@@ -63,6 +63,12 @@ OPENROUTER_BASE_URL=https://openrouter.ai/api/v1
 # MIMO_API_KEY=xxx
 # MIMO_BASE_URL=https://api.xiaomimimo.com/v1
 
+# --- Z.ai (for Coding Plan) ---
+# LANGCHAIN_PROVIDER=zai
+# LANGCHAIN_MODEL_NAME=glm-5.1
+# ZAI_API_KEY=xxx
+# ZAI_BASE_URL=https://api.z.ai/api/coding/paas/v4
+
 # --- Ollama (local) ---
 # LANGCHAIN_PROVIDER=ollama
 # LANGCHAIN_MODEL_NAME=qwen2.5:32b
diff --git a/agent/cli.py b/agent/cli.py
@@ -1519,6 +1519,16 @@ def _handle_prompt_command(
         "key_prefix": None,
         "key_placeholder": "api-key...",
     },
+    {
+        "label": "Z.ai (Coding platform)",
+        "provider": "zai",
+        "key_env": "ZAI_API_KEY",
+        "base_env": "ZAI_BASE_URL",
+        "base_url": "https://api.z.ai/api/coding/paas/v4",
+        "model": "glm-5.1",
+        "key_prefix": None,
+        "key_placeholder": "api-key...",
+    },
     {
         "label": "Ollama (local, free)",
         "provider": "ollama",
@@ -1564,6 +1574,8 @@ def _render_env_content(config: dict[str, str]) -> str:
         "MINIMAX_BASE_URL",
         "MIMO_API_KEY",
         "MIMO_BASE_URL",
+        "ZAI_API_KEY",
+        "ZAI_BASE_URL",
         "OLLAMA_BASE_URL",
         "LANGCHAIN_MODEL_NAME",
         "TUSHARE_TOKEN",
diff --git a/agent/src/providers/llm.py b/agent/src/providers/llm.py
@@ -79,6 +79,7 @@ def _sync_provider_env() -> None:
         "moonshot":   ("MOONSHOT_API_KEY",    "MOONSHOT_BASE_URL"),
         "minimax":    ("MINIMAX_API_KEY",     "MINIMAX_BASE_URL"),
         "mimo":       ("MIMO_API_KEY",        "MIMO_BASE_URL"),
+        "zai":        ("ZAI_API_KEY",         "ZAI_BASE_URL"),
         "ollama":     (None,                  "OLLAMA_BASE_URL"),
     }
 
diff --git a/agent/tests/test_llm.py b/agent/tests/test_llm.py
@@ -24,7 +24,7 @@ def _run_sync(self, env: dict[str, str]) -> dict[str, str]:
         import src.providers.llm as llm_mod
         llm_mod._dotenv_loaded = True  # pretend already loaded
 
-        clean = {k: v for k, v in os.environ.items() if not k.startswith(("OPENAI_", "LANGCHAIN_", "DEEPSEEK_", "GROQ_", "OLLAMA_", "DASHSCOPE_"))}
+        clean = {k: v for k, v in os.environ.items() if not k.startswith(("OPENAI_", "LANGCHAIN_", "DEEPSEEK_", "GROQ_", "OLLAMA_", "DASHSCOPE_", "ZAI_"))}
         clean.update(env)
         with patch.dict(os.environ, clean, clear=True):
             _sync_provider_env()
@@ -75,6 +75,15 @@ def test_qwen_alias_to_dashscope(self) -> None:
         })
         assert result["OPENAI_API_KEY"] == "qwen-key"
 
+    def test_zai_provider(self) -> None:
+        result = self._run_sync({
+            "LANGCHAIN_PROVIDER": "zai",
+            "ZAI_API_KEY": "zai-key-test",
+            "ZAI_BASE_URL": "https://api.z.ai/api/coding/paas/v4",
+        })
+        assert result["OPENAI_API_KEY"] == "zai-key-test"
+        assert result["OPENAI_API_BASE"] == "https://api.z.ai/api/coding/paas/v4"
+
     def test_unknown_provider_falls_back_to_openai(self) -> None:
         result = self._run_sync({
             "LANGCHAIN_PROVIDER": "unknown_provider_xyz",

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,7 @@ def _sync_provider_env() -> None:`
`79`	`79`	`"moonshot": ("MOONSHOT_API_KEY", "MOONSHOT_BASE_URL"),`
`80`	`80`	`"minimax": ("MINIMAX_API_KEY", "MINIMAX_BASE_URL"),`
`81`	`81`	`"mimo": ("MIMO_API_KEY", "MIMO_BASE_URL"),`
	`82`	`+ "zai": ("ZAI_API_KEY", "ZAI_BASE_URL"),`
`82`	`83`	`"ollama": (None, "OLLAMA_BASE_URL"),`
`83`	`84`	`}`
`84`	`85`