@arnilo/prism-provider-zai 0.0.2 → 0.0.4

package/CHANGELOG.md

@@ -7,6 +7,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.0.4] - 2026-07-14
+
+### Changed
+
+- Uses shared bounded transport/OpenAI helpers with GLM thinking/reasoning, structured-output, multimodal, tool-stream, telemetry, and protected-header behavior documented per model.
+
 ## [0.0.2] - 2026-07-05
 
 ### Added

package/dist/models.js

@@ -1,5 +1,5 @@
 export function defineZaiModel(config) {
-    return { ...config, provider: "zai", capabilities: { input: ["text"], output: ["text"], reasoning: true, tools: true, streaming: true, ...config.capabilities } };
+    return { ...config, provider: "zai", capabilities: { input: ["text"], output: ["text"], reasoning: true, tools: true, streaming: true, structuredOutput: "json_schema", ...config.capabilities } };
 }
 export const zaiModels = [
     defineZaiModel({

package/dist/provider.d.ts

@@ -1,4 +1,5 @@
-import type { AIProvider, CredentialValueSource, JsonObject, ProviderEvent, ProviderRequest } from "@arnilo/prism";
+import type { AIProvider, JsonObject, ProviderEvent, ProviderRequest } from "@arnilo/prism";
+import { type CredentialValueSource } from "@arnilo/prism";
 export interface ZaiProviderOptions {
     readonly id?: string;
     readonly baseUrl?: string;
@@ -7,4 +8,4 @@ export interface ZaiProviderOptions {
 }
 export declare function createZaiProvider(options?: ZaiProviderOptions): AIProvider;
 export declare function zaiBody(request: ProviderRequest): JsonObject;
-export declare function zaiEvents(body: ReadableStream<Uint8Array>): AsyncIterable<ProviderEvent>;
+export declare function zaiEvents(body: ReadableStream<Uint8Array>, signal?: AbortSignal): AsyncIterable<ProviderEvent>;

package/dist/provider.js

@@ -1,5 +1,6 @@
-import { providerDone, providerError, providerTextDelta, providerThinkingDelta, providerToolCall, providerToolCallDelta, providerUsage, resolveCredentialValue, toolCallContent } from "@arnilo/prism";
-import { readSseData } from "./sse.js";
+import { assertStructuredOutputRequestSupported, providerDone, providerError, providerTextDelta, providerThinkingDelta, providerToolCall, providerToolCallDelta, providerUsage, resolveCredentialValue, toolCallContent } from "@arnilo/prism";
+import { applyOpenAIChatStructuredOutput, mapOpenAIChatUsage, serializeOpenAIChatMessage, serializeOpenAITool, } from "@arnilo/prism/providers/openai";
+import { parseJsonObjectArguments, readBoundedResponseText, readSseData, } from "@arnilo/prism/providers/transport";
 import { zaiReasoningEffort, zaiThinking, zaiToolStream } from "./thinking.js";
 export function createZaiProvider(options = {}) {
     const id = options.id ?? "zai";
@@ -18,11 +19,12 @@ export function createZaiProvider(options = {}) {
                     body: JSON.stringify(zaiBody(request)),
                     signal: request.signal,
                 });
-                if (!response.ok)
-                    return yield providerError(new Error(`Z.AI request failed: ${response.status} ${await safeText(response)}`), secrets);
+                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);
+                yield* zaiEvents(response.body, request.signal);
             }
             catch (error) {
                 yield providerError(error, secrets);
@@ -31,11 +33,12 @@ export function createZaiProvider(options = {}) {
     };
 }
 export function zaiBody(request) {
+    assertStructuredOutputRequestSupported(request.model, request.options);
     const { maxTokens, ...parameters } = request.model.parameters ?? {};
-    return clean({
+    const body = {
         model: request.model.model,
-        messages: request.messages.map((message) => toMessage(message, request.model.capabilities ?? {})),
-        tools: request.tools?.map(toTool),
+        messages: request.messages.map((message) => serializeOpenAIChatMessage(message, request.model.capabilities ?? {})),
+        tools: request.tools?.map(serializeOpenAITool),
         stream: true,
         tool_stream: zaiToolStream(request),
         thinking: zaiThinking(request),
@@ -44,18 +47,23 @@ export function zaiBody(request) {
         max_tokens: maxTokens ?? request.model.limits?.maxOutputTokens,
         ...request.options?.compat,
         ...request.options?.extra,
-    });
+    };
+    applyOpenAIChatStructuredOutput(body, request.options?.structuredOutput);
+    return clean(body);
 }
-export async function* zaiEvents(body) {
+export async function* zaiEvents(body, signal) {
     const tools = new Map();
     let usage;
-    for await (const data of readSseData(body)) {
+    for await (const data of readSseData(body, { signal })) {
         if (data === "[DONE]")
             break;
         const chunk = JSON.parse(data);
-        usage = toUsage(chunk.usage) ?? usage;
-        if (chunk.usage)
-            yield providerUsage(toUsage(chunk.usage));
+        usage = mapOpenAIChatUsage(chunk.usage) ?? usage;
+        if (chunk.usage) {
+            const mapped = mapOpenAIChatUsage(chunk.usage);
+            if (mapped)
+                yield providerUsage(mapped);
+        }
         for (const choice of chunk.choices ?? []) {
             const delta = choice.delta ?? {};
             if (delta.content)
@@ -73,82 +81,14 @@ export async function* zaiEvents(body) {
             }
         }
     }
-    for (const call of tools.values())
-        if (call.id && call.name)
-            yield providerToolCall(toolCallContent(call.id, call.name, parseArgs(call.argumentsText)));
-    yield providerDone(usage);
-}
-function toMessage(message, capabilities = {}) {
-    if (message.role === "tool") {
-        const result = message.content.find((part) => part.type === "tool_result");
-        return {
-            role: "tool",
-            tool_call_id: result?.toolCallId ?? "",
-            content: result ? JSON.stringify(result.result ?? result.error ?? null) : "",
-        };
-    }
-    if (message.role === "assistant") {
-        const toolCalls = message.content.filter((part) => part.type === "tool_call");
-        const textParts = message.content.filter((part) => part.type === "text" || part.type === "thinking");
-        if (toolCalls.length > 0) {
-            return {
-                role: "assistant",
-                content: textParts.map((part) => part.text).join("\n") || null,
-                tool_calls: toolCalls.map((call) => ({
-                    id: call.id,
-                    type: "function",
-                    function: { name: call.name, arguments: JSON.stringify(call.arguments) },
-                })),
-            };
+    for (const call of tools.values()) {
+        if (call.id && call.name) {
+            yield providerToolCall(toolCallContent(call.id, call.name, parseJsonObjectArguments(call.argumentsText, { toolName: call.name })));
         }
     }
-    const content = [];
-    for (const part of message.content) {
-        if (part.type === "text" || part.type === "thinking") {
-            content.push({ type: "text", text: part.text });
-        }
-        else if (part.type === "image") {
-            if (!capabilities.input?.includes("image")) {
-                throw new Error(`Z.AI request includes image but model does not declare image input capability`);
-            }
-            const url = part.url ?? (part.data ? `data:${part.mimeType ?? "image/png"};base64,${part.data}` : undefined);
-            if (!url)
-                throw new Error("Z.AI image block missing url or data");
-            content.push({ type: "image_url", image_url: { url } });
-        }
-        else if (part.type === "tool_call") {
-            throw new Error("Z.AI assistant tool_call blocks must be the only content on the message");
-        }
-        else if (part.type === "tool_result") {
-            throw new Error("Z.AI tool_result blocks must appear in role=tool messages");
-        }
-    }
-    if (content.length === 1 && content[0].type === "text") {
-        return { role: message.role, content: content[0].text };
-    }
-    return { role: message.role, content };
-}
-function toTool(tool) {
-    return clean({ type: "function", function: { name: tool.name, description: tool.description, parameters: tool.parameters ?? { type: "object" } } });
-}
-function toUsage(usage) {
-    return usage ? { inputTokens: usage.prompt_tokens, outputTokens: usage.completion_tokens, totalTokens: usage.total_tokens, cacheReadTokens: usage.prompt_tokens_details?.cached_tokens, cacheWriteTokens: usage.prompt_tokens_details?.cache_write_tokens } : undefined;
-}
-function parseArgs(text) {
-    if (!text)
-        return {};
-    const parsed = JSON.parse(text);
-    return parsed && typeof parsed === "object" && !Array.isArray(parsed) ? parsed : {};
+    yield providerDone(usage);
 }
 function clean(value) {
     return Object.fromEntries(Object.entries(value).filter(([, item]) => item !== undefined));
 }
-async function safeText(response) {
-    try {
-        return await response.text();
-    }
-    catch {
-        return "";
-    }
-}
 //# sourceMappingURL=provider.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arnilo/prism-provider-zai",
-  "version": "0.0.2",
+  "version": "0.0.4",
   "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.2"
+    "@arnilo/prism": "0.0.4"
   },
   "devDependencies": {
     "@arnilo/prism": "file:../.."

package/dist/sse.d.ts

@@ -1 +0,0 @@
-export declare function readSseData(body: ReadableStream<Uint8Array>): AsyncIterable<string>;

package/dist/sse.js

@@ -1,24 +0,0 @@
-export async function* readSseData(body) {
-    const reader = body.getReader();
-    const decoder = new TextDecoder();
-    let buffer = "";
-    while (true) {
-        const { done, value } = await reader.read();
-        if (done)
-            break;
-        buffer += decoder.decode(value, { stream: true });
-        const parts = buffer.split(/\r?\n\r?\n/);
-        buffer = parts.pop() ?? "";
-        for (const part of parts)
-            yield* eventData(part);
-    }
-    buffer += decoder.decode();
-    if (buffer)
-        yield* eventData(buffer);
-}
-function* eventData(event) {
-    const data = event.split(/\r?\n/).filter((line) => line.startsWith("data:")).map((line) => line.slice(5).trimStart()).join("\n").trim();
-    if (data)
-        yield data;
-}
-//# sourceMappingURL=sse.js.map