@arnilo/prism-provider-zai 0.0.16 → 0.0.18

package/CHANGELOG.md

@@ -1,5 +1,15 @@
 # Changelog
 
+## [0.0.18] - 2026-07-30
+
+### Changed
+- Released with exact 0.0.18 graph.
+
+## [0.0.17] - 2026-07-29
+
+### Changed
+- Released with exact 0.0.17 graph.
+
 ## [0.0.16] - 2026-07-26
 
 ### Changed

package/dist/models.js

@@ -24,7 +24,7 @@ export function defineZaiModel(config) {
  */
 export async function listZaiModels(options = {}) {
     const provider = options.provider ?? "zai";
-    const baseUrl = (options.baseUrl ?? "https://api.z.ai/api/paas/v4").replace(/\/$/, "");
+    const baseUrl = (options.baseUrl ?? "https://api.z.ai/api/paas/v4").replace(/\/+$/, "");
     const token = await resolveCredentialValue(options.apiKey, { provider, name: "apiKey" });
     const response = await (options.fetch ?? fetch)(`${baseUrl}/models`, {
         method: "GET",

package/dist/provider.js

@@ -1,107 +1,43 @@
-import { assertStructuredOutputRequestSupported, providerDone, providerError, providerTextDelta, providerThinkingDelta, providerToolCall, providerToolCallDelta, providerUsage, resolveCredentialValue, toolCallFromArgumentsText, } from "@arnilo/prism";
-import { applyOpenAIChatStructuredOutput, mapOpenAIChatUsage, serializeOpenAITool } from "@arnilo/prism/providers/openai";
-import { readBoundedResponseText, readSseData } from "@arnilo/prism/providers/transport";
+import { applyOpenAIChatStructuredOutput } from "@arnilo/prism/providers/openai";
+import { buildOpenAIChatBody, createOpenAICompatibleProvider, openAIChatEvents } from "@arnilo/prism/providers/openai-compatible";
 import { zaiPreserveThinking, zaiReasoningEffort, zaiThinking, zaiToolStream } from "./thinking.js";
 /** Official international Chat Completions base (China `open.bigmodel.cn` remains overridable). */
 export const ZAI_DEFAULT_BASE_URL = "https://api.z.ai/api/paas/v4";
 export function createZaiProvider(options = {}) {
-    const id = options.id ?? "zai";
-    const baseUrl = (options.baseUrl ?? ZAI_DEFAULT_BASE_URL).replace(/\/$/, "");
-    return {
-        id,
-        async *generate(request) {
-            if (request.signal?.aborted)
-                throw request.signal.reason ?? new Error("aborted");
-            const token = await resolveCredentialValue(options.apiKey, { provider: id, name: "apiKey" });
-            const secrets = [token];
-            try {
-                const response = await (options.fetch ?? fetch)(`${baseUrl}/chat/completions`, {
-                    method: "POST",
-                    headers: {
-                        ...request.options?.headers,
-                        "content-type": "application/json",
-                        ...(token ? { authorization: `Bearer ${token}` } : {}),
-                    },
-                    body: JSON.stringify(zaiBody(request)),
-                    signal: request.signal,
-                });
-                if (!response.ok) {
-                    return yield providerError(new Error(`Z.AI request failed: ${response.status} ${await readBoundedResponseText(response, { secrets })}`), secrets);
-                }
-                if (!response.body)
-                    return yield providerError(new Error("Z.AI response had no body"), secrets);
-                yield* zaiEvents(response.body, request.signal);
-            }
-            catch (error) {
-                yield providerError(error, secrets);
-            }
-        },
-    };
+    return createOpenAICompatibleProvider({
+        id: options.id ?? "zai",
+        baseUrl: (options.baseUrl ?? ZAI_DEFAULT_BASE_URL).replace(/\/+$/, ""),
+        apiKey: options.apiKey,
+        fetch: options.fetch,
+        doneUsage: true,
+        requestFailedPrefix: "Z.AI request failed",
+        serializeMessage: (message, request) => toZaiMessage(message, request.model, zaiPreserveThinking(request)),
+        transformBody: (body, request) => zaiTransform(body, request),
+    });
 }
 export function zaiBody(request) {
-    assertStructuredOutputRequestSupported(request.model, request.options);
-    const { maxTokens, ...parameters } = request.model.parameters ?? {};
-    const compatRest = stripZaiManagedCompat(request.options?.compat);
-    const preserveThinking = zaiPreserveThinking(request);
-    const body = {
-        model: request.model.model,
-        messages: request.messages.map((message) => toZaiMessage(message, request.model, preserveThinking)),
-        tools: request.tools?.map(serializeOpenAITool),
-        stream: true,
-        ...parameters,
+    return buildOpenAIChatBody(request, {
+        serializeMessage: (message, req) => toZaiMessage(message, req.model, zaiPreserveThinking(req)),
+        transformBody: (body, req) => zaiTransform(body, req),
+    });
+}
+function zaiTransform(body, request) {
+    const { maxTokens, stream_options: _streamOptions, ...rest } = body;
+    const transformed = {
+        ...rest,
         max_tokens: maxTokens ?? request.model.limits?.maxOutputTokens,
-        ...compatRest,
+        ...stripZaiManagedCompat(request.options?.compat),
         ...request.options?.extra,
         // Resolved official fields win over raw compat/extra escape hatches.
         thinking: zaiThinking(request),
         reasoning_effort: zaiReasoningEffort(request),
         tool_stream: zaiToolStream(request),
     };
-    applyOpenAIChatStructuredOutput(body, request.options?.structuredOutput);
-    return clean(body);
+    applyOpenAIChatStructuredOutput(transformed, request.options?.structuredOutput);
+    return clean(transformed);
 }
-export async function* zaiEvents(body, signal) {
-    const tools = new Map();
-    let usage;
-    for await (const data of readSseData(body, { signal })) {
-        if (data === "[DONE]")
-            break;
-        const chunk = JSON.parse(data);
-        if (chunk.usage) {
-            const mapped = mapOpenAIChatUsage(chunk.usage);
-            if (mapped) {
-                usage = mapped;
-                yield providerUsage(mapped);
-            }
-        }
-        for (const choice of chunk.choices ?? []) {
-            const delta = choice.delta ?? {};
-            if (delta.content)
-                yield providerTextDelta(delta.content);
-            if (delta.reasoning_content)
-                yield providerThinkingDelta(delta.reasoning_content);
-            for (const tool of delta.tool_calls ?? []) {
-                const index = tool.index ?? 0;
-                const current = tools.get(index) ?? { argumentsText: "" };
-                current.id = tool.id ?? current.id;
-                current.name = tool.function?.name ?? current.name;
-                current.argumentsText += tool.function?.arguments ?? "";
-                tools.set(index, current);
-                yield providerToolCallDelta({
-                    index,
-                    id: tool.id,
-                    name: tool.function?.name,
-                    argumentsText: tool.function?.arguments,
-                });
-            }
-        }
-    }
-    for (const call of tools.values()) {
-        if (call.id && call.name) {
-            yield providerToolCall(toolCallFromArgumentsText(call.id, call.name, call.argumentsText));
-        }
-    }
-    yield providerDone(usage);
+export function zaiEvents(body, signal) {
+    return openAIChatEvents(body, { signal, doneUsage: true });
 }
 /**
  * Serialize Prism messages for Z.AI Chat Completions.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arnilo/prism-provider-zai",
-  "version": "0.0.16",
+  "version": "0.0.18",
   "description": "Z.AI provider package for Prism.",
   "type": "module",
   "main": "./dist/index.js",
@@ -25,7 +25,7 @@
     "pack:dry-run": "npm pack --dry-run"
   },
   "peerDependencies": {
-    "@arnilo/prism": "0.0.16"
+    "@arnilo/prism": "0.0.18"
   },
   "devDependencies": {
     "@arnilo/prism": "file:../.."