modaic 0.1.0

package/esm/modaic/signatures.js

@@ -0,0 +1,290 @@
+import { z } from "zod";
+/**
+ * A Signature defines the interface for a DSPy-style predictor.
+ * It includes optional instructions, an input schema, and an output schema.
+ * Both input and output are defined using Zod schemas.
+ *
+ * This is a port of the `@modaic/dsts` Signature with tool support removed —
+ * Arbiters are pure judges and never call tools.
+ */
+export class Signature {
+    instructions;
+    input;
+    output;
+    prefixes = {};
+    // `instructions?: string | undefined` (not just `?: string`) so callers may pass
+    // a `string | undefined` through under `exactOptionalPropertyTypes`.
+    constructor(sig) {
+        this.instructions =
+            sig.instructions ||
+                this.generateDefaultInstructions(sig.input, sig.output);
+        this.input = sig.input;
+        this.output = sig.output;
+        // Initialize default prefixes
+        const allFields = { ...sig.input.shape, ...sig.output.shape };
+        for (const name in allFields) {
+            this.prefixes[name] = infer_prefix(name) + ":";
+        }
+    }
+    generateDefaultInstructions(input, output) {
+        const inputFields = Object.keys(input.shape);
+        const outputFields = Object.keys(output.shape);
+        const inputStr = inputFields.map((f) => `\`${f}\``).join(", ");
+        const outputStr = outputFields.map((f) => `\`${f}\``).join(", ");
+        return `Given the fields ${inputStr}, produce the fields ${outputStr}.`;
+    }
+    get input_fields() {
+        return this.input.shape;
+    }
+    get output_fields() {
+        return this.output.shape;
+    }
+    /**
+     * Static helper to create a Signature from a string definition.
+     * e.g. Signature.parse("question -> answer", "Answer the question.")
+     */
+    static parse(sigStr, instructions) {
+        const parsed = parseStringSignature(sigStr);
+        return new Signature({
+            instructions,
+            input: parsed.input,
+            output: parsed.output,
+        });
+    }
+    /**
+     * Returns a new Signature with the specified field removed from either input or output.
+     */
+    delete(inputOrOutput, name) {
+        const inputShape = this.input.shape;
+        const outputShape = this.output.shape;
+        let newInput = this.input;
+        let newOutput = this.output;
+        // Remove from input if present and requested
+        if (name in inputShape && inputOrOutput === "input") {
+            newInput = this.input.omit({ [name]: true });
+        }
+        // Remove from output if present and requested
+        if (name in outputShape && inputOrOutput === "output") {
+            newOutput = this.output.omit({
+                [name]: true,
+            });
+        }
+        return new Signature({
+            instructions: this.instructions,
+            input: newInput,
+            output: newOutput,
+        });
+    }
+    /**
+     * Returns a new Signature with updated instructions.
+     */
+    withInstructions(instructions) {
+        return new Signature({
+            instructions,
+            input: this.input,
+            output: this.output,
+        });
+    }
+    dump_state() {
+        const fields = Object.entries({
+            ...this.input.shape,
+            ...this.output.shape,
+        }).map(([name, field]) => ({
+            prefix: this.prefixes[name],
+            description: field.meta?.()?.description ||
+                field.meta?.()?.desc ||
+                field._def?.description ||
+                "",
+        }));
+        return {
+            instructions: this.instructions,
+            fields,
+        };
+    }
+    load_state(state) {
+        const prefixToName = {};
+        for (const [name, prefix] of Object.entries(this.prefixes)) {
+            prefixToName[prefix] = name;
+        }
+        const inputPatch = {};
+        const outputPatch = {};
+        for (const fieldState of state.fields) {
+            const name = prefixToName[fieldState.prefix];
+            if (!name)
+                continue;
+            const isInput = name in this.input.shape;
+            const field = (isInput ? this.input.shape[name] : this.output.shape[name]);
+            let updatedField = field;
+            const description = fieldState.description;
+            updatedField = updatedField.describe(description);
+            if (typeof updatedField.meta === "function") {
+                const currentMeta = updatedField.meta() || {};
+                updatedField = updatedField.meta({
+                    ...currentMeta,
+                    description,
+                    desc: description,
+                });
+            }
+            if (isInput) {
+                inputPatch[name] = updatedField;
+            }
+            else {
+                outputPatch[name] = updatedField;
+            }
+        }
+        return new Signature({
+            instructions: state.instructions,
+            input: this.input.extend(inputPatch),
+            output: this.output.extend(outputPatch),
+        });
+    }
+}
+/**
+ * Maps common string type names to Zod schemas.
+ */
+const mapTypeToZod = (typeStr) => {
+    switch (typeStr?.toLowerCase()) {
+        case "number":
+            return z.number();
+        case "boolean":
+            return z.boolean();
+        case "string":
+        case undefined:
+        case "":
+            return z.string();
+        default:
+            // Default to string for unknown types, similar to DSPy behavior
+            return z.string();
+    }
+};
+/**
+ * Parses a string-based signature like "input1: string, input2: number -> output: string"
+ */
+const parseStringSignature = (sigStr) => {
+    const parts = sigStr.split("->");
+    if (parts.length !== 2) {
+        throw new Error(`Invalid signature string: "${sigStr}". Must contain exactly one "->" separator.`);
+    }
+    const [inputsPart, outputsPart] = parts.map((s) => s.trim());
+    const parsePart = (part) => {
+        const shape = {};
+        const fields = part
+            .split(",")
+            .map((f) => f.trim())
+            .filter(Boolean);
+        for (const field of fields) {
+            const [name, type] = field.split(":").map((s) => s.trim());
+            if (!name)
+                continue;
+            shape[name] = mapTypeToZod(type);
+        }
+        return z.object(shape);
+    };
+    return {
+        input: parsePart(inputsPart),
+        output: parsePart(outputsPart),
+    };
+};
+/**
+ * Infer a prefix from an attribute name by converting it to a human-readable format.
+ * Matches the logic in DSPy's infer_prefix.
+ *
+ * Examples:
+ *   "camelCaseText" -> "Camel Case Text"
+ *   "snake_case_text" -> "Snake Case Text"
+ *   "text2number" -> "Text 2 Number"
+ *   "HTMLParser" -> "HTML Parser"
+ */
+export function infer_prefix(attributeName) {
+    // Step 1: Convert camelCase to snake_case
+    // Example: "camelCase" -> "camel_Case"
+    const s1 = attributeName.replace(/(.)([A-Z][a-z]+)/g, "$1_$2");
+    // Handle consecutive capitals
+    // Example: "camel_Case" -> "camel_case"
+    const intermediateName = s1.replace(/([a-z0-9])([A-Z])/g, "$1_$2");
+    // Step 2: Handle numbers by adding underscores around them
+    // Example: "text2number" -> "text_2_number"
+    let withUnderscoresAroundNumbers = intermediateName.replace(/([a-zA-Z])(\d)/g, "$1_$2");
+    // Example: "2text" -> "2_text"
+    withUnderscoresAroundNumbers = withUnderscoresAroundNumbers.replace(/(\d)([a-zA-Z])/g, "$1_$2");
+    // Step 3: Convert to Title Case while preserving acronyms
+    const words = withUnderscoresAroundNumbers.split("_");
+    const titleCasedWords = words.map((word) => {
+        if (word === word.toUpperCase() && word !== word.toLowerCase()) {
+            // Preserve acronyms like 'HTML', 'API' as-is
+            return word;
+        }
+        // Capitalize first letter: 'text' -> 'Text'
+        return word.charAt(0).toUpperCase() + word.slice(1).toLowerCase();
+    });
+    // Join words with spaces
+    // Example: ["Text", "2", "Number"] -> "Text 2 Number"
+    return titleCasedWords.join(" ");
+}
+/**
+ * `modaic.Image` — a multimodal image field, the TS analog of `dspy.Image`.
+ *
+ * Use {@link Image.field} inside a `Signature`'s input/output schema, and construct
+ * `new Image({ url })` to carry a value when calling `predict`.
+ */
+export class Image {
+    url;
+    constructor(init) {
+        this.url = init.url;
+    }
+    /**
+     * The Zod schema to place in a `Signature`. Serializes to a `dspy.Image` `$ref`.
+     * Chain `.describe(...)` to set the field description.
+     */
+    static field() {
+        return z.custom().meta({ __modaic_type: "image" });
+    }
+}
+/**
+ * `modaic.Audio` — a multimodal audio field, the TS analog of `dspy.Audio`.
+ *
+ * Use {@link Audio.field} inside a `Signature`'s input/output schema, and construct
+ * `new Audio({ url })` to carry a value when calling `predict`.
+ */
+export class Audio {
+    url;
+    constructor(init) {
+        this.url = init.url;
+    }
+    /**
+     * The Zod schema to place in a `Signature`. Serializes to a `dspy.Audio` `$ref`.
+     * Chain `.describe(...)` to set the field description.
+     */
+    static field() {
+        return z.custom().meta({ __modaic_type: "audio" });
+    }
+}
+/**
+ * `modaic.Scale(lo, hi)` — an integer rating in `[lo, hi]` inclusive.
+ *
+ * Serializes to a plain integer enum `{"enum":[lo, lo+1, …, hi], "type":"integer"}`,
+ * matching Python `modaic.Scale[lo, hi]` (which presents to the model as a Literal of ints).
+ */
+export function Scale(lo, hi) {
+    if (!Number.isInteger(lo) || !Number.isInteger(hi)) {
+        throw new Error("Scale values must be integers");
+    }
+    if (lo > hi) {
+        throw new Error(`Scale lo (${lo}) must be <= hi (${hi})`);
+    }
+    return z
+        .int()
+        .gte(lo)
+        .lte(hi)
+        .meta({ __modaic_type: "scale", lo, hi });
+}
+/**
+ * `modaic.Enum(...values)` — a string choice from a fixed set.
+ *
+ * Serializes to `{"enum":[...], "type":"string"}` (a single value serializes as
+ * `{"const": v, "type":"string"}`), matching Python `modaic.Enum[...]`.
+ */
+export function Enum(...values) {
+    return z.enum(values);
+}
+//# sourceMappingURL=signatures.js.map

