@workglow/openai 0.2.33 → 0.2.35

package/dist/ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,cAAc,YAAY,CAAC"}
+{"version":3,"file":"ai.d.ts","sourceRoot":"","sources":["../src/ai.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,cAAc,YAAY,CAAC"}

package/dist/ai.js

@@ -8,7 +8,6 @@ var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require
 
 // src/ai/common/OpenAI_Constants.ts
 var OPENAI = "OPENAI";
-
 // src/ai/common/OpenAI_ImageValidation.ts
 import { AiImageOutputTask, ProviderUnsupportedFeatureError } from "@workglow/ai";
 function registerOpenAiImageValidator() {
@@ -22,7 +21,6 @@ function registerOpenAiImageValidator() {
     }
   });
 }
-
 // src/ai/common/OpenAI_ModelSchema.ts
 import { ModelConfigSchema, ModelRecordSchema } from "@workglow/ai/worker";
 var OpenAiModelSchema = {
@@ -81,7 +79,6 @@ var OpenAiModelConfigSchema = {
   required: [...ModelConfigSchema.required, ...OpenAiModelSchema.required],
   additionalProperties: false
 };
-
 // src/ai/common/OpenAI_ModelSearch.ts
 import { filterLabeledModelsByQuery } from "@workglow/ai/provider-utils";
 
@@ -139,8 +136,8 @@ var OPENAI_FALLBACK = [
   { label: "o1-mini", value: "o1-mini" }
 ];
 var OPENAI_IMAGE_MODELS = [
-  { value: "gpt-image-2", tasks: ["ImageGenerateTask", "ImageEditTask"] },
-  { value: "dall-e-3", tasks: ["ImageGenerateTask"] }
+  { value: "gpt-image-2", capabilities: ["image.generation", "image.editing"] },
+  { value: "dall-e-3", capabilities: ["image.generation"] }
 ];
 async function listOpenAiModels(credentialKey) {
   const client = await getClient({
@@ -172,7 +169,7 @@ function mapModelList(models) {
         provider: OPENAI,
         title: m.value,
         description: "",
-        tasks: imageEntry?.tasks ?? [],
+        capabilities: imageEntry?.capabilities ?? [],
         provider_config: { model_name: m.value },
         metadata: {}
       },
@@ -180,7 +177,7 @@ function mapModelList(models) {
     };
   });
 }
-var OpenAI_ModelSearch = async (input) => {
+var OpenAI_ModelSearch_Stream = async (input, _model, _signal, emit) => {
   let models;
   if (!input.credential_key) {
     models = OPENAI_FALLBACK;
@@ -188,34 +185,91 @@ var OpenAI_ModelSearch = async (input) => {
     models = await listOpenAiModels(input.credential_key);
   }
   models = filterLabeledModelsByQuery(models, input.query);
-  return { results: mapModelList(models) };
+  emit({ type: "finish", data: { results: mapModelList(models) } });
 };
-
 // src/ai/registerOpenAi.ts
 import { registerProviderWithWorker } from "@workglow/ai/provider-utils";
 
 // src/ai/OpenAiQueuedProvider.ts
 import { AiProvider } from "@workglow/ai";
 import { createCloudProviderClass } from "@workglow/ai/provider-utils";
-var OPENAI_QUEUED_TASK_TYPES = [
-  "TextGenerationTask",
-  "TextEmbeddingTask",
-  "TextRewriterTask",
-  "TextSummaryTask",
-  "CountTokensTask",
-  "ModelInfoTask",
-  "StructuredGenerationTask",
-  "ToolCallingTask",
-  "ModelSearchTask",
-  "ImageGenerateTask",
-  "ImageEditTask"
+
+// src/ai/common/OpenAI_CapabilitySets.ts
+var OPENAI_TEXT_GENERATION = ["text.generation"];
+var OPENAI_TOOL_USE = ["text.generation", "tool-use"];
+var OPENAI_JSON_MODE = ["text.generation", "json-mode"];
+var OPENAI_TEXT_REWRITER = ["text.rewriter"];
+var OPENAI_TEXT_SUMMARY = ["text.summary"];
+var OPENAI_TEXT_EMBEDDING = ["text.embedding"];
+var OPENAI_IMAGE_GENERATION = ["image.generation"];
+var OPENAI_IMAGE_EDITING = ["image.editing"];
+var OPENAI_COUNT_TOKENS = ["model.count-tokens"];
+var OPENAI_MODEL_SEARCH = ["model.search"];
+var OPENAI_MODEL_INFO = ["model.info"];
+var OPENAI_CAPABILITY_SETS = [
+  OPENAI_TEXT_GENERATION,
+  OPENAI_TOOL_USE,
+  OPENAI_JSON_MODE,
+  OPENAI_TEXT_REWRITER,
+  OPENAI_TEXT_SUMMARY,
+  OPENAI_TEXT_EMBEDDING,
+  OPENAI_IMAGE_GENERATION,
+  OPENAI_IMAGE_EDITING,
+  OPENAI_COUNT_TOKENS,
+  OPENAI_MODEL_SEARCH,
+  OPENAI_MODEL_INFO
 ];
 
+// src/ai/common/OpenAI_Capabilities.ts
+var OPENAI_RUN_FN_SPECS = OPENAI_CAPABILITY_SETS.map((serves) => ({ serves }));
+function openAiWorkerRunFnSpecs() {
+  return OPENAI_RUN_FN_SPECS;
+}
+function inferOpenAiCapabilities(model) {
+  const id = String(model.model_id ?? model.provider_config?.model_name ?? "");
+  if (/^text-embedding/i.test(id)) {
+    return ["text.embedding", "model.count-tokens", "model.info", "model.search"];
+  }
+  if (/^dall-e/i.test(id)) {
+    return ["image.generation", "model.info", "model.search"];
+  }
+  if (/^gpt-image/i.test(id)) {
+    return ["image.generation", "image.editing", "model.info", "model.search"];
+  }
+  if (/^gpt-/i.test(id) || /^o\d/i.test(id)) {
+    const caps = [
+      "text.generation",
+      "text.rewriter",
+      "text.summary",
+      "tool-use",
+      "json-mode",
+      "model.count-tokens",
+      "model.info",
+      "model.search"
+    ];
+    const supportsVision = /gpt-4o|gpt-4\.1|gpt-5|gpt-4-vision|gpt-4-turbo/i.test(id) || /^o\d/i.test(id);
+    if (supportsVision) {
+      caps.push("vision-input");
+    }
+    return caps;
+  }
+  const declared = model.capabilities ?? [];
+  if (declared.length > 0)
+    return declared;
+  return ["model.search", "model.info"];
+}
+
+// src/ai/OpenAiQueuedProvider.ts
 class OpenAiQueuedProvider extends createCloudProviderClass(AiProvider, {
   name: OPENAI,
-  displayName: "OpenAI",
-  taskTypes: OPENAI_QUEUED_TASK_TYPES
+  displayName: "OpenAI"
 }) {
+  inferCapabilities(model) {
+    return inferOpenAiCapabilities(model);
+  }
+  workerRunFnSpecs() {
+    return openAiWorkerRunFnSpecs();
+  }
 }
 
 // src/ai/registerOpenAi.ts
@@ -223,14 +277,483 @@ async function registerOpenAi(options) {
   registerOpenAiImageValidator();
   await registerProviderWithWorker(new OpenAiQueuedProvider, "OpenAI", options);
 }
+// src/ai/common/OpenAI_CountTokens.ts
+var _tiktoken;
+async function loadTiktoken() {
+  if (!_tiktoken) {
+    try {
+      _tiktoken = await import("tiktoken");
+    } catch {
+      throw new Error("tiktoken is required for OpenAI token counting. Install it with: bun add tiktoken");
+    }
+  }
+  return _tiktoken;
+}
+var _encoderCache = new Map;
+async function getEncoder(modelName) {
+  const tiktoken = await loadTiktoken();
+  if (!_encoderCache.has(modelName)) {
+    try {
+      _encoderCache.set(modelName, tiktoken.encoding_for_model(modelName));
+    } catch {
+      const fallback = "cl100k_base";
+      if (!_encoderCache.has(fallback)) {
+        _encoderCache.set(fallback, tiktoken.get_encoding(fallback));
+      }
+      _encoderCache.set(modelName, _encoderCache.get(fallback));
+    }
+  }
+  return _encoderCache.get(modelName);
+}
+async function countTokens(input, model) {
+  const enc = await getEncoder(getModelName(model));
+  const tokens = enc.encode(input.text);
+  return { count: tokens.length };
+}
+var OpenAI_CountTokens_Stream = async (input, model, _signal, emit) => {
+  const result = await countTokens(input, model);
+  emit({ type: "finish", data: result });
+};
+var OpenAI_CountTokens_Preview = async (input, model) => {
+  return countTokens(input, model);
+};
+
+// src/ai/common/OpenAI_ImageEdit.ts
+import { ImageGenerationContentPolicyError, ImageGenerationProviderError } from "@workglow/ai";
+import { dataUriToImageValue, imageValueToPngBytes } from "@workglow/ai/provider-utils";
+function aspectRatioToSize(aspectRatio) {
+  switch (aspectRatio) {
+    case "16:9":
+    case "4:3":
+      return "1536x1024";
+    case "9:16":
+    case "3:4":
+      return "1024x1536";
+    case "1:1":
+    default:
+      return "1024x1024";
+  }
+}
+function modelIdOf(model) {
+  return model?.model_id ?? model?.provider_config?.model_name ?? "openai";
+}
+async function gpuImageToOpenAiFile(image, name) {
+  const bytes = await imageValueToPngBytes(image);
+  const buffer = bytes.buffer instanceof ArrayBuffer ? bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength) : new Uint8Array(bytes).buffer;
+  const sdk = await import("openai");
+  if (typeof sdk.toFile === "function") {
+    return sdk.toFile(new Blob([buffer], { type: "image/png" }), name, { type: "image/png" });
+  }
+  return new File([buffer], name, { type: "image/png" });
+}
+async function decodeB64Png(b64) {
+  return dataUriToImageValue(`data:image/png;base64,${b64}`);
+}
+async function buildEditPayload(input, model) {
+  const modelName = getModelName(model);
+  const primary = await gpuImageToOpenAiFile(input.image, "image.png");
+  const additionalFiles = input.additionalImages && input.additionalImages.length > 0 ? await Promise.all(input.additionalImages.map((g, i) => gpuImageToOpenAiFile(g, `image-${i + 1}.png`))) : [];
+  const imageField = additionalFiles.length === 0 ? primary : [primary, ...additionalFiles];
+  const payload = {
+    model: modelName,
+    prompt: input.prompt,
+    image: imageField,
+    size: aspectRatioToSize(input.aspectRatio),
+    quality: input.quality,
+    n: 1,
+    ...input.providerOptions ?? {}
+  };
+  if (input.mask) {
+    payload.mask = await gpuImageToOpenAiFile(input.mask, "mask.png");
+  }
+  return payload;
+}
+var OpenAI_ImageEdit_Stream = async (input, model, signal, emit) => {
+  const client = await getClient(model);
+  try {
+    const payload = await buildEditPayload(input, model);
+    const stream = await client.images.edit({ ...payload, stream: true, partial_images: 3 }, { signal });
+    for await (const event of stream) {
+      if (signal.aborted)
+        return;
+      const b64 = event.b64_json;
+      if (!b64)
+        continue;
+      const image = await decodeB64Png(b64);
+      emit({ type: "snapshot", data: { image } });
+    }
+    emit({ type: "finish", data: {} });
+  } catch (err) {
+    if (err instanceof ImageGenerationProviderError || err instanceof ImageGenerationContentPolicyError) {
+      throw err;
+    }
+    const msg = err instanceof Error ? err.message : "unknown error";
+    if (/safety|policy|moderation/i.test(msg)) {
+      throw new ImageGenerationContentPolicyError(modelIdOf(model), msg);
+    }
+    throw new ImageGenerationProviderError(modelIdOf(model), msg, { cause: err });
+  }
+};
+
+// src/ai/common/OpenAI_ImageGenerate.ts
+import { ImageGenerationContentPolicyError as ImageGenerationContentPolicyError2, ImageGenerationProviderError as ImageGenerationProviderError2 } from "@workglow/ai";
+import { dataUriToImageValue as dataUriToImageValue2 } from "@workglow/ai/provider-utils";
+function aspectRatioToSize2(aspectRatio) {
+  switch (aspectRatio) {
+    case "16:9":
+    case "4:3":
+      return "1536x1024";
+    case "9:16":
+    case "3:4":
+      return "1024x1536";
+    case "1:1":
+    default:
+      return "1024x1024";
+  }
+}
+async function decodeB64Png2(b64) {
+  return dataUriToImageValue2(`data:image/png;base64,${b64}`);
+}
+function modelIdOf2(model) {
+  return model?.model_id ?? model?.provider_config?.model_name ?? "openai";
+}
+var OpenAI_ImageGenerate_Stream = async (input, model, signal, emit) => {
+  const client = await getClient(model);
+  const modelName = getModelName(model);
+  const size = aspectRatioToSize2(input.aspectRatio);
+  if (modelName.startsWith("dall-e")) {
+    try {
+      const resp = await client.images.generate({
+        model: modelName,
+        prompt: input.prompt,
+        size,
+        quality: input.quality,
+        n: 1,
+        response_format: "b64_json",
+        ...input.providerOptions ?? {}
+      }, { signal });
+      const b64 = resp.data?.[0]?.b64_json;
+      if (!b64) {
+        throw new ImageGenerationProviderError2(modelIdOf2(model), "Empty response (no b64_json)");
+      }
+      const image = await decodeB64Png2(b64);
+      emit({ type: "snapshot", data: { image } });
+      emit({ type: "finish", data: {} });
+      return;
+    } catch (err) {
+      if (err instanceof ImageGenerationProviderError2 || err instanceof ImageGenerationContentPolicyError2) {
+        throw err;
+      }
+      const msg = err instanceof Error ? err.message : "unknown error";
+      if (/safety|policy|moderation/i.test(msg)) {
+        throw new ImageGenerationContentPolicyError2(modelIdOf2(model), msg);
+      }
+      throw new ImageGenerationProviderError2(modelIdOf2(model), msg, { cause: err });
+    }
+  }
+  try {
+    const stream = await client.images.generate({
+      model: modelName,
+      prompt: input.prompt,
+      size,
+      quality: input.quality,
+      n: 1,
+      stream: true,
+      partial_images: 3,
+      ...input.providerOptions ?? {}
+    }, { signal });
+    for await (const event of stream) {
+      if (signal.aborted)
+        return;
+      const b64 = event.b64_json;
+      if (!b64)
+        continue;
+      const image = await decodeB64Png2(b64);
+      emit({ type: "snapshot", data: { image } });
+    }
+    emit({ type: "finish", data: {} });
+  } catch (err) {
+    if (err instanceof ImageGenerationProviderError2 || err instanceof ImageGenerationContentPolicyError2) {
+      throw err;
+    }
+    const msg = err instanceof Error ? err.message : "unknown error";
+    if (/safety|policy|moderation/i.test(msg)) {
+      throw new ImageGenerationContentPolicyError2(modelIdOf2(model), msg);
+    }
+    throw new ImageGenerationProviderError2(modelIdOf2(model), msg, { cause: err });
+  }
+};
+
+// src/ai/common/OpenAI_ModelInfo.ts
+var OPENAI_EMBEDDING_DIMENSIONS = {
+  "text-embedding-3-small": { native_dimensions: 1536, mrl: true },
+  "text-embedding-3-large": { native_dimensions: 3072, mrl: true },
+  "text-embedding-ada-002": { native_dimensions: 1536, mrl: false }
+};
+var OpenAI_ModelInfo_Stream = async (input, model, _signal, emit) => {
+  if (input.detail === "dimensions") {
+    const pc = model?.provider_config;
+    let native_dimensions = typeof pc?.native_dimensions === "number" ? pc.native_dimensions : undefined;
+    let mrl = typeof pc?.mrl === "boolean" ? pc.mrl : undefined;
+    if (native_dimensions === undefined) {
+      const modelName = pc?.model_name ?? "";
+      const known = OPENAI_EMBEDDING_DIMENSIONS[modelName];
+      if (known) {
+        native_dimensions = known.native_dimensions;
+        mrl = mrl ?? known.mrl;
+      }
+    }
+    emit({
+      type: "finish",
+      data: {
+        model: input.model,
+        is_local: false,
+        is_remote: true,
+        supports_browser: true,
+        supports_node: true,
+        is_cached: false,
+        is_loaded: false,
+        file_sizes: null,
+        ...native_dimensions !== undefined ? { native_dimensions } : {},
+        ...mrl !== undefined ? { mrl } : {}
+      }
+    });
+    return;
+  }
+  emit({
+    type: "finish",
+    data: {
+      model: input.model,
+      is_local: false,
+      is_remote: true,
+      supports_browser: true,
+      supports_node: true,
+      is_cached: false,
+      is_loaded: false,
+      file_sizes: null
+    }
+  });
+};
+
+// src/ai/common/OpenAI_StructuredGeneration.ts
+import { parsePartialJson } from "@workglow/util/worker";
+var OpenAI_StructuredGeneration_Stream = async (input, model, signal, emit, outputSchema) => {
+  const client = await getClient(model);
+  const modelName = getModelName(model);
+  const schema = input.outputSchema ?? outputSchema;
+  const stream = await client.chat.completions.create({
+    model: modelName,
+    messages: [{ role: "user", content: input.prompt }],
+    response_format: {
+      type: "json_schema",
+      json_schema: {
+        name: "structured_output",
+        schema,
+        strict: true
+      }
+    },
+    max_completion_tokens: input.maxTokens,
+    temperature: input.temperature,
+    stream: true
+  }, { signal });
+  let accumulatedJson = "";
+  for await (const chunk of stream) {
+    const delta = chunk.choices[0]?.delta?.content ?? "";
+    if (delta) {
+      accumulatedJson += delta;
+      const partial = parsePartialJson(accumulatedJson);
+      if (partial !== undefined) {
+        emit({ type: "object-delta", port: "object", objectDelta: partial });
+      }
+    }
+  }
+  let finalObject;
+  try {
+    finalObject = JSON.parse(accumulatedJson);
+  } catch {
+    finalObject = parsePartialJson(accumulatedJson) ?? {};
+  }
+  emit({ type: "finish", data: { object: finalObject } });
+};
+
+// src/ai/common/OpenAI_TextEmbedding.ts
+import { getLogger } from "@workglow/util/worker";
+var OpenAI_TextEmbedding_Stream = async (input, model, signal, emit) => {
+  const logger = getLogger();
+  const timerLabel = `openai:TextEmbedding:${getModelName(model)}`;
+  logger.time(timerLabel, { model: getModelName(model) });
+  try {
+    const client = await getClient(model);
+    const modelName = getModelName(model);
+    const response = await client.embeddings.create({
+      model: modelName,
+      input: input.text
+    }, { signal });
+    const result = Array.isArray(input.text) ? {
+      vector: response.data.map((item) => new Float32Array(item.embedding))
+    } : { vector: new Float32Array(response.data[0].embedding) };
+    emit({ type: "finish", data: result });
+  } finally {
+    logger.timeEnd(timerLabel, { model: getModelName(model) });
+  }
+};
+
+// src/ai/common/OpenAI_TextGeneration.ts
+import { toOpenAIMessages } from "@workglow/ai/worker";
+import { getLogger as getLogger2 } from "@workglow/util/worker";
+function buildChatParams(input, model) {
+  const hasMessages = Array.isArray(input.messages) && input.messages.length > 0;
+  const messages = hasMessages ? toOpenAIMessages({
+    messages: input.messages,
+    systemPrompt: input.systemPrompt,
+    prompt: "",
+    tools: []
+  }) : [{ role: "user", content: input.prompt }];
+  const params = {
+    model: getModelName(model),
+    messages
+  };
+  if (input.maxTokens !== undefined)
+    params.max_completion_tokens = input.maxTokens;
+  if (input.temperature !== undefined)
+    params.temperature = input.temperature;
+  if (input.topP !== undefined)
+    params.top_p = input.topP;
+  if (input.frequencyPenalty !== undefined)
+    params.frequency_penalty = input.frequencyPenalty;
+  if (input.presencePenalty !== undefined)
+    params.presence_penalty = input.presencePenalty;
+  return params;
+}
+var OpenAI_TextGeneration_Stream = async (input, model, signal, emit) => {
+  const logger = getLogger2();
+  const timerLabel = `openai:TextGeneration:${getModelName(model)}`;
+  logger.time(timerLabel, { model: getModelName(model) });
+  try {
+    const client = await getClient(model);
+    const params = buildChatParams(input, model);
+    const stream = await client.chat.completions.create({ ...params, stream: true }, { signal });
+    for await (const chunk of stream) {
+      const delta = chunk.choices?.[0]?.delta?.content ?? "";
+      if (delta) {
+        emit({ type: "text-delta", port: "text", textDelta: delta });
+      }
+    }
+    emit({ type: "finish", data: {} });
+  } finally {
+    logger.timeEnd(timerLabel, { model: getModelName(model) });
+  }
+};
+
+// src/ai/common/OpenAI_TextRewriter.ts
+var OpenAI_TextRewriter_Stream = async (input, model, signal, emit) => {
+  const client = await getClient(model);
+  const modelName = getModelName(model);
+  const stream = await client.chat.completions.create({
+    model: modelName,
+    messages: [
+      { role: "system", content: input.prompt },
+      { role: "user", content: input.text }
+    ],
+    stream: true
+  }, { signal });
+  for await (const chunk of stream) {
+    const delta = chunk.choices[0]?.delta?.content ?? "";
+    if (delta) {
+      emit({ type: "text-delta", port: "text", textDelta: delta });
+    }
+  }
+  emit({ type: "finish", data: {} });
+};
+
+// src/ai/common/OpenAI_TextSummary.ts
+var OpenAI_TextSummary_Stream = async (input, model, signal, emit) => {
+  const client = await getClient(model);
+  const modelName = getModelName(model);
+  const stream = await client.chat.completions.create({
+    model: modelName,
+    messages: [
+      { role: "system", content: "Summarize the following text concisely." },
+      { role: "user", content: input.text }
+    ],
+    stream: true
+  }, { signal });
+  for await (const chunk of stream) {
+    const delta = chunk.choices[0]?.delta?.content ?? "";
+    if (delta) {
+      emit({ type: "text-delta", port: "text", textDelta: delta });
+    }
+  }
+  emit({ type: "finish", data: {} });
+};
+
+// src/ai/common/OpenAI_ToolCalling.ts
+import { filterValidToolCalls, toOpenAIMessages as toOpenAIMessages2 } from "@workglow/ai/worker";
+import {
+  accumulateOpenAIStream,
+  buildOpenAITools,
+  mapOpenAIToolChoice
+} from "@workglow/ai/provider-utils";
+var OpenAI_ToolCalling_Stream = async (input, model, signal, emit) => {
+  const client = await getClient(model);
+  const modelName = getModelName(model);
+  const tools = buildOpenAITools(input.tools);
+  const messages = toOpenAIMessages2(input);
+  const toolChoice = mapOpenAIToolChoice(input.toolChoice, true);
+  const stream = await client.chat.completions.create({
+    model: modelName,
+    messages,
+    max_completion_tokens: input.maxTokens,
+    temperature: input.temperature,
+    stream: true,
+    tools,
+    tool_choice: toolChoice
+  }, { signal });
+  for await (const event of accumulateOpenAIStream(stream)) {
+    if (event.type === "object-delta" && event.port === "toolCalls") {
+      const validated = filterValidToolCalls(event.objectDelta, input.tools);
+      if (validated.length > 0) {
+        emit({ type: "object-delta", port: "toolCalls", objectDelta: validated });
+      }
+      continue;
+    }
+    emit(event);
+  }
+};
+
+// src/ai/common/OpenAI_JobRunFns.ts
+var OPENAI_RUN_FNS = [
+  { serves: OPENAI_TEXT_GENERATION, runFn: OpenAI_TextGeneration_Stream },
+  { serves: OPENAI_TOOL_USE, runFn: OpenAI_ToolCalling_Stream },
+  { serves: OPENAI_JSON_MODE, runFn: OpenAI_StructuredGeneration_Stream },
+  { serves: OPENAI_TEXT_REWRITER, runFn: OpenAI_TextRewriter_Stream },
+  { serves: OPENAI_TEXT_SUMMARY, runFn: OpenAI_TextSummary_Stream },
+  { serves: OPENAI_TEXT_EMBEDDING, runFn: OpenAI_TextEmbedding_Stream },
+  { serves: OPENAI_IMAGE_GENERATION, runFn: OpenAI_ImageGenerate_Stream },
+  { serves: OPENAI_IMAGE_EDITING, runFn: OpenAI_ImageEdit_Stream },
+  { serves: OPENAI_COUNT_TOKENS, runFn: OpenAI_CountTokens_Stream },
+  { serves: OPENAI_MODEL_SEARCH, runFn: OpenAI_ModelSearch_Stream },
+  { serves: OPENAI_MODEL_INFO, runFn: OpenAI_ModelInfo_Stream }
+];
+var OPENAI_PREVIEW_TASKS = {
+  CountTokensTask: OpenAI_CountTokens_Preview
+};
+
+// src/ai/index.ts
+var _testOnly = {
+  OpenAiQueuedProvider,
+  OPENAI_RUN_FN_SPECS,
+  OPENAI_RUN_FNS
+};
 export {
   registerOpenAiImageValidator,
   registerOpenAi,
+  _testOnly,
   OpenAiModelSchema,
   OpenAiModelRecordSchema,
   OpenAiModelConfigSchema,
-  OpenAI_ModelSearch,
+  OpenAI_ModelSearch_Stream,
   OPENAI
 };
 
-//# debugId=FF32F432A0E2ACB064756E2164756E21
+//# debugId=F363500CFFD8B36764756E2164756E21