@workglow/ai 0.1.0 → 0.1.1

package/dist/browser.js

@@ -137,13 +137,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;
 }
@@ -898,6 +901,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;
@@ -6533,4 +6540,4 @@ export {
   AiJob
 };
 
-//# debugId=CEEB72C6AA2136B964756E2164756E21
+//# debugId=402C8B26C7EE876D64756E2164756E21