package/esm/modaic/signatures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"signatures.js","sourceRoot":"","sources":["../../src/modaic/signatures.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;;;;;;GAOG;AACH,MAAM,OAAO,SAAS;IAIb,YAAY,CAAU;IACtB,KAAK,CAAI;IACT,MAAM,CAAI;IACV,QAAQ,GAA2B,EAAE,CAAC;IAa7C,iFAAiF;IACjF,qEAAqE;IACrE,YAAY,GAA+D,EAAE;QAC3E,IAAI,CAAC,YAAY;YACf,GAAG,CAAC,YAAY;gBAChB,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;QACvB,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;QAEzB,8BAA8B;QAC9B,MAAM,SAAS,GAAG,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC9D,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC;QACjD,CAAC;IAAA,CACF;IAEO,2BAA2B,CAAC,KAAQ,EAAE,MAAS,EAAU;QAC/D,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,OAAO,oBAAoB,QAAQ,wBAAwB,SAAS,GAAG,CAAC;IAAA,CACzE;IAED,IAAI,YAAY,GAAwB;QACtC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;IAAA,CACzB;IAED,IAAI,aAAa,GAAwB;QACvC,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAAA,CAC1B;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CACV,MAAc,EACd,YAAqB,EAC0B;QAC/C,MAAM,MAAM,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAC5C,OAAO,IAAI,SAAS,CAAC;YACnB,YAAY;YACZ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE,MAAM,CAAC,MAAM;SACtB,CAAC,CAAC;IAAA,CACJ;IAED;;OAEG;IACH,MAAM,CACJ,aAAiC,EACjC,IAAY,EACmC;QAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;QACpC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAEtC,IAAI,QAAQ,GAAqB,IAAI,CAAC,KAAK,CAAC;QAC5C,IAAI,SAAS,GAAqB,IAAI,CAAC,MAAM,CAAC;QAE9C,6CAA6C;QAC7C,IAAI,IAAI,IAAI,UAAU,IAAI,aAAa,KAAK,OAAO,EAAE,CAAC;YACpD,QAAQ,GAAI,IAAI,CAAC,KAA0B,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,IAAI,EAAS,CAAC,CAAC;QAC5E,CAAC;QAED,8CAA8C;QAC9C,IAAI,IAAI,IAAI,WAAW,IAAI,aAAa,KAAK,QAAQ,EAAE,CAAC;YACtD,SAAS,GAAI,IAAI,CAAC,MAA2B,CAAC,IAAI,CAAC;gBACjD,CAAC,IAAI,CAAC,EAAE,IAAI;aACN,CAAC,CAAC;QACZ,CAAC;QAED,OAAO,IAAI,SAAS,CAAC;YACnB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC;IAAA,CACJ;IAED;;OAEG;IACH,gBAAgB,CAAC,YAAoB,EAAmB;QACtD,OAAO,IAAI,SAAS,CAAC;YACnB,YAAY;YACZ,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;IAAA,CACJ;IAED,UAAU,GAAQ;QAChB,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;YAC5B,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK;YACnB,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK;SACrB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YACzB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC3B,WAAW,EACR,KAAa,CAAC,IAAI,EAAE,EAAE,EAAE,WAAW;gBACnC,KAAa,CAAC,IAAI,EAAE,EAAE,EAAE,IAAI;gBAC5B,KAAa,CAAC,IAAI,EAAE,WAAW;gBAChC,EAAE;SACL,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,MAAM;SACP,CAAC;IAAA,CACH;IAED,UAAU,CAAC,KAAU,EAAiD;QACpE,MAAM,YAAY,GAA2B,EAAE,CAAC;QAChD,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3D,YAAY,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;QAC9B,CAAC;QAED,MAAM,UAAU,GAA8B,EAAE,CAAC;QACjD,MAAM,WAAW,GAA8B,EAAE,CAAC;QAElD,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC7C,IAAI,CAAC,IAAI;gBAAE,SAAS;YAEpB,MAAM,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;YACzC,MAAM,KAAK,GAAG,CACZ,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAC9C,CAAC;YAEf,IAAI,YAAY,GAAG,KAAK,CAAC;YACzB,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;YAE3C,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;YAElD,IAAI,OAAQ,YAAoB,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACrD,MAAM,WAAW,GAAI,YAAoB,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;gBACvD,YAAY,GAAI,YAAoB,CAAC,IAAI,CAAC;oBACxC,GAAG,WAAW;oBACd,WAAW;oBACX,IAAI,EAAE,WAAW;iBAClB,CAAC,CAAC;YACL,CAAC;YAED,IAAI,OAAO,EAAE,CAAC;gBACZ,UAAU,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;YACnC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,SAAS,CAAC;YACnB,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC;YACpC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC;SACxC,CAAC,CAAC;IAAA,CACJ;CACF;AAYD;;GAEG;AACH,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAa,EAAE,CAAC;IACpD,QAAQ,OAAO,EAAE,WAAW,EAAE,EAAE,CAAC;QAC/B,KAAK,QAAQ;YACX,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACpB,KAAK,SAAS;YACZ,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC;QACrB,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS,CAAC;QACf,KAAK,EAAE;YACL,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACpB;YACE,gEAAgE;YAChE,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;IACtB,CAAC;AAAA,CACF,CAAC;AAEF;;GAEG;AACH,MAAM,oBAAoB,GAAG,CAC3B,MAAc,EACyC,EAAE,CAAC;IAC1D,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,8BAA8B,MAAM,6CAA6C,CAClF,CAAC;IACJ,CAAC;IAED,MAAM,CAAC,UAAU,EAAE,WAAW,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAE7D,MAAM,SAAS,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC;QAClC,MAAM,KAAK,GAA8B,EAAE,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI;aAChB,KAAK,CAAC,GAAG,CAAC;aACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;aACpB,MAAM,CAAC,OAAO,CAAC,CAAC;QAEnB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,KAAK,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAAA,CACxB,CAAC;IAEF,OAAO;QACL,KAAK,EAAE,SAAS,CAAC,UAAW,CAAC;QAC7B,MAAM,EAAE,SAAS,CAAC,WAAY,CAAC;KAChC,CAAC;AAAA,CACH,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,UAAU,YAAY,CAAC,aAAqB,EAAU;IAC1D,0CAA0C;IAC1C,uCAAuC;IACvC,MAAM,EAAE,GAAG,aAAa,CAAC,OAAO,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;IAE/D,8BAA8B;IAC9B,wCAAwC;IACxC,MAAM,gBAAgB,GAAG,EAAE,CAAC,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;IAEnE,2DAA2D;IAC3D,4CAA4C;IAC5C,IAAI,4BAA4B,GAAG,gBAAgB,CAAC,OAAO,CACzD,iBAAiB,EACjB,OAAO,CACR,CAAC;IACF,+BAA+B;IAC/B,4BAA4B,GAAG,4BAA4B,CAAC,OAAO,CACjE,iBAAiB,EACjB,OAAO,CACR,CAAC;IAEF,0DAA0D;IAC1D,MAAM,KAAK,GAAG,4BAA4B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACtD,MAAM,eAAe,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QAC1C,IAAI,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,IAAI,IAAI,KAAK,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YAC/D,6CAA6C;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC;QACD,4CAA4C;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IAAA,CACnE,CAAC,CAAC;IAEH,yBAAyB;IACzB,sDAAsD;IACtD,OAAO,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAAA,CAClC;AA2BD;;;;;GAKG;AACH,MAAM,OAAO,KAAK;IAChB,GAAG,CAAS;IAEZ,YAAY,IAAqB,EAAE;QACjC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;IAAA,CACrB;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,GAAc;QACxB,OAAO,CAAC,CAAC,MAAM,EAAS,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;IAAA,CAC3D;CACF;AAED;;;;;GAKG;AACH,MAAM,OAAO,KAAK;IAChB,GAAG,CAAS;IAEZ,YAAY,IAAqB,EAAE;QACjC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;IAAA,CACrB;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,GAAc;QACxB,OAAO,CAAC,CAAC,MAAM,EAAS,CAAC,IAAI,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAC;IAAA,CAC3D;CACF;AAED;;;;;GAKG;AACH,MAAM,UAAU,KAAK,CAAC,EAAU,EAAE,EAAU,EAAa;IACvD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,aAAa,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;IAC5D,CAAC;IACD,OAAO,CAAC;SACL,GAAG,EAAE;SACL,GAAG,CAAC,EAAE,CAAC;SACP,GAAG,CAAC,EAAE,CAAC;SACP,IAAI,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;AAAA,CAC7C;AAED;;;;;GAKG;AACH,MAAM,UAAU,IAAI,CAAmB,GAAG,MAAmB,EAAa;IACxE,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAAA,CACvB"}

