[Components] fal_ai - New action components

Author: jcortes

components/fal_ai/actions/add-request-to-queue/add-request-to-queue.mjs

@@ -0,0 +1,63 @@
+import app from "../../fal_ai.app.mjs";
+
+export default {
+  key: "fal_ai-add-request-to-queue",
+  name: "Add Request to Queue",
+  description: "Adds a request to the queue for asynchronous processing, including specifying a webhook URL for receiving updates.",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    appId: {
+      propDefinition: [
+        app,
+        "appId",
+      ],
+    },
+    data: {
+      type: "object",
+      label: "Data",
+      description: "Additional data to include with the request. [See the documentation](https://fal.ai/models/fal-ai/lora/api#schema-input) for more input fields.",
+      default: {
+        model_name: "stabilityai/stable-diffusion-xl-base-1.0",
+        prompt: "Photo of a european medieval 40 year old queen, silver hair, highly detailed face, detailed eyes, head shot, intricate crown, age spots, wrinkles",
+      },
+    },
+    falWebhook: {
+      propDefinition: [
+        app,
+        "falWebhook",
+      ],
+    },
+  },
+  methods: {
+    addToQueue({
+      appId, ...args
+    } = {}) {
+      return this.app.post({
+        path: `/${appId}`,
+        ...args,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      addToQueue,
+      appId,
+      data,
+      falWebhook,
+    } = this;
+
+    const response = await addToQueue({
+      $,
+      appId,
+      params: {
+        fal_webhook: falWebhook,
+      },
+      data,
+    });
+
+    $.export("$summary", `Successfully added request to the queue with ID \`${response.request_id}\`.`);
+    return response;
+  },
+};

components/fal_ai/actions/cancel-request/cancel-request.mjs

@@ -0,0 +1,50 @@
+import app from "../../fal_ai.app.mjs";
+
+export default {
+  key: "fal_ai-cancel-request",
+  name: "Cancel Request",
+  description: "Cancels a request in the queue. This allows you to stop a long-running task if it's no longer needed.",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    appId: {
+      propDefinition: [
+        app,
+        "appId",
+      ],
+    },
+    requestId: {
+      propDefinition: [
+        app,
+        "requestId",
+      ],
+    },
+  },
+  methods: {
+    cancelRequest({
+      appId, requestId, ...args
+    } = {}) {
+      return this.app.put({
+        path: `/${appId}/requests/${requestId}/cancel`,
+        ...args,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      cancelRequest,
+      appId,
+      requestId,
+    } = this;
+
+    const response = await cancelRequest({
+      $,
+      appId,
+      requestId,
+    });
+
+    $.export("$summary", "Successfully canceled request.");
+    return response;
+  },
+};

components/fal_ai/actions/get-request-response/get-request-response.mjs

@@ -0,0 +1,50 @@
+import app from "../../fal_ai.app.mjs";
+
+export default {
+  key: "fal_ai-get-request-response",
+  name: "Get Request Response",
+  description: "Gets the response of a completed request in the queue. This retrieves the results of your asynchronous task.",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    appId: {
+      propDefinition: [
+        app,
+        "appId",
+      ],
+    },
+    requestId: {
+      propDefinition: [
+        app,
+        "requestId",
+      ],
+    },
+  },
+  methods: {
+    getRequestResponse({
+      appId, requestId, ...args
+    } = {}) {
+      return this.app._makeRequest({
+        path: `/${appId}/requests/${requestId}`,
+        ...args,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      getRequestResponse,
+      appId,
+      requestId,
+    } = this;
+
+    const response = await getRequestResponse({
+      $,
+      appId,
+      requestId,
+    });
+
+    $.export("$summary", "Successfully retrieved the request response.");
+    return response;
+  },
+};

components/fal_ai/actions/get-request-status/get-request-status.mjs

@@ -0,0 +1,63 @@
+import app from "../../fal_ai.app.mjs";
+
+export default {
+  key: "fal_ai-get-request-status",
+  name: "Get Request Status",
+  description: "Gets the status of a request in the queue. This allows you to monitor the progress of your asynchronous tasks.",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    appId: {
+      propDefinition: [
+        app,
+        "appId",
+      ],
+    },
+    requestId: {
+      propDefinition: [
+        app,
+        "requestId",
+      ],
+    },
+    logs: {
+      propDefinition: [
+        app,
+        "logs",
+      ],
+    },
+  },
+  methods: {
+    getRequestStatus({
+      appId, requestId, ...args
+    } = {}) {
+      return this.app._makeRequest({
+        path: `/${appId}/requests/${requestId}/status`,
+        ...args,
+      });
+    },
+  },
+  async run({ $ }) {
+    const {
+      getRequestStatus,
+      appId,
+      requestId,
+      logs,
+    } = this;
+
+    const response = await getRequestStatus({
+      $,
+      appId,
+      requestId,
+      params: {
+        logs: logs
+          ? 1
+          : undefined,
+      },
+    });
+
+    $.export("$summary", `Successfully retrieved status for request ID \`${response.request_id}\`.`);
+
+    return response;
+  },
+};

components/fal_ai/fal_ai.app.mjs

@@ -1,11 +1,63 @@
+import { axios } from "@pipedream/platform";
+
 export default {
   type: "app",
   app: "fal_ai",
-  propDefinitions: {},
+  propDefinitions: {
+    appId: {
+      type: "string",
+      label: "App ID",
+      description: "The unique identifier for the app. Eg. `lora`.",
+    },
+    falWebhook: {
+      type: "string",
+      label: "Webhook URL",
+      description: "The URL to receive updates via webhook.",
+      optional: true,
+    },
+    requestId: {
+      type: "string",
+      label: "Request ID",
+      description: "The unique identifier for the request.",
+    },
+    logs: {
+      type: "boolean",
+      label: "Enable Logs",
+      description: "Specify if logs should be enabled for the request status.",
+      optional: true,
+    },
+  },
   methods: {
-    // this.$auth contains connected account data
-    authKeys() {
-      console.log(Object.keys(this.$auth));
+    getUrl(path) {
+      return `https://queue.fal.run/fal-ai${path}`;
+    },
+    getHeaders(headers) {
+      return {
+        ...headers,
+        "Authorization": `Key ${this.$auth.api_key}`,
+        "Content-Type": "application/json",
+      };
+    },
+    _makeRequest({
+      $ = this, path, headers, ...args
+    } = {}) {
+      return axios($, {
+        ...args,
+        url: this.getUrl(path),
+        headers: this.getHeaders(headers),
+      });
+    },
+    post(args = {}) {
+      return this._makeRequest({
+        method: "POST",
+        ...args,
+      });
+    },
+    put(args = {}) {
+      return this._makeRequest({
+        method: "PUT",
+        ...args,
+      });
     },
   },
 };

components/fal_ai/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pipedream/fal_ai",
-  "version": "0.0.1",
+  "version": "0.1.0",
   "description": "Pipedream fal.ai Components",
   "main": "fal_ai.app.mjs",
   "keywords": [
@@ -11,5 +11,8 @@
   "author": "Pipedream <[email protected]> (https://pipedream.com/)",
   "publishConfig": {
     "access": "public"
+  },
+  "dependencies": {
+    "@pipedream/platform": "3.0.3"
   }
-}
+}