@superlinked/sie-sdk 0.1.9 → 0.2.0

package/dist/index.cjs

@@ -107,7 +107,7 @@ var SDK_VERSION_HEADER = "X-SIE-SDK-Version";
 var SERVER_VERSION_HEADER = "X-SIE-Server-Version";
 
 // src/version.ts
-var SDK_VERSION = "0.1.9";
+var SDK_VERSION = "0.2.0";
 var EXT_TYPE_NUMPY = 78;
 function parseDtype(dtype) {
   const typeChar = dtype.slice(-2, -1);
@@ -424,7 +424,28 @@ function parseEntity(data) {
 function parseExtractResult(data) {
   return {
     id: data.id,
-    entities: data.entities.map(parseEntity)
+    entities: data.entities.map(parseEntity),
+    relations: (data.relations ?? []).map(
+      (r) => ({
+        head: r.head,
+        tail: r.tail,
+        relation: r.relation,
+        score: r.score
+      })
+    ),
+    classifications: (data.classifications ?? []).map(
+      (c) => ({
+        label: c.label,
+        score: c.score
+      })
+    ),
+    objects: (data.objects ?? []).map(
+      (o) => ({
+        label: o.label,
+        score: o.score,
+        bbox: o.bbox
+      })
+    )
   };
 }
 function parseExtractResults(data) {
@@ -572,6 +593,28 @@ var SIEClient = class {
       maxSequenceLength: m.max_sequence_length
     }));
   }
+  /**
+   * Get details for a specific model.
+   *
+   * Returns model metadata including dimensions, supported inputs/outputs,
+   * loaded status, and max sequence length. This is a lightweight call that
+   * reads from model config — it does not load the model or trigger inference.
+   *
+   * @param name - Model name (e.g., "BAAI/bge-m3")
+   * @returns Model information
+   */
+  async getModel(name) {
+    const response = await this.requestJson(`/v1/models/${encodeURIComponent(name)}`, "GET");
+    const data = await response.json();
+    return {
+      name: data.name,
+      loaded: data.loaded,
+      inputs: data.inputs,
+      outputs: data.outputs,
+      dims: data.dims,
+      maxSequenceLength: data.max_sequence_length
+    };
+  }
   /**
    * Stream real-time status updates from a worker or router.
    *
@@ -676,9 +719,6 @@ var SIEClient = class {
       query,
       items
     };
-    if (options.topK !== void 0) {
-      body.top_k = options.topK;
-    }
     const waitForCapacity = options.waitForCapacity ?? this.defaultWaitForCapacity;
     const { pool, gpu } = this.parseGpuParam(options.gpu);
     const response = await this.requestWithRetry(
@@ -1295,6 +1335,50 @@ function toFloat32Array(arr) {
   return new Float32Array(arr);
 }
 
+// src/encoding.ts
+function denseEmbedding(result, strict = true) {
+  const dense = result.dense;
+  if (!dense) {
+    if (strict) {
+      throw new Error("Encode result missing dense embedding");
+    }
+    return [];
+  }
+  return toNumberArray(dense);
+}
+function sparseEmbedding(result) {
+  const sparse = result.sparse;
+  if (!sparse) {
+    return { indices: [], values: [] };
+  }
+  return {
+    indices: toNumberArray(sparse.indices),
+    values: toNumberArray(sparse.values)
+  };
+}
+function sparseEmbeddingMap(result) {
+  const sparse = result.sparse;
+  if (!sparse) {
+    return /* @__PURE__ */ new Map();
+  }
+  const indices = toNumberArray(sparse.indices);
+  const values = toNumberArray(sparse.values);
+  const map = /* @__PURE__ */ new Map();
+  for (let i = 0; i < indices.length; i++) {
+    map.set(indices[i], values[i]);
+  }
+  return map;
+}
+function normalizeSparseVector(sparse) {
+  return {
+    indices: toNumberArray(sparse.indices),
+    values: toNumberArray(sparse.values)
+  };
+}
+function multivectorEmbedding(raw) {
+  return raw.map((v) => toNumberArray(v));
+}
+
 // src/scoring.ts
 function maxsim(query, document) {
   if (query.length === 0 || document.length === 0) {
@@ -1392,11 +1476,16 @@ exports.SIEClient = SIEClient;
 exports.SIEConnectionError = SIEConnectionError;
 exports.SIEError = SIEError;
 exports.ServerError = ServerError;
+exports.denseEmbedding = denseEmbedding;
 exports.detectImageFormat = detectImageFormat;
 exports.maxsim = maxsim;
 exports.maxsimBatch = maxsimBatch;
 exports.maxsimDocuments = maxsimDocuments;
+exports.multivectorEmbedding = multivectorEmbedding;
+exports.normalizeSparseVector = normalizeSparseVector;
 exports.packMessage = packMessage;
+exports.sparseEmbedding = sparseEmbedding;
+exports.sparseEmbeddingMap = sparseEmbeddingMap;
 exports.toFloat32Array = toFloat32Array;
 exports.toImageBytes = toImageBytes;
 exports.toImageWireFormat = toImageWireFormat;