@workglow/tasks 0.0.52 → 0.0.54

package/dist/index.js

@@ -176,9 +176,9 @@ var inputSchema3 = {
       description: "The body of the request"
     },
     response_type: {
-      enum: ["json", "text", "blob", "arraybuffer"],
+      anyOf: [{ type: "null" }, { enum: ["json", "text", "blob", "arraybuffer"] }],
       title: "Response Type",
-      default: "json"
+      default: null
     },
     timeout: {
       type: "number",
@@ -281,16 +281,32 @@ class FetchJob extends Job {
       signal: context.signal
     }, async (progress) => await context.updateProgress(progress));
     if (response.ok) {
-      if (input.response_type === "json") {
+      let responseType = input.response_type;
+      if (!responseType) {
+        const contentType = response.headers.get("content-type") ?? "";
+        if (contentType.includes("application/json")) {
+          responseType = "json";
+        } else if (contentType.includes("text/")) {
+          responseType = "text";
+        } else if (contentType.includes("application/octet-stream")) {
+          responseType = "arraybuffer";
+        } else if (contentType.includes("application/pdf") || contentType.includes("image/") || contentType.includes("application/zip")) {
+          responseType = "blob";
+        } else {
+          responseType = "json";
+        }
+        input.response_type = responseType;
+      }
+      if (responseType === "json") {
         return { json: await response.json() };
-      } else if (input.response_type === "text") {
+      } else if (responseType === "text") {
         return { text: await response.text() };
-      } else if (input.response_type === "blob") {
+      } else if (responseType === "blob") {
         return { blob: await response.blob() };
-      } else if (input.response_type === "arraybuffer") {
+      } else if (responseType === "arraybuffer") {
         return { arraybuffer: await response.arrayBuffer() };
       }
-      throw new TaskInvalidInputError(`Invalid response type: ${input.response_type}`);
+      throw new TaskInvalidInputError(`Invalid response type: ${responseType}`);
     } else {
       if (response.status === 429 || response.status === 503 || response.headers.get("Retry-After")) {
         let retryDate;
@@ -319,12 +335,44 @@ class FetchTask extends JobQueueTask {
   static category = "Input";
   static title = "Fetch";
   static description = "Fetches data from a URL with progress tracking and automatic retry handling";
+  static hasDynamicSchemas = true;
   static inputSchema() {
     return inputSchema3;
   }
   static outputSchema() {
     return outputSchema3;
   }
+  outputSchema() {
+    const responseType = this.runInputData?.response_type ?? this.defaults?.response_type ?? null;
+    if (responseType === null || responseType === undefined) {
+      return this.constructor.outputSchema();
+    }
+    const staticSchema = this.constructor.outputSchema();
+    if (typeof staticSchema === "boolean") {
+      return staticSchema;
+    }
+    if (!staticSchema.properties) {
+      return staticSchema;
+    }
+    const properties = {};
+    if (responseType === "json" && staticSchema.properties.json) {
+      properties.json = staticSchema.properties.json;
+    } else if (responseType === "text" && staticSchema.properties.text) {
+      properties.text = staticSchema.properties.text;
+    } else if (responseType === "blob" && staticSchema.properties.blob) {
+      properties.blob = staticSchema.properties.blob;
+    } else if (responseType === "arraybuffer" && staticSchema.properties.arraybuffer) {
+      properties.arraybuffer = staticSchema.properties.arraybuffer;
+    }
+    if (Object.keys(properties).length === 0) {
+      return staticSchema;
+    }
+    return {
+      type: "object",
+      properties,
+      additionalProperties: false
+    };
+  }
   constructor(input = {}, config = {}) {
     config.queue = input?.queue ?? config.queue;
     if (config.queue === undefined) {
@@ -333,6 +381,21 @@ class FetchTask extends JobQueueTask {
     super(input, config);
     this.jobClass = FetchJob;
   }
+  setInput(input) {
+    if (!("response_type" in input)) {
+      super.setInput(input);
+      return;
+    }
+    const getCurrentResponseType = () => {
+      return this.runInputData?.response_type ?? this.defaults?.response_type ?? null;
+    };
+    const previousResponseType = getCurrentResponseType();
+    super.setInput(input);
+    const newResponseType = getCurrentResponseType();
+    if (previousResponseType !== newResponseType) {
+      this.emitSchemaChange();
+    }
+  }
   async getDefaultQueueName(input) {
     if (!input.url) {
       return `fetch:${this.type}`;
@@ -5044,4 +5107,4 @@ export {
   DebugLog
 };
 
-//# debugId=C038D61CDD0021E964756E2164756E21
+//# debugId=F96A652558032ADF64756E2164756E21