@workglow/ai 0.1.0 → 0.1.2

package/dist/bun.js

@@ -138,13 +138,16 @@ function setAiProviderRegistry(pr) {
 
 // src/job/AiJob.ts
 var DEFAULT_AI_TIMEOUT_MS = 120000;
-var LOCAL_LLAMACPP_DEFAULT_TIMEOUT_MS = 600000;
+var LOCAL_MODEL_DEFAULT_TIMEOUT_MS = 600000;
 function resolveAiJobTimeoutMs(aiProvider, explicitMs) {
   if (explicitMs !== undefined) {
     return explicitMs;
   }
   if (aiProvider === "LOCAL_LLAMACPP") {
-    return LOCAL_LLAMACPP_DEFAULT_TIMEOUT_MS;
+    return LOCAL_MODEL_DEFAULT_TIMEOUT_MS;
+  }
+  if (aiProvider === "HF_TRANSFORMERS_ONNX" || aiProvider.startsWith("HF_TRANSFORMERS_ONNX")) {
+    return LOCAL_MODEL_DEFAULT_TIMEOUT_MS;
   }
   return DEFAULT_AI_TIMEOUT_MS;
 }
@@ -549,6 +552,7 @@ class InMemoryModelRepository extends ModelRepository {
 import {
   createServiceToken,
   globalServiceRegistry as globalServiceRegistry2,
+  registerInputCompactor,
   registerInputResolver
 } from "@workglow/util";
 var MODEL_REPOSITORY = createServiceToken("model.repository");
@@ -570,6 +574,19 @@ async function resolveModelFromRegistry(id, format, registry) {
   return model;
 }
 registerInputResolver("model", resolveModelFromRegistry);
+registerInputCompactor("model", async (value, _format, registry) => {
+  if (typeof value === "object" && value !== null && "model_id" in value) {
+    const id = value.model_id;
+    if (typeof id !== "string")
+      return;
+    const modelRepo = registry.has(MODEL_REPOSITORY) ? registry.get(MODEL_REPOSITORY) : getGlobalModelRepository();
+    const model = await modelRepo.findByName(id);
+    if (!model)
+      return;
+    return id;
+  }
+  return;
+});
 
 // src/provider/AiProvider.ts
 import {
@@ -899,6 +916,10 @@ class AiTask extends Task {
       aiProvider: model.provider,
       taskInput: input
     };
+    const taskTimeoutMs = this.config.timeout;
+    if (typeof taskTimeoutMs === "number" && taskTimeoutMs > 0) {
+      jobInput.timeoutMs = taskTimeoutMs;
+    }
     const inputOutputSchema = input.outputSchema;
     if (inputOutputSchema && typeof inputOutputSchema === "object" && !Array.isArray(inputOutputSchema) && typeof inputOutputSchema.type === "string") {
       jobInput.outputSchema = inputOutputSchema;
@@ -6534,4 +6555,4 @@ export {
   AiJob
 };
 
-//# debugId=B0C9F29E92462AAF64756E2164756E21
+//# debugId=B891FF379EE6413B64756E2164756E21