package/esm/models/annotate-example-request.d.ts

@@ -0,0 +1,16 @@
+import * as z from "zod/v4-mini";
+import { PredictionAnnotation, PredictionAnnotation$Outbound } from "./prediction-annotation.js";
+/**
+ * Annotate an example with one or more prediction annotations.
+ */
+export type AnnotateExampleRequest = {
+    annotations: Array<PredictionAnnotation>;
+};
+/** @internal */
+export type AnnotateExampleRequest$Outbound = {
+    annotations: Array<PredictionAnnotation$Outbound>;
+};
+/** @internal */
+export declare const AnnotateExampleRequest$outboundSchema: z.ZodMiniType<AnnotateExampleRequest$Outbound, AnnotateExampleRequest>;
+export declare function annotateExampleRequestToJSON(annotateExampleRequest: AnnotateExampleRequest): string;
+//# sourceMappingURL=annotate-example-request.d.ts.map

package/esm/models/annotate-example-request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"annotate-example-request.d.ts","sourceRoot":"","sources":["../../src/models/annotate-example-request.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EACL,oBAAoB,EACpB,6BAA6B,EAE9B,MAAM,4BAA4B,CAAC;AAEpC;;GAEG;AACH,MAAM,MAAM,sBAAsB,GAAG;IACnC,WAAW,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC;CAC1C,CAAC;AAEF,gBAAgB;AAChB,MAAM,MAAM,+BAA+B,GAAG;IAC5C,WAAW,EAAE,KAAK,CAAC,6BAA6B,CAAC,CAAC;CACnD,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,qCAAqC,EAAE,CAAC,CAAC,WAAW,CAC/D,+BAA+B,EAC/B,sBAAsB,CAGtB,CAAC;AAEH,wBAAgB,4BAA4B,CAC1C,sBAAsB,EAAE,sBAAsB,GAC7C,MAAM,CAIR"}

