@workglow/ai-provider 0.0.90 → 0.0.91

package/dist/google-gemini/index.js.map

@@ -0,0 +1,14 @@
+{
+  "version": 3,
+  "sources": ["../src/google-gemini/common/Gemini_Constants.ts", "../src/google-gemini/common/Gemini_JobRunFns.ts", "../src/google-gemini/common/Gemini_ModelSchema.ts", "../src/google-gemini/Gemini_Worker.ts", "../src/google-gemini/GoogleGeminiProvider.ts"],
+  "sourcesContent": [
+    "/**\n * @license\n * Copyright 2025 Steven Roussey <sroussey@gmail.com>\n * SPDX-License-Identifier: Apache-2.0\n */\n\nexport const GOOGLE_GEMINI = \"GOOGLE_GEMINI\";\n",
+    "/**\n * @license\n * Copyright 2025 Steven Roussey <sroussey@gmail.com>\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport type { TaskType } from \"@google/generative-ai\";\nimport type {\n  AiProviderRunFn,\n  AiProviderStreamFn,\n  TextEmbeddingTaskInput,\n  TextEmbeddingTaskOutput,\n  TextGenerationTaskInput,\n  TextGenerationTaskOutput,\n  TextRewriterTaskInput,\n  TextRewriterTaskOutput,\n  TextSummaryTaskInput,\n  TextSummaryTaskOutput,\n} from \"@workglow/ai\";\nimport type { StreamEvent } from \"@workglow/task-graph\";\nimport type { GeminiModelConfig } from \"./Gemini_ModelSchema\";\n\nlet _sdk: typeof import(\"@google/generative-ai\") | undefined;\nasync function loadGeminiSDK() {\n  if (!_sdk) {\n    try {\n      _sdk = await import(\"@google/generative-ai\");\n    } catch {\n      throw new Error(\n        \"@google/generative-ai is required for Gemini tasks. Install it with: bun add @google/generative-ai\"\n      );\n    }\n  }\n  return _sdk.GoogleGenerativeAI;\n}\n\nfunction getApiKey(model: GeminiModelConfig | undefined): string {\n  const apiKey =\n    model?.provider_config?.api_key ||\n    (typeof process !== \"undefined\"\n      ? process.env?.GOOGLE_API_KEY || process.env?.GEMINI_API_KEY\n      : undefined);\n  if (!apiKey) {\n    throw new Error(\n      \"Missing Google API key: set provider_config.api_key or the GOOGLE_API_KEY / GEMINI_API_KEY environment variable.\"\n    );\n  }\n  return apiKey;\n}\n\nfunction getModelName(model: GeminiModelConfig | undefined): string {\n  const name = model?.provider_config?.model_name;\n  if (!name) {\n    throw new Error(\"Missing model name in provider_config.model_name.\");\n  }\n  return name;\n}\n\nexport const Gemini_TextGeneration: AiProviderRunFn<\n  TextGenerationTaskInput,\n  TextGenerationTaskOutput,\n  GeminiModelConfig\n> = async (input, model, update_progress, signal) => {\n  update_progress(0, \"Starting Gemini text generation\");\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    generationConfig: {\n      maxOutputTokens: input.maxTokens,\n      temperature: input.temperature,\n      topP: input.topP,\n    },\n  });\n\n  const result = await genModel.generateContent({\n    contents: [{ role: \"user\", parts: [{ text: input.prompt }] }],\n  });\n\n  const text = result.response.text();\n  update_progress(100, \"Completed Gemini text generation\");\n  return { text };\n};\n\nexport const Gemini_TextEmbedding: AiProviderRunFn<\n  TextEmbeddingTaskInput,\n  TextEmbeddingTaskOutput,\n  GeminiModelConfig\n> = async (input, model, update_progress, signal) => {\n  update_progress(0, \"Starting Gemini text embedding\");\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const embeddingModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n  });\n\n  const taskType =\n    (model?.provider_config?.embedding_task_type as TaskType) || (\"RETRIEVAL_DOCUMENT\" as TaskType);\n\n  if (Array.isArray(input.text)) {\n    const result = await embeddingModel.batchEmbedContents({\n      requests: input.text.map((t) => ({\n        content: { role: \"user\", parts: [{ text: t }] },\n        taskType,\n      })),\n    });\n    update_progress(100, \"Completed Gemini text embedding\");\n    return {\n      vector: result.embeddings.map((e) => new Float32Array(e.values)),\n    };\n  }\n\n  const result = await embeddingModel.embedContent({\n    content: { role: \"user\", parts: [{ text: input.text }] },\n    taskType,\n  });\n\n  update_progress(100, \"Completed Gemini text embedding\");\n  return { vector: new Float32Array(result.embedding.values) };\n};\n\nexport const Gemini_TextRewriter: AiProviderRunFn<\n  TextRewriterTaskInput,\n  TextRewriterTaskOutput,\n  GeminiModelConfig\n> = async (input, model, update_progress, signal) => {\n  update_progress(0, \"Starting Gemini text rewriting\");\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    systemInstruction: input.prompt,\n  });\n\n  const result = await genModel.generateContent({\n    contents: [{ role: \"user\", parts: [{ text: input.text }] }],\n  });\n\n  const text = result.response.text();\n  update_progress(100, \"Completed Gemini text rewriting\");\n  return { text };\n};\n\nexport const Gemini_TextSummary: AiProviderRunFn<\n  TextSummaryTaskInput,\n  TextSummaryTaskOutput,\n  GeminiModelConfig\n> = async (input, model, update_progress, signal) => {\n  update_progress(0, \"Starting Gemini text summarization\");\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    systemInstruction: \"Summarize the following text concisely.\",\n  });\n\n  const result = await genModel.generateContent({\n    contents: [{ role: \"user\", parts: [{ text: input.text }] }],\n  });\n\n  const text = result.response.text();\n  update_progress(100, \"Completed Gemini text summarization\");\n  return { text };\n};\n\n// ========================================================================\n// Streaming implementations (append mode)\n// ========================================================================\n\nexport const Gemini_TextGeneration_Stream: AiProviderStreamFn<\n  TextGenerationTaskInput,\n  TextGenerationTaskOutput,\n  GeminiModelConfig\n> = async function* (input, model, signal): AsyncIterable<StreamEvent<TextGenerationTaskOutput>> {\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    generationConfig: {\n      maxOutputTokens: input.maxTokens,\n      temperature: input.temperature,\n      topP: input.topP,\n    },\n  });\n\n  const result = await genModel.generateContentStream({\n    contents: [{ role: \"user\", parts: [{ text: input.prompt }] }],\n  });\n\n  for await (const chunk of result.stream) {\n    const text = chunk.text();\n    if (text) {\n      yield { type: \"text-delta\", textDelta: text };\n    }\n  }\n  yield { type: \"finish\", data: {} as TextGenerationTaskOutput };\n};\n\nexport const Gemini_TextRewriter_Stream: AiProviderStreamFn<\n  TextRewriterTaskInput,\n  TextRewriterTaskOutput,\n  GeminiModelConfig\n> = async function* (input, model, signal): AsyncIterable<StreamEvent<TextRewriterTaskOutput>> {\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    systemInstruction: input.prompt,\n  });\n\n  const result = await genModel.generateContentStream({\n    contents: [{ role: \"user\", parts: [{ text: input.text }] }],\n  });\n\n  for await (const chunk of result.stream) {\n    const text = chunk.text();\n    if (text) {\n      yield { type: \"text-delta\", textDelta: text };\n    }\n  }\n  yield { type: \"finish\", data: {} as TextRewriterTaskOutput };\n};\n\nexport const Gemini_TextSummary_Stream: AiProviderStreamFn<\n  TextSummaryTaskInput,\n  TextSummaryTaskOutput,\n  GeminiModelConfig\n> = async function* (input, model, signal): AsyncIterable<StreamEvent<TextSummaryTaskOutput>> {\n  const GoogleGenerativeAI = await loadGeminiSDK();\n  const genAI = new GoogleGenerativeAI(getApiKey(model));\n  const genModel = genAI.getGenerativeModel({\n    model: getModelName(model),\n    systemInstruction: \"Summarize the following text concisely.\",\n  });\n\n  const result = await genModel.generateContentStream({\n    contents: [{ role: \"user\", parts: [{ text: input.text }] }],\n  });\n\n  for await (const chunk of result.stream) {\n    const text = chunk.text();\n    if (text) {\n      yield { type: \"text-delta\", textDelta: text };\n    }\n  }\n  yield { type: \"finish\", data: {} as TextSummaryTaskOutput };\n};\n\n// ========================================================================\n// Task registries\n// ========================================================================\n\nexport const GEMINI_TASKS: Record<string, AiProviderRunFn<any, any, GeminiModelConfig>> = {\n  TextGenerationTask: Gemini_TextGeneration,\n  TextEmbeddingTask: Gemini_TextEmbedding,\n  TextRewriterTask: Gemini_TextRewriter,\n  TextSummaryTask: Gemini_TextSummary,\n};\n\nexport const GEMINI_STREAM_TASKS: Record<\n  string,\n  AiProviderStreamFn<any, any, GeminiModelConfig>\n> = {\n  TextGenerationTask: Gemini_TextGeneration_Stream,\n  TextRewriterTask: Gemini_TextRewriter_Stream,\n  TextSummaryTask: Gemini_TextSummary_Stream,\n};\n",
+    "/**\n * @license\n * Copyright 2025 Steven Roussey <sroussey@gmail.com>\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { ModelConfigSchema, ModelRecordSchema } from \"@workglow/ai\";\nimport { DataPortSchemaObject, FromSchema } from \"@workglow/util\";\nimport { GOOGLE_GEMINI } from \"./Gemini_Constants\";\n\nexport const GeminiModelSchema = {\n  type: \"object\",\n  properties: {\n    provider: {\n      const: GOOGLE_GEMINI,\n      description: \"Discriminator: Google Gemini cloud provider.\",\n    },\n    provider_config: {\n      type: \"object\",\n      description: \"Google Gemini-specific configuration.\",\n      properties: {\n        model_name: {\n          type: \"string\",\n          description:\n            \"The Gemini model identifier (e.g., 'gemini-2.0-flash', 'text-embedding-004').\",\n        },\n        api_key: {\n          type: \"string\",\n          description: \"Google AI API key. Falls back to default API key if not set.\",\n        },\n        embedding_task_type: {\n          oneOf: [\n            { type: \"null\" },\n            {\n              type: \"string\",\n              enum: [\n                \"RETRIEVAL_QUERY\",\n                \"RETRIEVAL_DOCUMENT\",\n                \"SEMANTIC_SIMILARITY\",\n                \"CLASSIFICATION\",\n                \"CLUSTERING\",\n              ],\n            },\n          ],\n          description: \"Task type hint for embedding models.\",\n          default: null,\n        },\n      },\n      required: [\"model_name\"],\n      additionalProperties: false,\n    },\n  },\n  required: [\"provider\", \"provider_config\"],\n  additionalProperties: true,\n} as const satisfies DataPortSchemaObject;\n\nexport const GeminiModelRecordSchema = {\n  type: \"object\",\n  properties: {\n    ...ModelRecordSchema.properties,\n    ...GeminiModelSchema.properties,\n  },\n  required: [...ModelRecordSchema.required, ...GeminiModelSchema.required],\n  additionalProperties: false,\n} as const satisfies DataPortSchemaObject;\n\nexport type GeminiModelRecord = FromSchema<typeof GeminiModelRecordSchema>;\n\nexport const GeminiModelConfigSchema = {\n  type: \"object\",\n  properties: {\n    ...ModelConfigSchema.properties,\n    ...GeminiModelSchema.properties,\n  },\n  required: [...ModelConfigSchema.required, ...GeminiModelSchema.required],\n  additionalProperties: false,\n} as const satisfies DataPortSchemaObject;\n\nexport type GeminiModelConfig = FromSchema<typeof GeminiModelConfigSchema>;\n",
+    "/**\n * @license\n * Copyright 2025 Steven Roussey <sroussey@gmail.com>\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { globalServiceRegistry, parentPort, WORKER_SERVER } from \"@workglow/util\";\nimport { GEMINI_STREAM_TASKS, GEMINI_TASKS } from \"./common/Gemini_JobRunFns\";\nimport { GoogleGeminiProvider } from \"./GoogleGeminiProvider\";\n\nexport function GEMINI_WORKER_JOBRUN_REGISTER() {\n  const workerServer = globalServiceRegistry.get(WORKER_SERVER);\n  new GoogleGeminiProvider(GEMINI_TASKS, GEMINI_STREAM_TASKS).registerOnWorkerServer(workerServer);\n  parentPort.postMessage({ type: \"ready\" });\n  console.log(\"GEMINI_WORKER_JOBRUN registered\");\n}\n",
+    "/**\n * @license\n * Copyright 2025 Steven Roussey <sroussey@gmail.com>\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { AiProvider, type AiProviderRunFn, type AiProviderStreamFn } from \"@workglow/ai\";\nimport { GOOGLE_GEMINI } from \"./common/Gemini_Constants\";\nimport type { GeminiModelConfig } from \"./common/Gemini_ModelSchema\";\n\n/**\n * AI provider for Google Gemini cloud models.\n *\n * Supports text generation, text embedding, text rewriting, and text summarization\n * via the Google Generative AI API using the `@google/generative-ai` SDK.\n *\n * Task run functions are injected via the constructor so that the SDK\n * is only imported where actually needed (inline mode, worker server), not on\n * the main thread in worker mode.\n *\n * @example\n * ```typescript\n * // Worker mode (main thread) -- lightweight, no SDK import:\n * await new GoogleGeminiProvider().register({\n *   mode: \"worker\",\n *   worker: new Worker(new URL(\"./worker_gemini.ts\", import.meta.url), { type: \"module\" }),\n * });\n *\n * // Inline mode -- caller provides the tasks:\n * import { GEMINI_TASKS } from \"@workglow/ai-provider/google-gemini\";\n * await new GoogleGeminiProvider(GEMINI_TASKS).register({ mode: \"inline\" });\n * ```\n */\nexport class GoogleGeminiProvider extends AiProvider<GeminiModelConfig> {\n  readonly name = GOOGLE_GEMINI;\n\n  readonly taskTypes = [\n    \"TextGenerationTask\",\n    \"TextEmbeddingTask\",\n    \"TextRewriterTask\",\n    \"TextSummaryTask\",\n  ] as const;\n\n  constructor(\n    tasks?: Record<string, AiProviderRunFn<any, any, GeminiModelConfig>>,\n    streamTasks?: Record<string, AiProviderStreamFn<any, any, GeminiModelConfig>>\n  ) {\n    super(tasks, streamTasks);\n  }\n}\n"
+  ],
+  "mappings": ";;;;;;;;;;;;;;;;;;;;AAMO,IAAM,gBAAgB;;ACgB7B,IAAI;AACJ,eAAe,aAAa,GAAG;AAAA,EAC7B,IAAI,CAAC,MAAM;AAAA,IACT,IAAI;AAAA,MACF,OAAO,MAAa;AAAA,MACpB,MAAM;AAAA,MACN,MAAM,IAAI,MACR,oGACF;AAAA;AAAA,EAEJ;AAAA,EACA,OAAO,KAAK;AAAA;AAGd,SAAS,SAAS,CAAC,OAA8C;AAAA,EAC/D,MAAM,SACJ,OAAO,iBAAiB,YACvB,OAAO,YAAY,cAChB,QAAQ,KAAK,kBAAkB,QAAQ,KAAK,iBAC5C;AAAA,EACN,IAAI,CAAC,QAAQ;AAAA,IACX,MAAM,IAAI,MACR,kHACF;AAAA,EACF;AAAA,EACA,OAAO;AAAA;AAGT,SAAS,YAAY,CAAC,OAA8C;AAAA,EAClE,MAAM,OAAO,OAAO,iBAAiB;AAAA,EACrC,IAAI,CAAC,MAAM;AAAA,IACT,MAAM,IAAI,MAAM,mDAAmD;AAAA,EACrE;AAAA,EACA,OAAO;AAAA;AAGF,IAAM,wBAIT,OAAO,OAAO,OAAO,iBAAiB,WAAW;AAAA,EACnD,gBAAgB,GAAG,iCAAiC;AAAA,EACpD,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,kBAAkB;AAAA,MAChB,iBAAiB,MAAM;AAAA,MACvB,aAAa,MAAM;AAAA,MACnB,MAAM,MAAM;AAAA,IACd;AAAA,EACF,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,gBAAgB;AAAA,IAC5C,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,OAAO,CAAC,EAAE,CAAC;AAAA,EAC9D,CAAC;AAAA,EAED,MAAM,OAAO,OAAO,SAAS,KAAK;AAAA,EAClC,gBAAgB,KAAK,kCAAkC;AAAA,EACvD,OAAO,EAAE,KAAK;AAAA;AAGT,IAAM,uBAIT,OAAO,OAAO,OAAO,iBAAiB,WAAW;AAAA,EACnD,gBAAgB,GAAG,gCAAgC;AAAA,EACnD,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,iBAAiB,MAAM,mBAAmB;AAAA,IAC9C,OAAO,aAAa,KAAK;AAAA,EAC3B,CAAC;AAAA,EAED,MAAM,WACH,OAAO,iBAAiB,uBAAqC;AAAA,EAEhE,IAAI,MAAM,QAAQ,MAAM,IAAI,GAAG;AAAA,IAC7B,MAAM,UAAS,MAAM,eAAe,mBAAmB;AAAA,MACrD,UAAU,MAAM,KAAK,IAAI,CAAC,OAAO;AAAA,QAC/B,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;AAAA,QAC9C;AAAA,MACF,EAAE;AAAA,IACJ,CAAC;AAAA,IACD,gBAAgB,KAAK,iCAAiC;AAAA,IACtD,OAAO;AAAA,MACL,QAAQ,QAAO,WAAW,IAAI,CAAC,MAAM,IAAI,aAAa,EAAE,MAAM,CAAC;AAAA,IACjE;AAAA,EACF;AAAA,EAEA,MAAM,SAAS,MAAM,eAAe,aAAa;AAAA,IAC/C,SAAS,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,KAAK,CAAC,EAAE;AAAA,IACvD;AAAA,EACF,CAAC;AAAA,EAED,gBAAgB,KAAK,iCAAiC;AAAA,EACtD,OAAO,EAAE,QAAQ,IAAI,aAAa,OAAO,UAAU,MAAM,EAAE;AAAA;AAGtD,IAAM,sBAIT,OAAO,OAAO,OAAO,iBAAiB,WAAW;AAAA,EACnD,gBAAgB,GAAG,gCAAgC;AAAA,EACnD,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,mBAAmB,MAAM;AAAA,EAC3B,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,gBAAgB;AAAA,IAC5C,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,KAAK,CAAC,EAAE,CAAC;AAAA,EAC5D,CAAC;AAAA,EAED,MAAM,OAAO,OAAO,SAAS,KAAK;AAAA,EAClC,gBAAgB,KAAK,iCAAiC;AAAA,EACtD,OAAO,EAAE,KAAK;AAAA;AAGT,IAAM,qBAIT,OAAO,OAAO,OAAO,iBAAiB,WAAW;AAAA,EACnD,gBAAgB,GAAG,oCAAoC;AAAA,EACvD,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,mBAAmB;AAAA,EACrB,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,gBAAgB;AAAA,IAC5C,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,KAAK,CAAC,EAAE,CAAC;AAAA,EAC5D,CAAC;AAAA,EAED,MAAM,OAAO,OAAO,SAAS,KAAK;AAAA,EAClC,gBAAgB,KAAK,qCAAqC;AAAA,EAC1D,OAAO,EAAE,KAAK;AAAA;AAOT,IAAM,+BAIT,gBAAgB,CAAC,OAAO,OAAO,QAA8D;AAAA,EAC/F,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,kBAAkB;AAAA,MAChB,iBAAiB,MAAM;AAAA,MACvB,aAAa,MAAM;AAAA,MACnB,MAAM,MAAM;AAAA,IACd;AAAA,EACF,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,sBAAsB;AAAA,IAClD,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,OAAO,CAAC,EAAE,CAAC;AAAA,EAC9D,CAAC;AAAA,EAED,iBAAiB,SAAS,OAAO,QAAQ;AAAA,IACvC,MAAM,OAAO,MAAM,KAAK;AAAA,IACxB,IAAI,MAAM;AAAA,MACR,MAAM,EAAE,MAAM,cAAc,WAAW,KAAK;AAAA,IAC9C;AAAA,EACF;AAAA,EACA,MAAM,EAAE,MAAM,UAAU,MAAM,CAAC,EAA8B;AAAA;AAGxD,IAAM,6BAIT,gBAAgB,CAAC,OAAO,OAAO,QAA4D;AAAA,EAC7F,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,mBAAmB,MAAM;AAAA,EAC3B,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,sBAAsB;AAAA,IAClD,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,KAAK,CAAC,EAAE,CAAC;AAAA,EAC5D,CAAC;AAAA,EAED,iBAAiB,SAAS,OAAO,QAAQ;AAAA,IACvC,MAAM,OAAO,MAAM,KAAK;AAAA,IACxB,IAAI,MAAM;AAAA,MACR,MAAM,EAAE,MAAM,cAAc,WAAW,KAAK;AAAA,IAC9C;AAAA,EACF;AAAA,EACA,MAAM,EAAE,MAAM,UAAU,MAAM,CAAC,EAA4B;AAAA;AAGtD,IAAM,4BAIT,gBAAgB,CAAC,OAAO,OAAO,QAA2D;AAAA,EAC5F,MAAM,qBAAqB,MAAM,cAAc;AAAA,EAC/C,MAAM,QAAQ,IAAI,mBAAmB,UAAU,KAAK,CAAC;AAAA,EACrD,MAAM,WAAW,MAAM,mBAAmB;AAAA,IACxC,OAAO,aAAa,KAAK;AAAA,IACzB,mBAAmB;AAAA,EACrB,CAAC;AAAA,EAED,MAAM,SAAS,MAAM,SAAS,sBAAsB;AAAA,IAClD,UAAU,CAAC,EAAE,MAAM,QAAQ,OAAO,CAAC,EAAE,MAAM,MAAM,KAAK,CAAC,EAAE,CAAC;AAAA,EAC5D,CAAC;AAAA,EAED,iBAAiB,SAAS,OAAO,QAAQ;AAAA,IACvC,MAAM,OAAO,MAAM,KAAK;AAAA,IACxB,IAAI,MAAM;AAAA,MACR,MAAM,EAAE,MAAM,cAAc,WAAW,KAAK;AAAA,IAC9C;AAAA,EACF;AAAA,EACA,MAAM,EAAE,MAAM,UAAU,MAAM,CAAC,EAA2B;AAAA;AAOrD,IAAM,eAA6E;AAAA,EACxF,oBAAoB;AAAA,EACpB,mBAAmB;AAAA,EACnB,kBAAkB;AAAA,EAClB,iBAAiB;AACnB;AAEO,IAAM,sBAGT;AAAA,EACF,oBAAoB;AAAA,EACpB,kBAAkB;AAAA,EAClB,iBAAiB;AACnB;;ACpQA;AAIO,IAAM,oBAAoB;AAAA,EAC/B,MAAM;AAAA,EACN,YAAY;AAAA,IACV,UAAU;AAAA,MACR,OAAO;AAAA,MACP,aAAa;AAAA,IACf;AAAA,IACA,iBAAiB;AAAA,MACf,MAAM;AAAA,MACN,aAAa;AAAA,MACb,YAAY;AAAA,QACV,YAAY;AAAA,UACV,MAAM;AAAA,UACN,aACE;AAAA,QACJ;AAAA,QACA,SAAS;AAAA,UACP,MAAM;AAAA,UACN,aAAa;AAAA,QACf;AAAA,QACA,qBAAqB;AAAA,UACnB,OAAO;AAAA,YACL,EAAE,MAAM,OAAO;AAAA,YACf;AAAA,cACE,MAAM;AAAA,cACN,MAAM;AAAA,gBACJ;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA;AAAA,cACF;AAAA,YACF;AAAA,UACF;AAAA,UACA,aAAa;AAAA,UACb,SAAS;AAAA,QACX;AAAA,MACF;AAAA,MACA,UAAU,CAAC,YAAY;AAAA,MACvB,sBAAsB;AAAA,IACxB;AAAA,EACF;AAAA,EACA,UAAU,CAAC,YAAY,iBAAiB;AAAA,EACxC,sBAAsB;AACxB;AAEO,IAAM,0BAA0B;AAAA,EACrC,MAAM;AAAA,EACN,YAAY;AAAA,OACP,kBAAkB;AAAA,OAClB,kBAAkB;AAAA,EACvB;AAAA,EACA,UAAU,CAAC,GAAG,kBAAkB,UAAU,GAAG,kBAAkB,QAAQ;AAAA,EACvE,sBAAsB;AACxB;AAIO,IAAM,0BAA0B;AAAA,EACrC,MAAM;AAAA,EACN,YAAY;AAAA,OACP,kBAAkB;AAAA,OAClB,kBAAkB;AAAA,EACvB;AAAA,EACA,UAAU,CAAC,GAAG,kBAAkB,UAAU,GAAG,kBAAkB,QAAQ;AAAA,EACvE,sBAAsB;AACxB;;ACtEA;;;ACAA;AA2BO,MAAM,6BAA6B,WAA8B;AAAA,EAC7D,OAAO;AAAA,EAEP,YAAY;AAAA,IACnB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAAA,EAEA,WAAW,CACT,OACA,aACA;AAAA,IACA,MAAM,OAAO,WAAW;AAAA;AAE5B;;;ADvCO,SAAS,6BAA6B,GAAG;AAAA,EAC9C,MAAM,eAAe,sBAAsB,IAAI,aAAa;AAAA,EAC5D,IAAI,qBAAqB,cAAc,mBAAmB,EAAE,uBAAuB,YAAY;AAAA,EAC/F,WAAW,YAAY,EAAE,MAAM,QAAQ,CAAC;AAAA,EACxC,QAAQ,IAAI,iCAAiC;AAAA;",
+  "debugId": "94FDE8B04713DF4C64756E2164756E21",
+  "names": []
+}

