@workglow/huggingface-transformers 0.2.36 → 0.3.0

package/dist/ai.js

@@ -987,21 +987,6 @@ var HFT_BackgroundRemoval = async (input, model, signal, emit) => {
 // src/ai/common/HFT_Chat.ts
 init_HFT_Pipeline();
 
-// src/ai/common/HFT_ToolCalling.ts
-init_HFT_Pipeline();
-import {
-  buildToolDescription,
-  filterValidToolCalls,
-  toTextFlatMessages
-} from "@workglow/ai/worker";
-import {
-  adaptParserResult,
-  forcedToolSelection,
-  getAvailableParsers,
-  getGenerationPrefix,
-  parseToolCalls
-} from "@workglow/ai/provider-utils";
-
 // src/ai/common/HFT_Streaming.ts
 function createStreamingTextStreamer(tokenizer, onText, textStreamer) {
   return new textStreamer(tokenizer, {
@@ -1025,6 +1010,19 @@ function createTextStreamer(tokenizer, updateProgress, textStreamer) {
 }
 
 // src/ai/common/HFT_ToolCalling.ts
+init_HFT_Pipeline();
+import {
+  adaptParserResult,
+  forcedToolSelection,
+  getAvailableParsers,
+  getGenerationPrefix,
+  parseToolCalls
+} from "@workglow/ai/provider-utils";
+import {
+  buildToolDescription,
+  filterValidToolCalls,
+  toTextFlatMessages
+} from "@workglow/ai/worker";
 function getModelTextCandidates(model) {
   return [model.model_id, model.title, model.description, model.provider_config.model_path].filter((value) => typeof value === "string" && value.length > 0).map((value) => value.toLowerCase());
 }
@@ -1444,8 +1442,8 @@ var HFT_ImageClassification = async (input, model, signal, emit) => {
 
 // src/ai/common/HFT_ImageEmbedding.ts
 init_HFT_Pipeline();
-import { getLogger as getLogger3 } from "@workglow/util/worker";
 import { imageValueToBlob as imageValueToBlob3 } from "@workglow/ai/provider-utils";
+import { getLogger as getLogger3 } from "@workglow/util/worker";
 var HFT_ImageEmbedding = async (input, model, signal, emit) => {
   const logger = getLogger3();
   const timerLabel = `hft:ImageEmbedding:${model?.provider_config.model_path}`;
@@ -1477,6 +1475,29 @@ var HFT_ImageEmbedding = async (input, model, signal, emit) => {
 // src/ai/common/HFT_ImageSegmentation.ts
 init_HFT_Pipeline();
 import { imageValueToBlob as imageValueToBlob4 } from "@workglow/ai/provider-utils";
+import { imageValueFromBuffer } from "@workglow/util/media";
+function rawImageToImageValue(image) {
+  const raw = image;
+  const pixelCount = raw.width * raw.height;
+  const rgba = new Uint8Array(pixelCount * 4);
+  for (let pixel = 0;pixel < pixelCount; pixel++) {
+    const sourceOffset = pixel * raw.channels;
+    const targetOffset = pixel * 4;
+    if (raw.channels === 1) {
+      const value = raw.data[sourceOffset] ?? 0;
+      rgba[targetOffset] = value;
+      rgba[targetOffset + 1] = value;
+      rgba[targetOffset + 2] = value;
+      rgba[targetOffset + 3] = 255;
+    } else {
+      rgba[targetOffset] = raw.data[sourceOffset] ?? 0;
+      rgba[targetOffset + 1] = raw.data[sourceOffset + 1] ?? raw.data[sourceOffset] ?? 0;
+      rgba[targetOffset + 2] = raw.data[sourceOffset + 2] ?? raw.data[sourceOffset] ?? 0;
+      rgba[targetOffset + 3] = raw.channels >= 4 ? raw.data[sourceOffset + 3] ?? 255 : 255;
+    }
+  }
+  return imageValueFromBuffer(Buffer.from(rgba), "raw-rgba", raw.width, raw.height);
+}
 var HFT_ImageSegmentation = async (input, model, signal, emit) => {
   const segmenter = await getPipeline(model, emit, {}, signal);
   const imageArg = await imageValueToBlob4(input.image);
@@ -1485,11 +1506,13 @@ var HFT_ImageSegmentation = async (input, model, signal, emit) => {
     mask_threshold: input.maskThreshold
   });
   const masks = Array.isArray(result) ? result : [result];
-  const processedMasks = await Promise.all(masks.map(async (mask) => ({
-    label: mask.label || "",
-    score: mask.score || 0,
-    mask: {}
-  })));
+  const processedMasks = masks.map((mask) => {
+    return {
+      label: mask.label || "",
+      score: mask.score || 0,
+      mask: rawImageToImageValue(mask.mask)
+    };
+  });
   emit({
     type: "finish",
     data: {
@@ -1616,7 +1639,7 @@ var HFT_ModelInfo = async (input, model, _signal, emit) => {
 };
 
 // src/ai/common/HFT_ModelSearch.ts
-import { searchHfModels, mapHfModelResult } from "@workglow/ai/provider-utils";
+import { mapHfModelResult, searchHfModels } from "@workglow/ai/provider-utils";
 var HFT_ModelSearch = async (input, _model, signal, emit) => {
   const entries = await searchHfModels(input.query?.trim() ?? "", { filter: "onnx" }, ["siblings"], signal);
   const results = entries.map((entry) => {
@@ -2127,4 +2150,4 @@ export {
   HF_TRANSFORMERS_ONNX
 };
 
-//# debugId=FC830CCDC8DBF09664756E2164756E21
+//# debugId=F1F15E5800BCD03B64756E2164756E21