package/esm/models/annotate-example-request.js

@@ -0,0 +1,13 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { PredictionAnnotation$outboundSchema, } from "./prediction-annotation.js";
+/** @internal */
+export const AnnotateExampleRequest$outboundSchema = z.object({
+    annotations: z.array(PredictionAnnotation$outboundSchema),
+});
+export function annotateExampleRequestToJSON(annotateExampleRequest) {
+    return JSON.stringify(AnnotateExampleRequest$outboundSchema.parse(annotateExampleRequest));
+}
+//# sourceMappingURL=annotate-example-request.js.map

package/esm/models/annotate-example-request.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"annotate-example-request.js","sourceRoot":"","sources":["../../src/models/annotate-example-request.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAGL,mCAAmC,GACpC,MAAM,4BAA4B,CAAC;AAcpC,gBAAgB;AAChB,MAAM,CAAC,MAAM,qCAAqC,GAG9C,CAAC,CAAC,MAAM,CAAC;IACX,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,mCAAmC,CAAC;CAC1D,CAAC,CAAC;AAEH,MAAM,UAAU,4BAA4B,CAC1C,sBAA8C,EACtC;IACR,OAAO,IAAI,CAAC,SAAS,CACnB,qCAAqC,CAAC,KAAK,CAAC,sBAAsB,CAAC,CACpE,CAAC;AAAA,CACH"}

