@workglow/ai 0.1.0 → 0.1.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (56) hide show
  1. package/dist/browser.js +10 -3
  2. package/dist/browser.js.map +49 -49
  3. package/dist/bun.js +10 -3
  4. package/dist/bun.js.map +49 -49
  5. package/dist/job/AiJob.d.ts.map +1 -1
  6. package/dist/node.js +10 -3
  7. package/dist/node.js.map +49 -49
  8. package/dist/task/BackgroundRemovalTask.d.ts.map +1 -1
  9. package/dist/task/ChunkRetrievalTask.d.ts.map +1 -1
  10. package/dist/task/ChunkToVectorTask.d.ts.map +1 -1
  11. package/dist/task/ChunkVectorHybridSearchTask.d.ts.map +1 -1
  12. package/dist/task/ChunkVectorSearchTask.d.ts.map +1 -1
  13. package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
  14. package/dist/task/ContextBuilderTask.d.ts.map +1 -1
  15. package/dist/task/CountTokensTask.d.ts.map +1 -1
  16. package/dist/task/DocumentEnricherTask.d.ts.map +1 -1
  17. package/dist/task/DownloadModelTask.d.ts.map +1 -1
  18. package/dist/task/FaceDetectorTask.d.ts.map +1 -1
  19. package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
  20. package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
  21. package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
  22. package/dist/task/HierarchicalChunkerTask.d.ts.map +1 -1
  23. package/dist/task/HierarchyJoinTask.d.ts.map +1 -1
  24. package/dist/task/ImageClassificationTask.d.ts.map +1 -1
  25. package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
  26. package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
  27. package/dist/task/ImageToTextTask.d.ts.map +1 -1
  28. package/dist/task/ModelInfoTask.d.ts.map +1 -1
  29. package/dist/task/ModelSearchTask.d.ts.map +1 -1
  30. package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
  31. package/dist/task/PoseLandmarkerTask.d.ts.map +1 -1
  32. package/dist/task/QueryExpanderTask.d.ts.map +1 -1
  33. package/dist/task/RerankerTask.d.ts.map +1 -1
  34. package/dist/task/StructuralParserTask.d.ts.map +1 -1
  35. package/dist/task/StructuredGenerationTask.d.ts.map +1 -1
  36. package/dist/task/TextChunkerTask.d.ts.map +1 -1
  37. package/dist/task/TextClassificationTask.d.ts.map +1 -1
  38. package/dist/task/TextEmbeddingTask.d.ts.map +1 -1
  39. package/dist/task/TextFillMaskTask.d.ts.map +1 -1
  40. package/dist/task/TextGenerationTask.d.ts.map +1 -1
  41. package/dist/task/TextLanguageDetectionTask.d.ts.map +1 -1
  42. package/dist/task/TextNamedEntityRecognitionTask.d.ts.map +1 -1
  43. package/dist/task/TextQuestionAnswerTask.d.ts.map +1 -1
  44. package/dist/task/TextRewriterTask.d.ts.map +1 -1
  45. package/dist/task/TextSummaryTask.d.ts.map +1 -1
  46. package/dist/task/TextTranslationTask.d.ts.map +1 -1
  47. package/dist/task/TopicSegmenterTask.d.ts.map +1 -1
  48. package/dist/task/UnloadModelTask.d.ts.map +1 -1
  49. package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
  50. package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
  51. package/dist/task/base/AiTask.d.ts.map +1 -1
  52. package/dist/task/base/AiVisionTask.d.ts.map +1 -1
  53. package/dist/task/base/StreamingAiTask.d.ts.map +1 -1
  54. package/dist/worker.js +6 -3
  55. package/dist/worker.js.map +3 -3
  56. package/package.json +11 -11
package/dist/bun.js CHANGED
@@ -138,13 +138,16 @@ function setAiProviderRegistry(pr) {
138
138
 
139
139
  // src/job/AiJob.ts
140
140
  var DEFAULT_AI_TIMEOUT_MS = 120000;
141
- var LOCAL_LLAMACPP_DEFAULT_TIMEOUT_MS = 600000;
141
+ var LOCAL_MODEL_DEFAULT_TIMEOUT_MS = 600000;
142
142
  function resolveAiJobTimeoutMs(aiProvider, explicitMs) {
143
143
  if (explicitMs !== undefined) {
144
144
  return explicitMs;
145
145
  }
146
146
  if (aiProvider === "LOCAL_LLAMACPP") {
147
- return LOCAL_LLAMACPP_DEFAULT_TIMEOUT_MS;
147
+ return LOCAL_MODEL_DEFAULT_TIMEOUT_MS;
148
+ }
149
+ if (aiProvider === "HF_TRANSFORMERS_ONNX" || aiProvider.startsWith("HF_TRANSFORMERS_ONNX")) {
150
+ return LOCAL_MODEL_DEFAULT_TIMEOUT_MS;
148
151
  }
149
152
  return DEFAULT_AI_TIMEOUT_MS;
150
153
  }
@@ -899,6 +902,10 @@ class AiTask extends Task {
899
902
  aiProvider: model.provider,
900
903
  taskInput: input
901
904
  };
905
+ const taskTimeoutMs = this.config.timeout;
906
+ if (typeof taskTimeoutMs === "number" && taskTimeoutMs > 0) {
907
+ jobInput.timeoutMs = taskTimeoutMs;
908
+ }
902
909
  const inputOutputSchema = input.outputSchema;
903
910
  if (inputOutputSchema && typeof inputOutputSchema === "object" && !Array.isArray(inputOutputSchema) && typeof inputOutputSchema.type === "string") {
904
911
  jobInput.outputSchema = inputOutputSchema;
@@ -6534,4 +6541,4 @@ export {
6534
6541
  AiJob
6535
6542
  };
6536
6543
 
6537
- //# debugId=B0C9F29E92462AAF64756E2164756E21
6544
+ //# debugId=833980E5CCD244A264756E2164756E21