ironside 0.1.0

package/dist/src/wrappers/anthropic.d.ts

@@ -0,0 +1,22 @@
+import type { IronsideClient, TraceHandle } from "../client.js";
+export interface WrapAnthropicOptions {
+    /** Attach generations to an existing trace instead of creating a new standalone trace per call. */
+    trace?: TraceHandle;
+}
+interface MessagesLike {
+    create: (...args: never[]) => unknown;
+}
+interface AnthropicLike {
+    messages: MessagesLike;
+}
+/**
+ * Wraps an Anthropic client instance so every messages.create() call —
+ * streaming or not — is automatically recorded as a generation. Mutates
+ * client.messages in place and returns the same client reference.
+ *
+ * Safe to call more than once on the same client — re-wrapping is detected
+ * and is a no-op, rather than nesting a second wrapper around the first
+ * (which would silently double-record every call).
+ */
+export declare function wrapAnthropic<T extends AnthropicLike>(client: T, ironside: IronsideClient, options?: WrapAnthropicOptions): T;
+export {};

package/dist/src/wrappers/anthropic.js

@@ -0,0 +1,188 @@
+import { errorEndOptions, instrumentAsyncIterable } from "./streaming.js";
+function extractModelParameters(body) {
+    if (!body)
+        return undefined;
+    const entries = ["temperature", "top_p", "top_k", "max_tokens"]
+        .filter((key) => body[key] !== undefined)
+        .map((key) => [key, body[key]]);
+    return entries.length > 0 ? Object.fromEntries(entries) : undefined;
+}
+function usageDetailsFrom(usage) {
+    if (!usage)
+        return undefined;
+    const details = {
+        ...(usage.input_tokens !== undefined && { input_tokens: usage.input_tokens }),
+        ...(usage.output_tokens !== undefined && { output_tokens: usage.output_tokens })
+    };
+    return Object.keys(details).length > 0 ? details : undefined;
+}
+function createEventAccumulator() {
+    const blocks = [];
+    let responseModel;
+    let inputTokens;
+    let outputTokens;
+    let stopReason;
+    return {
+        onEvent(raw) {
+            const event = raw;
+            switch (event.type) {
+                case "message_start":
+                    responseModel = event.message?.model;
+                    inputTokens = event.message?.usage?.input_tokens;
+                    outputTokens = event.message?.usage?.output_tokens;
+                    break;
+                case "content_block_start": {
+                    const block = event.content_block;
+                    if (event.index === undefined || !block)
+                        break;
+                    if (block.type === "text") {
+                        blocks[event.index] = { type: "text", text: block.text ?? "" };
+                    }
+                    else if (block.type === "tool_use") {
+                        blocks[event.index] = {
+                            type: "tool_use",
+                            ...(block.id && { id: block.id }),
+                            ...(block.name && { name: block.name }),
+                            partialJson: ""
+                        };
+                    }
+                    else if (block.type === "thinking") {
+                        // Extended-thinking blocks stream as thinking_delta text plus a
+                        // final signature_delta (the signature is required to round-trip
+                        // the block in later turns — dropping it would make the recorded
+                        // output unusable as a replay input). Review finding, PR #39.
+                        blocks[event.index] = {
+                            type: "thinking",
+                            thinking: block.thinking ?? "",
+                            signature: block.signature ?? ""
+                        };
+                    }
+                    else if (block.type) {
+                        // Other block kinds (redacted_thinking arrives complete at
+                        // start; future API additions) are kept as-is rather than
+                        // dropped — deltas for them aren't understood, but the block's
+                        // existence is real data.
+                        blocks[event.index] = { ...block, type: block.type };
+                    }
+                    break;
+                }
+                case "content_block_delta": {
+                    if (event.index === undefined)
+                        break;
+                    const block = blocks[event.index];
+                    if (!block)
+                        break;
+                    if (event.delta?.type === "text_delta" && block.type === "text") {
+                        block.text += event.delta.text ?? "";
+                    }
+                    else if (event.delta?.type === "input_json_delta" && block.type === "tool_use") {
+                        block.partialJson += event.delta.partial_json ?? "";
+                    }
+                    else if (event.delta?.type === "thinking_delta" && block.type === "thinking") {
+                        block.thinking += event.delta.thinking ?? "";
+                    }
+                    else if (event.delta?.type === "signature_delta" && block.type === "thinking") {
+                        block.signature = event.delta.signature ?? "";
+                    }
+                    break;
+                }
+                case "message_delta":
+                    if (event.usage?.output_tokens !== undefined)
+                        outputTokens = event.usage.output_tokens;
+                    if (event.usage?.input_tokens !== undefined)
+                        inputTokens = event.usage.input_tokens;
+                    if (event.delta?.stop_reason)
+                        stopReason = event.delta.stop_reason;
+                    break;
+            }
+        },
+        endOptions() {
+            const content = blocks.filter(Boolean).map((block) => {
+                if (block.type === "tool_use" && "partialJson" in block) {
+                    const { partialJson, ...rest } = block;
+                    const rawJson = typeof partialJson === "string" ? partialJson : "";
+                    let input;
+                    try {
+                        input = rawJson ? JSON.parse(rawJson) : {};
+                    }
+                    catch {
+                        // Truncated stream (early break mid-tool-call): keep the raw
+                        // fragment rather than losing it or throwing in a finalizer.
+                        input = { __partial_json: rawJson };
+                    }
+                    return { ...rest, input };
+                }
+                return block;
+            });
+            const usage = {
+                ...(inputTokens !== undefined && { input_tokens: inputTokens }),
+                ...(outputTokens !== undefined && { output_tokens: outputTokens })
+            };
+            return {
+                output: {
+                    role: "assistant",
+                    ...(responseModel && { model: responseModel }),
+                    content,
+                    ...(stopReason && { stop_reason: stopReason })
+                },
+                ...(Object.keys(usage).length > 0 && { usageDetails: usage }),
+                metadata: { streamed: "true" }
+            };
+        }
+    };
+}
+/**
+ * Wraps an Anthropic client instance so every messages.create() call —
+ * streaming or not — is automatically recorded as a generation. Mutates
+ * client.messages in place and returns the same client reference.
+ *
+ * Safe to call more than once on the same client — re-wrapping is detected
+ * and is a no-op, rather than nesting a second wrapper around the first
+ * (which would silently double-record every call).
+ */
+export function wrapAnthropic(client, ironside, options = {}) {
+    const messages = client.messages;
+    if (messages.__ironsideWrapped)
+        return client;
+    const originalCreate = messages.create.bind(messages);
+    const wrappedCreate = async (...args) => {
+        const requestBody = args[0];
+        const modelParameters = extractModelParameters(requestBody);
+        const trace = options.trace ?? ironside.trace({ name: "anthropic.messages.create" });
+        const generation = trace.generation({
+            name: "anthropic.messages.create",
+            ...(requestBody?.model && { model: requestBody.model }),
+            ...(modelParameters && { modelParameters }),
+            input: requestBody?.messages
+        });
+        try {
+            const result = await originalCreate(...args);
+            if (requestBody?.stream) {
+                const accumulator = createEventAccumulator();
+                return instrumentAsyncIterable(result, accumulator.onEvent, ({ error, consumed }) => {
+                    if (error)
+                        generation.end(errorEndOptions(error));
+                    else if (consumed)
+                        generation.end(accumulator.endOptions());
+                    else
+                        generation.end({ metadata: { streamed: "true" } });
+                });
+            }
+            const message = result;
+            const usageDetails = usageDetailsFrom(message.usage);
+            generation.end({
+                output: message,
+                ...(usageDetails && { usageDetails })
+            });
+            return result;
+        }
+        catch (error) {
+            generation.end(errorEndOptions(error));
+            throw error;
+        }
+    };
+    messages.create = wrappedCreate;
+    Object.defineProperty(messages, "__ironsideWrapped", { value: true, enumerable: false });
+    return client;
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/src/wrappers/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/wrappers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAqC1E,SAAS,sBAAsB,CAC7B,IAAwC;IAExC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,OAAO,GAAI,CAAC,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,YAAY,CAAW;SACvE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;SACxC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAqB,CAAC,CAAC;IACtD,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACtE,CAAC;AAsBD,SAAS,gBAAgB,CAAC,KAA+B;IACvD,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,OAAO,GAAG;QACd,GAAG,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,IAAI,EAAE,YAAY,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC;QAC7E,GAAG,CAAC,KAAK,CAAC,aAAa,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC;KACjF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/D,CAAC;AAiCD,SAAS,sBAAsB;IAC7B,MAAM,MAAM,GAAuB,EAAE,CAAC;IACtC,IAAI,aAAiC,CAAC;IACtC,IAAI,WAA+B,CAAC;IACpC,IAAI,YAAgC,CAAC;IACrC,IAAI,UAA8B,CAAC;IAEnC,OAAO;QACL,OAAO,CAAC,GAAY;YAClB,MAAM,KAAK,GAAG,GAAsB,CAAC;YACrC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,eAAe;oBAClB,aAAa,GAAG,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;oBACrC,WAAW,GAAG,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,YAAY,CAAC;oBACjD,YAAY,GAAG,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,aAAa,CAAC;oBACnD,MAAM;gBACR,KAAK,qBAAqB,CAAC,CAAC,CAAC;oBAC3B,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC;oBAClC,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK;wBAAE,MAAM;oBAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC1B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;oBACjE,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACrC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG;4BACpB,IAAI,EAAE,UAAU;4BAChB,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;4BACjC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;4BACvC,WAAW,EAAE,EAAE;yBAChB,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBACrC,gEAAgE;wBAChE,iEAAiE;wBACjE,iEAAiE;wBACjE,8DAA8D;wBAC9D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG;4BACpB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,KAAK,CAAC,QAAQ,IAAI,EAAE;4BAC9B,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,EAAE;yBACjC,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBACtB,2DAA2D;wBAC3D,0DAA0D;wBAC1D,+DAA+D;wBAC/D,0BAA0B;wBAC1B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;oBACvD,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;oBAC3B,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;wBAAE,MAAM;oBACrC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAClC,IAAI,CAAC,KAAK;wBAAE,MAAM;oBAClB,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBAC/D,KAA0B,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;oBAC7D,CAAC;yBAAM,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,kBAAkB,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAChF,KAAiC,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,IAAI,EAAE,CAAC;oBACnF,CAAC;yBAAM,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC9E,KAA8B,CAAC,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;oBACzE,CAAC;yBAAM,IAAI,KAAK,CAAC,KAAK,EAAE,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC/E,KAA+B,CAAC,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC;oBAC3E,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,eAAe;oBAClB,IAAI,KAAK,CAAC,KAAK,EAAE,aAAa,KAAK,SAAS;wBAAE,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,CAAC;oBACvF,IAAI,KAAK,CAAC,KAAK,EAAE,YAAY,KAAK,SAAS;wBAAE,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;oBACpF,IAAI,KAAK,CAAC,KAAK,EAAE,WAAW;wBAAE,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;oBACnE,MAAM;YACV,CAAC;QACH,CAAC;QACD,UAAU;YACR,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACnD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,aAAa,IAAI,KAAK,EAAE,CAAC;oBACxD,MAAM,EAAE,WAAW,EAAE,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC;oBACvC,MAAM,OAAO,GAAG,OAAO,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;oBACnE,IAAI,KAAc,CAAC;oBACnB,IAAI,CAAC;wBACH,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC7C,CAAC;oBAAC,MAAM,CAAC;wBACP,6DAA6D;wBAC7D,6DAA6D;wBAC7D,KAAK,GAAG,EAAE,cAAc,EAAE,OAAO,EAAE,CAAC;oBACtC,CAAC;oBACD,OAAO,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;gBAC5B,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YACH,MAAM,KAAK,GAAG;gBACZ,GAAG,CAAC,WAAW,KAAK,SAAS,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;gBAC/D,GAAG,CAAC,YAAY,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC;aACnE,CAAC;YACF,OAAO;gBACL,MAAM,EAAE;oBACN,IAAI,EAAE,WAAW;oBACjB,GAAG,CAAC,aAAa,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;oBAC9C,OAAO;oBACP,GAAG,CAAC,UAAU,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;iBAC/C;gBACD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;gBAC7D,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE;aAC/B,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,aAAa,CAC3B,MAAS,EACT,QAAwB,EACxB,UAAgC,EAAE;IAElC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAA0D,CAAC;IACnF,IAAI,QAAQ,CAAC,iBAAiB;QAAE,OAAO,MAAM,CAAC;IAE9C,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAE/B,CAAC;IAEtB,MAAM,aAAa,GAAG,KAAK,EAAE,GAAG,IAAe,EAAoB,EAAE;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,CAAC,CAEb,CAAC;QAEd,MAAM,eAAe,GAAG,sBAAsB,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,2BAA2B,EAAE,CAAC,CAAC;QACrF,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YAClC,IAAI,EAAE,2BAA2B;YACjC,GAAG,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,EAAE,CAAC;YACvD,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;YAC3C,KAAK,EAAE,WAAW,EAAE,QAAQ;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YAE7C,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;gBACxB,MAAM,WAAW,GAAG,sBAAsB,EAAE,CAAC;gBAC7C,OAAO,uBAAuB,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAClF,IAAI,KAAK;wBAAE,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;yBAC7C,IAAI,QAAQ;wBAAE,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;;wBACvD,UAAU,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBAC1D,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,OAAO,GAAG,MAAqB,CAAC;YACtC,MAAM,YAAY,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACrD,UAAU,CAAC,GAAG,CAAC;gBACb,MAAM,EAAE,OAAO;gBACf,GAAG,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,CAAC;aACtC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;YACvC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IACF,QAAQ,CAAC,MAAM,GAAG,aAAuC,CAAC;IAC1D,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;IAEzF,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/wrappers/openai.d.ts

@@ -0,0 +1,26 @@
+import type { IronsideClient, TraceHandle } from "../client.js";
+export interface WrapOpenAIOptions {
+    /** Attach generations to an existing trace instead of creating a new standalone trace per call. */
+    trace?: TraceHandle;
+}
+interface ChatCompletionsLike {
+    create: (...args: never[]) => unknown;
+}
+interface OpenAILike {
+    chat: {
+        completions: ChatCompletionsLike;
+    };
+}
+/**
+ * Wraps an OpenAI client instance so every chat.completions.create() call
+ * — streaming or not — is automatically recorded as a generation. Mutates
+ * client.chat.completions in place and returns the same client reference
+ * for convenient chaining (`const client = wrapOpenAI(new OpenAI(...), ironside)`).
+ *
+ * Safe to call more than once on the same client — re-wrapping is detected
+ * and is a no-op, rather than nesting a second wrapper around the first
+ * (which would silently double-record every call: two traces, two
+ * generations, per real API call).
+ */
+export declare function wrapOpenAI<T extends OpenAILike>(client: T, ironside: IronsideClient, options?: WrapOpenAIOptions): T;
+export {};

package/dist/src/wrappers/openai.js

@@ -0,0 +1,162 @@
+import { errorEndOptions, instrumentAsyncIterable } from "./streaming.js";
+function extractModelParameters(body) {
+    if (!body)
+        return undefined;
+    const entries = ["temperature", "top_p", "max_tokens", "max_completion_tokens", "presence_penalty", "frequency_penalty", "seed"]
+        .filter((key) => body[key] !== undefined)
+        .map((key) => [key, body[key]]);
+    return entries.length > 0 ? Object.fromEntries(entries) : undefined;
+}
+function usageDetailsFrom(usage) {
+    if (!usage)
+        return undefined;
+    const details = {
+        ...(usage.prompt_tokens !== undefined && { input_tokens: usage.prompt_tokens }),
+        ...(usage.completion_tokens !== undefined && { output_tokens: usage.completion_tokens })
+    };
+    return Object.keys(details).length > 0 ? details : undefined;
+}
+function assembleMessage(choice) {
+    const assembledToolCalls = choice.toolCalls.filter(Boolean);
+    return {
+        role: "assistant",
+        content: choice.sawContent ? choice.content : null,
+        ...(assembledToolCalls.length > 0 && { tool_calls: assembledToolCalls })
+    };
+}
+function createChunkAccumulator() {
+    // Keyed by choice.index — an n>1 request streams every choice's deltas
+    // interleaved in the same chunk sequence, so reading only choices[0]
+    // would silently truncate the response to a fraction of one choice
+    // (review finding, PR #39).
+    const choices = new Map();
+    let usage;
+    let responseModel;
+    return {
+        onChunk(raw) {
+            const chunk = raw;
+            if (chunk.model)
+                responseModel = chunk.model;
+            if (chunk.usage)
+                usage = chunk.usage;
+            for (const choice of chunk.choices ?? []) {
+                const index = choice.index ?? 0;
+                let slot = choices.get(index);
+                if (!slot) {
+                    slot = { content: "", sawContent: false, toolCalls: [] };
+                    choices.set(index, slot);
+                }
+                if (choice.finish_reason)
+                    slot.finishReason = choice.finish_reason;
+                if (typeof choice.delta?.content === "string") {
+                    slot.content += choice.delta.content;
+                    slot.sawContent = true;
+                }
+                for (const tc of choice.delta?.tool_calls ?? []) {
+                    const toolSlot = (slot.toolCalls[tc.index] ??= { function: { arguments: "" } });
+                    if (tc.id)
+                        toolSlot.id = tc.id;
+                    if (tc.type)
+                        toolSlot.type = tc.type;
+                    if (tc.function?.name)
+                        toolSlot.function.name = tc.function.name;
+                    if (tc.function?.arguments)
+                        toolSlot.function.arguments += tc.function.arguments;
+                }
+            }
+        },
+        endOptions() {
+            const sorted = [...choices.entries()].sort(([a], [b]) => a - b);
+            const usageDetails = usageDetailsFrom(usage);
+            // Single choice (the overwhelmingly common case): output is the
+            // assistant message itself. n>1: output mirrors the non-streaming
+            // completion's choices array so no choice is dropped.
+            const single = sorted.length === 0
+                ? { content: "", sawContent: false, toolCalls: [] } // empty stream: a bare content-null message
+                : sorted.length === 1
+                    ? sorted[0][1]
+                    : undefined;
+            const output = single
+                ? assembleMessage(single)
+                : {
+                    choices: sorted.map(([index, choice]) => ({
+                        index,
+                        message: assembleMessage(choice),
+                        ...(choice.finishReason && { finish_reason: choice.finishReason })
+                    }))
+                };
+            const finishReason = single?.finishReason;
+            return {
+                output,
+                ...(usageDetails && { usageDetails }),
+                metadata: {
+                    streamed: "true",
+                    ...(responseModel && { response_model: responseModel }),
+                    ...(finishReason && { finish_reason: finishReason })
+                }
+            };
+        }
+    };
+}
+/**
+ * Wraps an OpenAI client instance so every chat.completions.create() call
+ * — streaming or not — is automatically recorded as a generation. Mutates
+ * client.chat.completions in place and returns the same client reference
+ * for convenient chaining (`const client = wrapOpenAI(new OpenAI(...), ironside)`).
+ *
+ * Safe to call more than once on the same client — re-wrapping is detected
+ * and is a no-op, rather than nesting a second wrapper around the first
+ * (which would silently double-record every call: two traces, two
+ * generations, per real API call).
+ */
+export function wrapOpenAI(client, ironside, options = {}) {
+    const completions = client.chat.completions;
+    if (completions.__ironsideWrapped)
+        return client;
+    const originalCreate = completions.create.bind(completions);
+    const wrappedCreate = async (...args) => {
+        const requestBody = args[0];
+        const modelParameters = extractModelParameters(requestBody);
+        const trace = options.trace ?? ironside.trace({ name: "openai.chat.completions.create" });
+        const generation = trace.generation({
+            name: "openai.chat.completions.create",
+            ...(requestBody?.model && { model: requestBody.model }),
+            ...(modelParameters && { modelParameters }),
+            input: requestBody?.messages
+        });
+        try {
+            const result = await originalCreate(...args);
+            if (requestBody?.stream) {
+                const accumulator = createChunkAccumulator();
+                return instrumentAsyncIterable(result, accumulator.onChunk, ({ error, consumed }) => {
+                    if (error)
+                        generation.end(errorEndOptions(error));
+                    else if (consumed)
+                        generation.end(accumulator.endOptions());
+                    // !consumed: the result wasn't iterable at all (unexpected SDK
+                    // shape) — end with what we know rather than dangle forever.
+                    else
+                        generation.end({ metadata: { streamed: "true" } });
+                });
+            }
+            const completion = result;
+            const usageDetails = usageDetailsFrom(completion.usage);
+            generation.end({
+                output: completion,
+                ...(usageDetails && { usageDetails })
+            });
+            return result;
+        }
+        catch (error) {
+            generation.end(errorEndOptions(error));
+            throw error;
+        }
+    };
+    completions.create = wrappedCreate;
+    Object.defineProperty(completions, "__ironsideWrapped", {
+        value: true,
+        enumerable: false
+    });
+    return client;
+}
+//# sourceMappingURL=openai.js.map

package/dist/src/wrappers/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/wrappers/openai.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,uBAAuB,EAAE,MAAM,gBAAgB,CAAC;AAoD1E,SAAS,sBAAsB,CAC7B,IAAwC;IAExC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,OAAO,GACX,CAAC,aAAa,EAAE,OAAO,EAAE,YAAY,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,CAChH;SACE,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,SAAS,CAAC;SACxC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAqB,CAAC,CAAC;IACtD,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACtE,CAAC;AAYD,SAAS,gBAAgB,CAAC,KAAsC;IAC9D,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAC7B,MAAM,OAAO,GAAG;QACd,GAAG,CAAC,KAAK,CAAC,aAAa,KAAK,SAAS,IAAI,EAAE,YAAY,EAAE,KAAK,CAAC,aAAa,EAAE,CAAC;QAC/E,GAAG,CAAC,KAAK,CAAC,iBAAiB,KAAK,SAAS,IAAI,EAAE,aAAa,EAAE,KAAK,CAAC,iBAAiB,EAAE,CAAC;KACzF,CAAC;IACF,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AAC/D,CAAC;AAoCD,SAAS,eAAe,CAAC,MAAyB;IAChD,MAAM,kBAAkB,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC5D,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;QAClD,GAAG,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,kBAAkB,EAAE,CAAC;KACzE,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB;IAC7B,uEAAuE;IACvE,qEAAqE;IACrE,mEAAmE;IACnE,4BAA4B;IAC5B,MAAM,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IACrD,IAAI,KAAsC,CAAC;IAC3C,IAAI,aAAiC,CAAC;IAEtC,OAAO;QACL,OAAO,CAAC,GAAY;YAClB,MAAM,KAAK,GAAG,GAA8B,CAAC;YAC7C,IAAI,KAAK,CAAC,KAAK;gBAAE,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC;YAC7C,IAAI,KAAK,CAAC,KAAK;gBAAE,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YACrC,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC;gBAChC,IAAI,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC9B,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,IAAI,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;oBACzD,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAC3B,CAAC;gBACD,IAAI,MAAM,CAAC,aAAa;oBAAE,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;gBACnE,IAAI,OAAO,MAAM,CAAC,KAAK,EAAE,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAC9C,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;oBACrC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;gBACzB,CAAC;gBACD,KAAK,MAAM,EAAE,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,IAAI,EAAE,EAAE,CAAC;oBAChD,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;oBAChF,IAAI,EAAE,CAAC,EAAE;wBAAE,QAAQ,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;oBAC/B,IAAI,EAAE,CAAC,IAAI;wBAAE,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC;oBACrC,IAAI,EAAE,CAAC,QAAQ,EAAE,IAAI;wBAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACjE,IAAI,EAAE,CAAC,QAAQ,EAAE,SAAS;wBAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;gBACnF,CAAC;YACH,CAAC;QACH,CAAC;QACD,UAAU;YACR,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAChE,MAAM,YAAY,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC7C,gEAAgE;YAChE,kEAAkE;YAClE,sDAAsD;YACtD,MAAM,MAAM,GACV,MAAM,CAAC,MAAM,KAAK,CAAC;gBACjB,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,4CAA4C;gBAChG,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;oBACnB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC;oBACf,CAAC,CAAC,SAAS,CAAC;YAClB,MAAM,MAAM,GAAG,MAAM;gBACnB,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC;gBACzB,CAAC,CAAC;oBACE,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;wBACxC,KAAK;wBACL,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC;wBAChC,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,aAAa,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC;qBACnE,CAAC,CAAC;iBACJ,CAAC;YACN,MAAM,YAAY,GAAG,MAAM,EAAE,YAAY,CAAC;YAC1C,OAAO;gBACL,MAAM;gBACN,GAAG,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,CAAC;gBACrC,QAAQ,EAAE;oBACR,QAAQ,EAAE,MAAM;oBAChB,GAAG,CAAC,aAAa,IAAI,EAAE,cAAc,EAAE,aAAa,EAAE,CAAC;oBACvD,GAAG,CAAC,YAAY,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC;iBACrD;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAeD;;;;;;;;;;GAUG;AACH,MAAM,UAAU,UAAU,CACxB,MAAS,EACT,QAAwB,EACxB,UAA6B,EAAE;IAE/B,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,WAE/B,CAAC;IACF,IAAI,WAAW,CAAC,iBAAiB;QAAE,OAAO,MAAM,CAAC;IAEjD,MAAM,cAAc,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAErC,CAAC;IAEtB,MAAM,aAAa,GAAG,KAAK,EAAE,GAAG,IAAe,EAAoB,EAAE;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,CAAC,CAEb,CAAC;QAEd,MAAM,eAAe,GAAG,sBAAsB,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,gCAAgC,EAAE,CAAC,CAAC;QAC1F,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YAClC,IAAI,EAAE,gCAAgC;YACtC,GAAG,CAAC,WAAW,EAAE,KAAK,IAAI,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,EAAE,CAAC;YACvD,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;YAC3C,KAAK,EAAE,WAAW,EAAE,QAAQ;SAC7B,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YAE7C,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;gBACxB,MAAM,WAAW,GAAG,sBAAsB,EAAE,CAAC;gBAC7C,OAAO,uBAAuB,CAAC,MAAM,EAAE,WAAW,CAAC,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;oBAClF,IAAI,KAAK;wBAAE,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;yBAC7C,IAAI,QAAQ;wBAAE,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;oBAC5D,+DAA+D;oBAC/D,6DAA6D;;wBACxD,UAAU,CAAC,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;gBAC1D,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,MAA4B,CAAC;YAChD,MAAM,YAAY,GAAG,gBAAgB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACxD,UAAU,CAAC,GAAG,CAAC;gBACb,MAAM,EAAE,UAAU;gBAClB,GAAG,CAAC,YAAY,IAAI,EAAE,YAAY,EAAE,CAAC;aACtC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,UAAU,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC;YACvC,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IACF,WAAW,CAAC,MAAM,GAAG,aAA0C,CAAC;IAChE,MAAM,CAAC,cAAc,CAAC,WAAW,EAAE,mBAAmB,EAAE;QACtD,KAAK,EAAE,IAAI;QACX,UAAU,EAAE,KAAK;KAClB,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/src/wrappers/streaming.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Patches `stream`'s async iterator in place so every yielded chunk feeds
+ * `onChunk` and exactly one of done/break/error triggers `onFinish`.
+ * Returns the same object. If the value isn't async-iterable at all
+ * (unexpected SDK shape), it is returned untouched and `onFinish` fires
+ * immediately with `{ consumed: false }` so the generation isn't left
+ * dangling by our own bug.
+ */
+export declare function instrumentAsyncIterable<T>(stream: T, onChunk: (chunk: unknown) => void, onFinish: (outcome: {
+    error?: unknown;
+    consumed: boolean;
+}) => void): T;
+/** Shared error→end mapping so streamed and non-streamed failures record identically. */
+export declare function errorEndOptions(error: unknown): {
+    level: "error";
+    statusMessage: string;
+};

package/dist/src/wrappers/streaming.js

@@ -0,0 +1,86 @@
+// Shared machinery for tracing streamed responses (M9-07). Both provider
+// SDKs return a Stream object that is an async iterable with extra API
+// surface (.tee(), .controller, .toReadableStream(), ...). Returning our
+// own wrapper generator would silently break every caller using that
+// surface, so instead the stream's [Symbol.asyncIterator] is patched IN
+// PLACE and the SAME object is returned — identical to how the wrappers
+// patch client.chat.completions.create itself.
+//
+// The generation can only be finalized when the caller actually consumes
+// the stream (that's when the text/usage exists at all). Three exits all
+// funnel into one finalize call: normal completion (done), early break
+// (the iterator's return()), and a mid-stream error (next() rejecting or
+// throw()). A stream the caller never iterates records its start event
+// but never ends — visible in the UI as a dangling in-progress
+// generation, which is the honest representation of what happened.
+//
+// KNOWN LIMIT — .tee(): each branch iterates through the patched
+// asyncIterator, so chunks from both branches feed one accumulator
+// (double-counted text). finalize still fires exactly once. tee'd +
+// wrapped is rare enough that correct-single-stream beats a per-iterator
+// accumulator design that couldn't merge usage sanely anyway.
+/**
+ * Patches `stream`'s async iterator in place so every yielded chunk feeds
+ * `onChunk` and exactly one of done/break/error triggers `onFinish`.
+ * Returns the same object. If the value isn't async-iterable at all
+ * (unexpected SDK shape), it is returned untouched and `onFinish` fires
+ * immediately with `{ consumed: false }` so the generation isn't left
+ * dangling by our own bug.
+ */
+export function instrumentAsyncIterable(stream, onChunk, onFinish) {
+    const iterable = stream;
+    if (typeof iterable?.[Symbol.asyncIterator] !== "function") {
+        onFinish({ consumed: false });
+        return stream;
+    }
+    let finished = false;
+    const finishOnce = (outcome) => {
+        if (finished)
+            return;
+        finished = true;
+        onFinish({ ...outcome, consumed: true });
+    };
+    const originalFactory = iterable[Symbol.asyncIterator].bind(iterable);
+    iterable[Symbol.asyncIterator] = () => {
+        const inner = originalFactory();
+        return {
+            async next() {
+                try {
+                    const result = await inner.next();
+                    if (result.done)
+                        finishOnce({});
+                    else
+                        onChunk(result.value);
+                    return result;
+                }
+                catch (error) {
+                    finishOnce({ error });
+                    throw error;
+                }
+            },
+            // Called on `break`/`return` inside a for-await — the caller chose
+            // to stop early; what accumulated so far is the real output.
+            async return(value) {
+                finishOnce({});
+                if (inner.return)
+                    return inner.return(value);
+                return { done: true, value: undefined };
+            },
+            async throw(error) {
+                finishOnce({ error });
+                if (inner.throw)
+                    return inner.throw(error);
+                throw error;
+            }
+        };
+    };
+    return stream;
+}
+/** Shared error→end mapping so streamed and non-streamed failures record identically. */
+export function errorEndOptions(error) {
+    return {
+        level: "error",
+        statusMessage: error instanceof Error ? error.message : String(error)
+    };
+}
+//# sourceMappingURL=streaming.js.map

package/dist/src/wrappers/streaming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.js","sourceRoot":"","sources":["../../../src/wrappers/streaming.ts"],"names":[],"mappings":"AAAA,yEAAyE;AACzE,uEAAuE;AACvE,yEAAyE;AACzE,qEAAqE;AACrE,wEAAwE;AACxE,wEAAwE;AACxE,+CAA+C;AAC/C,EAAE;AACF,yEAAyE;AACzE,yEAAyE;AACzE,uEAAuE;AACvE,yEAAyE;AACzE,uEAAuE;AACvE,+DAA+D;AAC/D,mEAAmE;AACnE,EAAE;AACF,iEAAiE;AACjE,mEAAmE;AACnE,oEAAoE;AACpE,yEAAyE;AACzE,8DAA8D;AAE9D;;;;;;;GAOG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAS,EACT,OAAiC,EACjC,QAAmE;IAEnE,MAAM,QAAQ,GAAG,MAEhB,CAAC;IACF,IAAI,OAAO,QAAQ,EAAE,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,EAAE,CAAC;QAC3D,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9B,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,MAAM,UAAU,GAAG,CAAC,OAA4B,EAAE,EAAE;QAClD,IAAI,QAAQ;YAAE,OAAO;QACrB,QAAQ,GAAG,IAAI,CAAC;QAChB,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC;IAEF,MAAM,eAAe,GAAG,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvE,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,EAAE;QACpC,MAAM,KAAK,GAAG,eAAe,EAAE,CAAC;QAChC,OAAO;YACL,KAAK,CAAC,IAAI;gBACR,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;oBAClC,IAAI,MAAM,CAAC,IAAI;wBAAE,UAAU,CAAC,EAAE,CAAC,CAAC;;wBAC3B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC3B,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;oBACtB,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC;YACD,mEAAmE;YACnE,6DAA6D;YAC7D,KAAK,CAAC,MAAM,CAAC,KAAe;gBAC1B,UAAU,CAAC,EAAE,CAAC,CAAC;gBACf,IAAI,KAAK,CAAC,MAAM;oBAAE,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC7C,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YAC1C,CAAC;YACD,KAAK,CAAC,KAAK,CAAC,KAAe;gBACzB,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtB,IAAI,KAAK,CAAC,KAAK;oBAAE,OAAO,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC3C,MAAM,KAAK,CAAC;YACd,CAAC;SACF,CAAC;IACJ,CAAC,CAAC;IACF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,yFAAyF;AACzF,MAAM,UAAU,eAAe,CAAC,KAAc;IAI5C,OAAO;QACL,KAAK,EAAE,OAAO;QACd,aAAa,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;KACtE,CAAC;AACJ,CAAC"}

package/dist/src/wrappers/vercel-ai.d.ts

@@ -0,0 +1,29 @@
+import type { IronsideClient, TraceHandle } from "../client.js";
+export interface RecordGenerateTextOptions {
+    trace?: TraceHandle;
+    /** Passed through to the generation's name; defaults to "ai.generateText". */
+    name?: string;
+    /** The `model` id string you passed to generateText/streamText, since the result object doesn't always echo it back verbatim. */
+    model?: string;
+    /** The `prompt`/`messages` you passed in, for the recorded input. */
+    input?: unknown;
+    /** Sampling parameters (temperature, maxOutputTokens, ...) you passed to generateText/streamText — this recorder doesn't intercept the call, so these aren't available on the result and must be passed through explicitly. */
+    modelParameters?: Record<string, string | number | boolean | null>;
+}
+interface LanguageModelUsageLike {
+    inputTokens?: number;
+    outputTokens?: number;
+}
+interface GenerateTextResultLike {
+    text?: string;
+    usage?: LanguageModelUsageLike;
+}
+/**
+ * Records a completed generateText()/streamText() result as a generation.
+ * Call after awaiting the result (or after a stream finishes and its
+ * `usage` promise/property resolves) — this does not intercept the call
+ * itself, since the AI SDK's own `telemetry` option is the recommended
+ * hook for that.
+ */
+export declare function recordGenerateTextResult(ironside: IronsideClient, result: GenerateTextResultLike, options?: RecordGenerateTextOptions): void;
+export {};

package/dist/src/wrappers/vercel-ai.js

@@ -0,0 +1,30 @@
+/**
+ * Records a completed generateText()/streamText() result as a generation.
+ * Call after awaiting the result (or after a stream finishes and its
+ * `usage` promise/property resolves) — this does not intercept the call
+ * itself, since the AI SDK's own `telemetry` option is the recommended
+ * hook for that.
+ */
+export function recordGenerateTextResult(ironside, result, options = {}) {
+    const trace = options.trace ?? ironside.trace({ name: options.name ?? "ai.generateText" });
+    const generation = trace.generation({
+        name: options.name ?? "ai.generateText",
+        ...(options.model && { model: options.model }),
+        ...(options.modelParameters && { modelParameters: options.modelParameters }),
+        input: options.input
+    });
+    generation.end({
+        output: result.text,
+        ...(result.usage && {
+            usageDetails: {
+                ...(result.usage.inputTokens !== undefined && {
+                    input_tokens: result.usage.inputTokens
+                }),
+                ...(result.usage.outputTokens !== undefined && {
+                    output_tokens: result.usage.outputTokens
+                })
+            }
+        })
+    });
+}
+//# sourceMappingURL=vercel-ai.js.map

package/dist/src/wrappers/vercel-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../../../src/wrappers/vercel-ai.ts"],"names":[],"mappings":"AAuCA;;;;;;GAMG;AACH,MAAM,UAAU,wBAAwB,CACtC,QAAwB,EACxB,MAA8B,EAC9B,UAAqC,EAAE;IAEvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,QAAQ,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,iBAAiB,EAAE,CAAC,CAAC;IAC3F,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;QAClC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,iBAAiB;QACvC,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QAC9C,GAAG,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,CAAC;QAC5E,KAAK,EAAE,OAAO,CAAC,KAAK;KACrB,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,CAAC;QACb,MAAM,EAAE,MAAM,CAAC,IAAI;QACnB,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI;YAClB,YAAY,EAAE;gBACZ,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,IAAI;oBAC5C,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,WAAW;iBACvC,CAAC;gBACF,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,IAAI;oBAC7C,aAAa,EAAE,MAAM,CAAC,KAAK,CAAC,YAAY;iBACzC,CAAC;aACH;SACF,CAAC;KACH,CAAC,CAAC;AACL,CAAC"}

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "ironside",
+  "version": "0.1.0",
+  "description": "Ironside client SDK — instrument LLM traces in a few lines.",
+  "license": "SEE LICENSE IN LICENSE.md",
+  "keywords": [
+    "ai",
+    "llm",
+    "observability",
+    "opentelemetry",
+    "tracing"
+  ],
+  "type": "module",
+  "sideEffects": false,
+  "main": "./dist/src/index.js",
+  "types": "./dist/src/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/src/index.d.ts",
+      "import": "./dist/src/index.js"
+    }
+  },
+  "files": [
+    "dist/src",
+    "src",
+    "README.md",
+    "LICENSE.md"
+  ],
+  "engines": {
+    "node": ">=20"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/luka-zivkovic/ironside.git",
+    "directory": "packages/sdk"
+  },
+  "publishConfig": {
+    "access": "public",
+    "provenance": false,
+    "registry": "https://registry.npmjs.org/"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "prepack": "npm run build",
+    "test:package": "node ./scripts/smoke-package.mjs",
+    "typecheck": "tsc -p tsconfig.json --noEmit"
+  },
+  "dependencies": {
+    "ulid": "^3.0.2"
+  },
+  "devDependencies": {
+    "@anthropic-ai/sdk": "^0.111.0",
+    "openai": "^6.46.0"
+  }
+}