package/esm/models/arbiter-request.d.ts

@@ -0,0 +1,18 @@
+import * as z from "zod/v4-mini";
+export type ArbiterRequest = {
+    arbiterRepo: string;
+    arbiterRevision?: string | undefined;
+    groundTruth?: string | null | undefined;
+    groundReasoning?: string | undefined;
+};
+/** @internal */
+export type ArbiterRequest$Outbound = {
+    arbiter_repo: string;
+    arbiter_revision: string;
+    ground_truth?: string | null | undefined;
+    ground_reasoning: string;
+};
+/** @internal */
+export declare const ArbiterRequest$outboundSchema: z.ZodMiniType<ArbiterRequest$Outbound, ArbiterRequest>;
+export declare function arbiterRequestToJSON(arbiterRequest: ArbiterRequest): string;
+//# sourceMappingURL=arbiter-request.d.ts.map

package/esm/models/arbiter-request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"arbiter-request.d.ts","sourceRoot":"","sources":["../../src/models/arbiter-request.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AAGjC,MAAM,MAAM,cAAc,GAAG;IAC3B,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACrC,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACxC,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC,CAAC;AAEF,gBAAgB;AAChB,MAAM,MAAM,uBAAuB,GAAG;IACpC,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACzC,gBAAgB,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,6BAA6B,EAAE,CAAC,CAAC,WAAW,CACvD,uBAAuB,EACvB,cAAc,CAgBf,CAAC;AAEF,wBAAgB,oBAAoB,CAAC,cAAc,EAAE,cAAc,GAAG,MAAM,CAE3E"}

package/esm/models/arbiter-request.js

@@ -0,0 +1,23 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { remap as remap$ } from "../lib/primitives.js";
+/** @internal */
+export const ArbiterRequest$outboundSchema = z.pipe(z.object({
+    arbiterRepo: z.string(),
+    arbiterRevision: z._default(z.string(), "main"),
+    groundTruth: z.optional(z.nullable(z.string())),
+    groundReasoning: z._default(z.string(), ""),
+}), z.transform((v) => {
+    return remap$(v, {
+        arbiterRepo: "arbiter_repo",
+        arbiterRevision: "arbiter_revision",
+        groundTruth: "ground_truth",
+        groundReasoning: "ground_reasoning",
+    });
+}));
+export function arbiterRequestToJSON(arbiterRequest) {
+    return JSON.stringify(ArbiterRequest$outboundSchema.parse(arbiterRequest));
+}
+//# sourceMappingURL=arbiter-request.js.map

package/esm/models/arbiter-request.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"arbiter-request.js","sourceRoot":"","sources":["../../src/models/arbiter-request.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAiBvD,gBAAgB;AAChB,MAAM,CAAC,MAAM,6BAA6B,GAGtC,CAAC,CAAC,IAAI,CACR,CAAC,CAAC,MAAM,CAAC;IACP,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC;IAC/C,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;CAC5C,CAAC,EACF,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC,CAAC,EAAE;QACf,WAAW,EAAE,cAAc;QAC3B,eAAe,EAAE,kBAAkB;QACnC,WAAW,EAAE,cAAc;QAC3B,eAAe,EAAE,kBAAkB;KACpC,CAAC,CAAC;AAAA,CACJ,CAAC,CACH,CAAC;AAEF,MAAM,UAAU,oBAAoB,CAAC,cAA8B,EAAU;IAC3E,OAAO,IAAI,CAAC,SAAS,CAAC,6BAA6B,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;AAAA,CAC5E"}

package/esm/models/batch-arbiter.d.ts

@@ -0,0 +1,14 @@
+import * as z from "zod/v4-mini";
+export type BatchArbiter = {
+    arbiterRepo: string;
+    arbiterRevision?: string | undefined;
+};
+/** @internal */
+export type BatchArbiter$Outbound = {
+    arbiter_repo: string;
+    arbiter_revision: string;
+};
+/** @internal */
+export declare const BatchArbiter$outboundSchema: z.ZodMiniType<BatchArbiter$Outbound, BatchArbiter>;
+export declare function batchArbiterToJSON(batchArbiter: BatchArbiter): string;
+//# sourceMappingURL=batch-arbiter.d.ts.map

package/esm/models/batch-arbiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-arbiter.d.ts","sourceRoot":"","sources":["../../src/models/batch-arbiter.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AAGjC,MAAM,MAAM,YAAY,GAAG;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;CACtC,CAAC;AAEF,gBAAgB;AAChB,MAAM,MAAM,qBAAqB,GAAG;IAClC,YAAY,EAAE,MAAM,CAAC;IACrB,gBAAgB,EAAE,MAAM,CAAC;CAC1B,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,2BAA2B,EAAE,CAAC,CAAC,WAAW,CACrD,qBAAqB,EACrB,YAAY,CAYb,CAAC;AAEF,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM,CAErE"}

package/esm/models/batch-arbiter.js

@@ -0,0 +1,19 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { remap as remap$ } from "../lib/primitives.js";
+/** @internal */
+export const BatchArbiter$outboundSchema = z.pipe(z.object({
+    arbiterRepo: z.string(),
+    arbiterRevision: z._default(z.string(), "main"),
+}), z.transform((v) => {
+    return remap$(v, {
+        arbiterRepo: "arbiter_repo",
+        arbiterRevision: "arbiter_revision",
+    });
+}));
+export function batchArbiterToJSON(batchArbiter) {
+    return JSON.stringify(BatchArbiter$outboundSchema.parse(batchArbiter));
+}
+//# sourceMappingURL=batch-arbiter.js.map

