feat: add Z.AI env support and live test

src/agents/zai.live.test.ts

@@ -0,0 +1,31 @@
+import { completeSimple, getModel } from "@mariozechner/pi-ai";
+import { describe, expect, it } from "vitest";
+
+const ZAI_KEY = process.env.ZAI_API_KEY ?? process.env.Z_AI_API_KEY ?? "";
+const LIVE = process.env.ZAI_LIVE_TEST === "1" || process.env.LIVE === "1";
+
+const describeLive = LIVE && ZAI_KEY ? describe : describe.skip;
+
+describeLive("zai live", () => {
+  it("returns assistant text", async () => {
+    const model = getModel("zai", "glm-4.7");
+    const res = await completeSimple(
+      model,
+      {
+        messages: [
+          {
+            role: "user",
+            content: "Reply with the word ok.",
+            timestamp: Date.now(),
+          },
+        ],
+      },
+      { apiKey: ZAI_KEY, maxTokens: 64 },
+    );
+    const text = res.content
+      .filter((block) => block.type === "text")
+      .map((block) => block.text.trim())
+      .join(" ");
+    expect(text.length).toBeGreaterThan(0);
+  });
+});

src/index.ts

@@ -17,6 +17,7 @@ import {
   saveSessionStore,
 } from "./config/sessions.js";
 import { ensureBinary } from "./infra/binaries.js";
+import { normalizeEnv } from "./infra/env.js";
 import { isMainModule } from "./infra/is-main.js";
 import { ensureClawdisCliOnPath } from "./infra/path-env.js";
 import {
@@ -32,6 +33,7 @@ import { monitorWebProvider } from "./provider-web.js";
 import { assertProvider, normalizeE164, toWhatsappJid } from "./utils.js";
 
 dotenv.config({ quiet: true });
+normalizeEnv();
 ensureClawdisCliOnPath();
 
 // Capture all console output into structured logs while keeping stdout/stderr behavior.

src/infra/env.test.ts

@@ -0,0 +1,37 @@
+import { describe, expect, it } from "vitest";
+
+import { normalizeZaiEnv } from "./env.js";
+
+describe("normalizeZaiEnv", () => {
+  it("copies Z_AI_API_KEY to ZAI_API_KEY when missing", () => {
+    const prevZai = process.env.ZAI_API_KEY;
+    const prevZAi = process.env.Z_AI_API_KEY;
+    process.env.ZAI_API_KEY = "";
+    process.env.Z_AI_API_KEY = "zai-legacy";
+
+    normalizeZaiEnv();
+
+    expect(process.env.ZAI_API_KEY).toBe("zai-legacy");
+
+    if (prevZai === undefined) delete process.env.ZAI_API_KEY;
+    else process.env.ZAI_API_KEY = prevZai;
+    if (prevZAi === undefined) delete process.env.Z_AI_API_KEY;
+    else process.env.Z_AI_API_KEY = prevZAi;
+  });
+
+  it("does not override existing ZAI_API_KEY", () => {
+    const prevZai = process.env.ZAI_API_KEY;
+    const prevZAi = process.env.Z_AI_API_KEY;
+    process.env.ZAI_API_KEY = "zai-current";
+    process.env.Z_AI_API_KEY = "zai-legacy";
+
+    normalizeZaiEnv();
+
+    expect(process.env.ZAI_API_KEY).toBe("zai-current");
+
+    if (prevZai === undefined) delete process.env.ZAI_API_KEY;
+    else process.env.ZAI_API_KEY = prevZai;
+    if (prevZAi === undefined) delete process.env.Z_AI_API_KEY;
+    else process.env.Z_AI_API_KEY = prevZAi;
+  });
+});

src/infra/env.ts

@@ -0,0 +1,9 @@
+export function normalizeZaiEnv(): void {
+  if (!process.env.ZAI_API_KEY?.trim() && process.env.Z_AI_API_KEY?.trim()) {
+    process.env.ZAI_API_KEY = process.env.Z_AI_API_KEY;
+  }
+}
+
+export function normalizeEnv(): void {
+  normalizeZaiEnv();
+}