@superlinked/sie-sdk 0.6.22 → 0.6.24

package/dist/index.cjs

@@ -1191,7 +1191,7 @@ function extractDataPayload(block) {
 }
 
 // src/version.ts
-var SDK_VERSION = "0.6.22";
+var SDK_VERSION = "0.6.24";
 
 // src/client.ts
 function sleep2(ms) {
@@ -1245,6 +1245,33 @@ function validateGenerationSeed(seed) {
   }
   return seed;
 }
+function validateGenerateGrammar(grammar) {
+  if (typeof grammar !== "object" || grammar === null || Array.isArray(grammar)) {
+    throw new TypeError("grammar must be an object");
+  }
+  const allowed = /* @__PURE__ */ new Set(["json_schema", "regex", "ebnf", "label", "strict"]);
+  const unknown = Object.keys(grammar).filter((key) => !allowed.has(key));
+  if (unknown.length > 0) {
+    throw new TypeError(`grammar contains unsupported field(s): ${unknown.sort().join(", ")}`);
+  }
+  const variants = ["json_schema", "regex", "ebnf"].filter((key) => Object.hasOwn(grammar, key));
+  if (variants.length !== 1) {
+    throw new TypeError("grammar must contain exactly one of json_schema, regex, or ebnf");
+  }
+  const variant = variants[0];
+  const value = grammar[variant];
+  if (variant === "json_schema" ? typeof value !== "object" || value === null || Array.isArray(value) : typeof value !== "string") {
+    throw new TypeError(
+      variant === "json_schema" ? "grammar.json_schema must be an object" : `grammar.${variant} must be a string`
+    );
+  }
+  if (grammar.label !== void 0 && grammar.label !== null && typeof grammar.label !== "string") {
+    throw new TypeError("grammar.label must be a string");
+  }
+  if (grammar.strict !== void 0 && grammar.strict !== null && typeof grammar.strict !== "boolean") {
+    throw new TypeError("grammar.strict must be a boolean");
+  }
+}
 function applyGenerateOptions(body, options) {
   if (options.temperature !== void 0) body.temperature = options.temperature;
   if (options.topP !== void 0) body.top_p = options.topP;
@@ -1252,7 +1279,10 @@ function applyGenerateOptions(body, options) {
   if (options.stop !== void 0) body.stop = options.stop;
   if (options.frequencyPenalty !== void 0) body.frequency_penalty = options.frequencyPenalty;
   if (options.presencePenalty !== void 0) body.presence_penalty = options.presencePenalty;
-  if (options.grammar !== void 0) body.grammar = options.grammar;
+  if (options.grammar !== void 0) {
+    validateGenerateGrammar(options.grammar);
+    body.grammar = options.grammar;
+  }
   if (options.seed !== void 0) body.seed = validateGenerationSeed(options.seed);
   if (options.logitBias !== void 0) body.logit_bias = options.logitBias;
   if (options.routingKey !== void 0) body.routing_key = options.routingKey;
@@ -1292,6 +1322,25 @@ async function imageForWire(image) {
   }
   return toImageWireFormat(image);
 }
+function imageBytesToBase64(data) {
+  if (typeof Buffer !== "undefined") {
+    return Buffer.from(data.buffer, data.byteOffset, data.byteLength).toString("base64");
+  }
+  let binary = "";
+  const chunkSize = 32768;
+  for (let offset = 0; offset < data.length; offset += chunkSize) {
+    binary += String.fromCharCode(...data.subarray(offset, offset + chunkSize));
+  }
+  return btoa(binary);
+}
+async function generationImagesForWire(images) {
+  return Promise.all(
+    images.map(async (image) => {
+      const wire = await imageForWire(image);
+      return { data: imageBytesToBase64(wire.data), format: wire.format };
+    })
+  );
+}
 async function itemImagesForWire(item) {
   if (!item.images || item.images.length === 0) {
     return item;
@@ -1623,6 +1672,9 @@ var SIEClient = class {
       prompt,
       max_new_tokens: options.maxNewTokens
     };
+    if (options.images !== void 0) {
+      body.images = await generationImagesForWire(options.images);
+    }
     applyGenerateOptions(body, options);
     const { pool, gpu } = this.parseGpuParam(options.gpu);
     const headers = {
@@ -1843,6 +1895,9 @@ var SIEClient = class {
       max_new_tokens: options.maxNewTokens,
       stream: true
     };
+    if (options.images !== void 0) {
+      body.images = await generationImagesForWire(options.images);
+    }
     applyGenerateOptions(body, options);
     if (options.logprobs !== void 0) body.logprobs = options.logprobs;
     if (options.topLogprobs !== void 0) {