package/esm/models/batch-arbiter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-arbiter.js","sourceRoot":"","sources":["../../src/models/batch-arbiter.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAavD,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAGpC,CAAC,CAAC,IAAI,CACR,CAAC,CAAC,MAAM,CAAC;IACP,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;IACvB,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC;CAChD,CAAC,EACF,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC,CAAC,EAAE;QACf,WAAW,EAAE,cAAc;QAC3B,eAAe,EAAE,kBAAkB;KACpC,CAAC,CAAC;AAAA,CACJ,CAAC,CACH,CAAC;AAEF,MAAM,UAAU,kBAAkB,CAAC,YAA0B,EAAU;IACrE,OAAO,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;AAAA,CACxE"}

package/esm/models/batch-example.d.ts

@@ -0,0 +1,51 @@
+import * as z from "zod/v4-mini";
+import { ClosedEnum } from "../types/enums.js";
+export declare const BatchExampleSplit: {
+    readonly Train: "train";
+    readonly Test: "test";
+    readonly None: "none";
+};
+export type BatchExampleSplit = ClosedEnum<typeof BatchExampleSplit>;
+/**
+ * One row in a batch predictions request.
+ *
+ * @remarks
+ *
+ * Split contract (mirrors `examples/schemas.py::PredictedExample`):
+ * only `"train"` and `"test"` are valid downstream — `"none"` exists
+ * purely as the storage marker for unannotated rows. Callers MUST NOT
+ * send `"none"` explicitly; omit `split` instead and let `get_split`
+ * randomly assign train/test (80/20) when `ground_truth` is provided.
+ * Without that auto-assignment, annotated batch rows would land in
+ * ClickHouse with `split='none'` and be skipped by every downstream
+ * consumer (GEPA, scoring, dataset export).
+ */
+export type BatchExample = {
+    input: {
+        [k: string]: any;
+    };
+    altId?: string | null | undefined;
+    groundTruth?: {
+        [k: string]: any;
+    } | null | undefined;
+    groundReasoning?: string | undefined;
+    split?: BatchExampleSplit | null | undefined;
+};
+/** @internal */
+export declare const BatchExampleSplit$outboundSchema: z.ZodMiniEnum<typeof BatchExampleSplit>;
+/** @internal */
+export type BatchExample$Outbound = {
+    input: {
+        [k: string]: any;
+    };
+    alt_id?: string | null | undefined;
+    ground_truth?: {
+        [k: string]: any;
+    } | null | undefined;
+    ground_reasoning: string;
+    split?: string | null | undefined;
+};
+/** @internal */
+export declare const BatchExample$outboundSchema: z.ZodMiniType<BatchExample$Outbound, BatchExample>;
+export declare function batchExampleToJSON(batchExample: BatchExample): string;
+//# sourceMappingURL=batch-example.d.ts.map

package/esm/models/batch-example.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-example.d.ts","sourceRoot":"","sources":["../../src/models/batch-example.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AAEjC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,eAAO,MAAM,iBAAiB;;;;CAIpB,CAAC;AACX,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAErE;;;;;;;;;;;;;GAaG;AACH,MAAM,MAAM,YAAY,GAAG;IACzB,KAAK,EAAE;QAAE,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;IAC5B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IAClC,WAAW,CAAC,EAAE;QAAE,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,IAAI,GAAG,SAAS,CAAC;IACtD,eAAe,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACrC,KAAK,CAAC,EAAE,iBAAiB,GAAG,IAAI,GAAG,SAAS,CAAC;CAC9C,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,gCAAgC,EAAE,CAAC,CAAC,WAAW,CAC1D,OAAO,iBAAiB,CACG,CAAC;AAE9B,gBAAgB;AAChB,MAAM,MAAM,qBAAqB,GAAG;IAClC,KAAK,EAAE;QAAE,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,CAAC;IAC5B,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACnC,YAAY,CAAC,EAAE;QAAE,CAAC,CAAC,EAAE,MAAM,GAAG,GAAG,CAAA;KAAE,GAAG,IAAI,GAAG,SAAS,CAAC;IACvD,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;CACnC,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,2BAA2B,EAAE,CAAC,CAAC,WAAW,CACrD,qBAAqB,EACrB,YAAY,CAgBb,CAAC;AAEF,wBAAgB,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM,CAErE"}

package/esm/models/batch-example.js

@@ -0,0 +1,30 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { remap as remap$ } from "../lib/primitives.js";
+export const BatchExampleSplit = {
+    Train: "train",
+    Test: "test",
+    None: "none",
+};
+/** @internal */
+export const BatchExampleSplit$outboundSchema = z.enum(BatchExampleSplit);
+/** @internal */
+export const BatchExample$outboundSchema = z.pipe(z.object({
+    input: z.record(z.string(), z.any()),
+    altId: z.optional(z.nullable(z.string())),
+    groundTruth: z.optional(z.nullable(z.record(z.string(), z.any()))),
+    groundReasoning: z._default(z.string(), ""),
+    split: z.optional(z.nullable(BatchExampleSplit$outboundSchema)),
+}), z.transform((v) => {
+    return remap$(v, {
+        altId: "alt_id",
+        groundTruth: "ground_truth",
+        groundReasoning: "ground_reasoning",
+    });
+}));
+export function batchExampleToJSON(batchExample) {
+    return JSON.stringify(BatchExample$outboundSchema.parse(batchExample));
+}
+//# sourceMappingURL=batch-example.js.map