package/dist/hf-transformers/HFT_Worker.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Steven Roussey <sroussey@gmail.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare function HFT_WORKER_JOBRUN_REGISTER(): void;
+//# sourceMappingURL=HFT_Worker.d.ts.map

package/dist/hf-transformers/HFT_Worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HFT_Worker.d.ts","sourceRoot":"","sources":["../../src/hf-transformers/HFT_Worker.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,wBAAgB,0BAA0B,SAOzC"}

package/dist/hf-transformers/HuggingFaceTransformersProvider.d.ts

@@ -0,0 +1,41 @@
+/**
+ * @license
+ * Copyright 2025 Steven Roussey <sroussey@gmail.com>
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AiProvider, type AiProviderRegisterOptions, type AiProviderRunFn, type AiProviderStreamFn } from "@workglow/ai";
+import type { HfTransformersOnnxModelConfig } from "./common/HFT_ModelSchema";
+/**
+ * AI provider for HuggingFace Transformers ONNX models.
+ *
+ * Supports text, vision, and multimodal tasks via the @sroussey/transformers library.
+ *
+ * Task run functions are injected via the constructor so that the heavy
+ * `@sroussey/transformers` library is only imported where actually needed
+ * (inline mode, worker server), not on the main thread in worker mode.
+ *
+ * @example
+ * ```typescript
+ * // Worker mode (main thread) -- lightweight, no heavy imports:
+ * await new HuggingFaceTransformersProvider().register({
+ *   mode: "worker",
+ *   worker: new Worker(new URL("./worker_hft.ts", import.meta.url), { type: "module" }),
+ * });
+ *
+ * // Inline mode -- caller provides the tasks:
+ * import { HFT_TASKS } from "@workglow/ai-provider/hf-transformers";
+ * await new HuggingFaceTransformersProvider(HFT_TASKS).register({ mode: "inline" });
+ *
+ * // Worker side -- caller provides the tasks:
+ * import { HFT_TASKS } from "@workglow/ai-provider/hf-transformers";
+ * new HuggingFaceTransformersProvider(HFT_TASKS).registerOnWorkerServer(workerServer);
+ * ```
+ */
+export declare class HuggingFaceTransformersProvider extends AiProvider<HfTransformersOnnxModelConfig> {
+    readonly name = "HF_TRANSFORMERS_ONNX";
+    readonly taskTypes: readonly ["DownloadModelTask", "UnloadModelTask", "TextEmbeddingTask", "TextGenerationTask", "TextQuestionAnswerTask", "TextLanguageDetectionTask", "TextClassificationTask", "TextFillMaskTask", "TextNamedEntityRecognitionTask", "TextRewriterTask", "TextSummaryTask", "TextTranslationTask", "ImageSegmentationTask", "ImageToTextTask", "BackgroundRemovalTask", "ImageEmbeddingTask", "ImageClassificationTask", "ObjectDetectionTask"];
+    constructor(tasks?: Record<string, AiProviderRunFn<any, any, HfTransformersOnnxModelConfig>>, streamTasks?: Record<string, AiProviderStreamFn<any, any, HfTransformersOnnxModelConfig>>);
+    protected onInitialize(options: AiProviderRegisterOptions): Promise<void>;
+    dispose(): Promise<void>;
+}
+//# sourceMappingURL=HuggingFaceTransformersProvider.d.ts.map

package/dist/hf-transformers/HuggingFaceTransformersProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HuggingFaceTransformersProvider.d.ts","sourceRoot":"","sources":["../../src/hf-transformers/HuggingFaceTransformersProvider.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,UAAU,EACV,KAAK,yBAAyB,EAC9B,KAAK,eAAe,EACpB,KAAK,kBAAkB,EACxB,MAAM,cAAc,CAAC;AAEtB,OAAO,KAAK,EAAE,6BAA6B,EAAE,MAAM,0BAA0B,CAAC;AAE9E;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,qBAAa,+BAAgC,SAAQ,UAAU,CAAC,6BAA6B,CAAC;IAC5F,QAAQ,CAAC,IAAI,0BAAwB;IAErC,QAAQ,CAAC,SAAS,ibAmBP;gBAGT,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,6BAA6B,CAAC,CAAC,EAChF,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,kBAAkB,CAAC,GAAG,EAAE,GAAG,EAAE,6BAA6B,CAAC,CAAC;cAKlE,YAAY,CAAC,OAAO,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC;IAQzE,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAMxC"}