package/esm/models/batch-example.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-example.js","sourceRoot":"","sources":["../../src/models/batch-example.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,MAAM,EAAE,MAAM,sBAAsB,CAAC;AAGvD,MAAM,CAAC,MAAM,iBAAiB,GAAG;IAC/B,KAAK,EAAE,OAAO;IACd,IAAI,EAAE,MAAM;IACZ,IAAI,EAAE,MAAM;CACJ,CAAC;AAyBX,gBAAgB;AAChB,MAAM,CAAC,MAAM,gCAAgC,GAEzC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAW9B,gBAAgB;AAChB,MAAM,CAAC,MAAM,2BAA2B,GAGpC,CAAC,CAAC,IAAI,CACR,CAAC,CAAC,MAAM,CAAC;IACP,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC;IACpC,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IACzC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IAClE,eAAe,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;IAC3C,KAAK,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,gCAAgC,CAAC,CAAC;CAChE,CAAC,EACF,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC,CAAC,EAAE;QACf,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,cAAc;QAC3B,eAAe,EAAE,kBAAkB;KACpC,CAAC,CAAC;AAAA,CACJ,CAAC,CACH,CAAC;AAEF,MAAM,UAAU,kBAAkB,CAAC,YAA0B,EAAU;IACrE,OAAO,IAAI,CAAC,SAAS,CAAC,2BAA2B,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;AAAA,CACxE"}

package/esm/models/batch-predictions-job-request.d.ts

@@ -0,0 +1,30 @@
+import * as z from "zod/v4-mini";
+import { BatchArbiter, BatchArbiter$Outbound } from "./batch-arbiter.js";
+import { BatchExample, BatchExample$Outbound } from "./batch-example.js";
+/**
+ * Request body for ``POST /api/v1/jobs/batch/predictions``.
+ *
+ * @remarks
+ *
+ * Either ``examples`` or ``example_ids`` must be set, not both. The
+ * ``example_ids`` path re-predicts existing examples (by their ClickHouse
+ * ids) instead of ingesting brand-new ones; predictions are inserted as new
+ * versions on the existing example rows.
+ */
+export type BatchPredictionsJobRequest = {
+    arbiters: Array<BatchArbiter>;
+    examples?: Array<BatchExample> | null | undefined;
+    exampleIds?: Array<string> | null | undefined;
+    computeConfidence?: boolean | undefined;
+};
+/** @internal */
+export type BatchPredictionsJobRequest$Outbound = {
+    arbiters: Array<BatchArbiter$Outbound>;
+    examples?: Array<BatchExample$Outbound> | null | undefined;
+    example_ids?: Array<string> | null | undefined;
+    compute_confidence: boolean;
+};
+/** @internal */
+export declare const BatchPredictionsJobRequest$outboundSchema: z.ZodMiniType<BatchPredictionsJobRequest$Outbound, BatchPredictionsJobRequest>;
+export declare function batchPredictionsJobRequestToJSON(batchPredictionsJobRequest: BatchPredictionsJobRequest): string;
+//# sourceMappingURL=batch-predictions-job-request.d.ts.map

package/esm/models/batch-predictions-job-request.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-predictions-job-request.d.ts","sourceRoot":"","sources":["../../src/models/batch-predictions-job-request.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AAEjC,OAAO,EACL,YAAY,EACZ,qBAAqB,EAEtB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACL,YAAY,EACZ,qBAAqB,EAEtB,MAAM,oBAAoB,CAAC;AAE5B;;;;;;;;;GASG;AACH,MAAM,MAAM,0BAA0B,GAAG;IACvC,QAAQ,EAAE,KAAK,CAAC,YAAY,CAAC,CAAC;IAC9B,QAAQ,CAAC,EAAE,KAAK,CAAC,YAAY,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IAClD,UAAU,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IAC9C,iBAAiB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACzC,CAAC;AAEF,gBAAgB;AAChB,MAAM,MAAM,mCAAmC,GAAG;IAChD,QAAQ,EAAE,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACvC,QAAQ,CAAC,EAAE,KAAK,CAAC,qBAAqB,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IAC3D,WAAW,CAAC,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IAC/C,kBAAkB,EAAE,OAAO,CAAC;CAC7B,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,yCAAyC,EAAE,CAAC,CAAC,WAAW,CACnE,mCAAmC,EACnC,0BAA0B,CAc3B,CAAC;AAEF,wBAAgB,gCAAgC,CAC9C,0BAA0B,EAAE,0BAA0B,GACrD,MAAM,CAIR"}

package/esm/models/batch-predictions-job-request.js

@@ -0,0 +1,23 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { remap as remap$ } from "../lib/primitives.js";
+import { BatchArbiter$outboundSchema, } from "./batch-arbiter.js";
+import { BatchExample$outboundSchema, } from "./batch-example.js";
+/** @internal */
+export const BatchPredictionsJobRequest$outboundSchema = z.pipe(z.object({
+    arbiters: z.array(BatchArbiter$outboundSchema),
+    examples: z.optional(z.nullable(z.array(BatchExample$outboundSchema))),
+    exampleIds: z.optional(z.nullable(z.array(z.string()))),
+    computeConfidence: z._default(z.boolean(), false),
+}), z.transform((v) => {
+    return remap$(v, {
+        exampleIds: "example_ids",
+        computeConfidence: "compute_confidence",
+    });
+}));
+export function batchPredictionsJobRequestToJSON(batchPredictionsJobRequest) {
+    return JSON.stringify(BatchPredictionsJobRequest$outboundSchema.parse(batchPredictionsJobRequest));
+}
+//# sourceMappingURL=batch-predictions-job-request.js.map

package/esm/models/batch-predictions-job-request.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-predictions-job-request.js","sourceRoot":"","sources":["../../src/models/batch-predictions-job-request.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,MAAM,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAGL,2BAA2B,GAC5B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAGL,2BAA2B,GAC5B,MAAM,oBAAoB,CAAC;AA2B5B,gBAAgB;AAChB,MAAM,CAAC,MAAM,yCAAyC,GAGlD,CAAC,CAAC,IAAI,CACR,CAAC,CAAC,MAAM,CAAC;IACP,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC;IAC9C,QAAQ,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAC;IACtE,UAAU,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACvD,iBAAiB,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC;CAClD,CAAC,EACF,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC,CAAC,EAAE;QACf,UAAU,EAAE,aAAa;QACzB,iBAAiB,EAAE,oBAAoB;KACxC,CAAC,CAAC;AAAA,CACJ,CAAC,CACH,CAAC;AAEF,MAAM,UAAU,gCAAgC,CAC9C,0BAAsD,EAC9C;IACR,OAAO,IAAI,CAAC,SAAS,CACnB,yCAAyC,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAC5E,CAAC;AAAA,CACH"}

package/esm/models/chat-completion-choice.d.ts

@@ -0,0 +1,13 @@
+import * as z from "zod/v4-mini";
+import { Result as SafeParseResult } from "../types/fp.js";
+import { ChatMessage } from "./chat-message.js";
+import { SDKValidationError } from "./errors/sdk-validation-error.js";
+export type ChatCompletionChoice = {
+    index: number;
+    message: ChatMessage;
+    finishReason?: string | null | undefined;
+};
+/** @internal */
+export declare const ChatCompletionChoice$inboundSchema: z.ZodMiniType<ChatCompletionChoice, unknown>;
+export declare function chatCompletionChoiceFromJSON(jsonString: string): SafeParseResult<ChatCompletionChoice, SDKValidationError>;
+//# sourceMappingURL=chat-completion-choice.d.ts.map

package/esm/models/chat-completion-choice.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-completion-choice.d.ts","sourceRoot":"","sources":["../../src/models/chat-completion-choice.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AAGjC,OAAO,EAAE,MAAM,IAAI,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAE3D,OAAO,EAAE,WAAW,EAA6B,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAAE,kBAAkB,EAAE,MAAM,kCAAkC,CAAC;AAEtE,MAAM,MAAM,oBAAoB,GAAG;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,WAAW,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;CAC1C,CAAC;AAEF,gBAAgB;AAChB,eAAO,MAAM,kCAAkC,EAAE,CAAC,CAAC,WAAW,CAC5D,oBAAoB,EACpB,OAAO,CAYR,CAAC;AAEF,wBAAgB,4BAA4B,CAC1C,UAAU,EAAE,MAAM,GACjB,eAAe,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,CAM3D"}

package/esm/models/chat-completion-choice.js

@@ -0,0 +1,22 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+import * as z from "zod/v4-mini";
+import { remap as remap$ } from "../lib/primitives.js";
+import { safeParse } from "../lib/schemas.js";
+import * as types from "../types/primitives.js";
+import { ChatMessage$inboundSchema } from "./chat-message.js";
+/** @internal */
+export const ChatCompletionChoice$inboundSchema = z.pipe(z.object({
+    index: types.number(),
+    message: ChatMessage$inboundSchema,
+    finish_reason: z.optional(z.nullable(types.string())),
+}), z.transform((v) => {
+    return remap$(v, {
+        "finish_reason": "finishReason",
+    });
+}));
+export function chatCompletionChoiceFromJSON(jsonString) {
+    return safeParse(jsonString, (x) => ChatCompletionChoice$inboundSchema.parse(JSON.parse(x)), `Failed to parse 'ChatCompletionChoice' from JSON`);
+}
+//# sourceMappingURL=chat-completion-choice.js.map

package/esm/models/chat-completion-choice.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat-completion-choice.js","sourceRoot":"","sources":["../../src/models/chat-completion-choice.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,CAAC,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,KAAK,IAAI,MAAM,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAE9C,OAAO,KAAK,KAAK,MAAM,wBAAwB,CAAC;AAChD,OAAO,EAAe,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAS3E,gBAAgB;AAChB,MAAM,CAAC,MAAM,kCAAkC,GAG3C,CAAC,CAAC,IAAI,CACR,CAAC,CAAC,MAAM,CAAC;IACP,KAAK,EAAE,KAAK,CAAC,MAAM,EAAE;IACrB,OAAO,EAAE,yBAAyB;IAClC,aAAa,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;CACtD,CAAC,EACF,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;IACjB,OAAO,MAAM,CAAC,CAAC,EAAE;QACf,eAAe,EAAE,cAAc;KAChC,CAAC,CAAC;AAAA,CACJ,CAAC,CACH,CAAC;AAEF,MAAM,UAAU,4BAA4B,CAC1C,UAAkB,EACyC;IAC3D,OAAO,SAAS,CACd,UAAU,EACV,CAAC,CAAC,EAAE,EAAE,CAAC,kCAAkC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAC9D,kDAAkD,CACnD,CAAC;AAAA,CACH"}