dust init

Author: luancazarine

components/dust/actions/talk-assistant/talk-assistant.mjs

@@ -0,0 +1,38 @@
+import dust from "../../dust.app.mjs";
+import { axios } from "@pipedream/platform";
+
+export default {
+  key: "dust-talk-assistant",
+  name: "Talk to Assistant",
+  description: "Send a message to an assistant on Dust and receive an answer. [See the documentation](https://docs.dust.tt/reference/post_api-v1-w-wid-assistant-conversations-cid-messages)",
+  version: "0.0.{{ts}}",
+  type: "action",
+  props: {
+    dust,
+    content: {
+      propDefinition: [
+        dust,
+        "content",
+      ],
+    },
+    assistantId: {
+      type: "string",
+      label: "Assistant ID",
+      description: "The unique identifier of the assistant.",
+    },
+    conversationId: {
+      type: "string",
+      label: "Conversation ID",
+      description: "The unique identifier of the conversation.",
+    },
+  },
+  async run({ $ }) {
+    const response = await this.dust.sendMessageToAssistant({
+      content: this.content,
+      assistantId: this.assistantId,
+      conversationId: this.conversationId,
+    });
+    $.export("$summary", "Successfully sent message to assistant");
+    return response;
+  },
+};

components/dust/actions/upload-document/upload-document.mjs

@@ -0,0 +1,34 @@
+import dust from "../../dust.app.mjs";
+import { axios } from "@pipedream/platform";
+
+export default {
+  key: "dust-upload-document",
+  name: "Upload Document",
+  description: "Adds a document to a chosen Dust data source or folder. [See the documentation](https://docs.dust.tt/reference/post_api-v1-w-wid-data-sources-name-documents-documentid)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    dust,
+    document: {
+      propDefinition: [
+        dust,
+        "document",
+      ],
+    },
+    destination: {
+      propDefinition: [
+        dust,
+        "destination",
+      ],
+    },
+  },
+  async run({ $ }) {
+    const response = await this.dust.upsertDocument({
+      document: this.document,
+      destination: this.destination,
+    });
+
+    $.export("$summary", `Successfully uploaded document to destination ${this.destination}`);
+    return response;
+  },
+};

components/dust/dust.app.mjs

@@ -1,11 +1,83 @@
+import { axios } from "@pipedream/platform";
+
 export default {
   type: "app",
   app: "dust",
-  propDefinitions: {},
+  propDefinitions: {
+    content: {
+      type: "string",
+      label: "Message Content",
+      description: "The content of the message to be sent to the assistant",
+    },
+    document: {
+      type: "string",
+      label: "Document",
+      description: "The content of the document to be uploaded",
+    },
+    destination: {
+      type: "string",
+      label: "Destination",
+      description: "The destination data source or folder where the document will be uploaded",
+      async options() {
+        const dataSources = await this.listDataSources();
+        return dataSources.map((dataSource) => ({
+          label: dataSource.name,
+          value: dataSource.id,
+        }));
+      },
+    },
+  },
   methods: {
-    // this.$auth contains connected account data
-    authKeys() {
-      console.log(Object.keys(this.$auth));
+    _baseUrl() {
+      return "https://dust.tt/api/v1";
+    },
+    async _makeRequest(opts = {}) {
+      const {
+        $ = this,
+        method = "GET",
+        path = "/",
+        headers,
+        ...otherOpts
+      } = opts;
+
+      return axios($, {
+        ...otherOpts,
+        method,
+        url: this._baseUrl() + path,
+        headers: {
+          ...headers,
+          Authorization: `Bearer ${this.$auth.oauth_access_token}`,
+        },
+      });
+    },
+    async listDataSources(opts = {}) {
+      return this._makeRequest({
+        path: `/w/${this.$auth.workspace_id}/data_sources`,
+        ...opts,
+      });
+    },
+    async sendMessageToAssistant({
+      content, assistantId, conversationId,
+    }) {
+      return this._makeRequest({
+        method: "POST",
+        path: `/w/${this.$auth.workspace_id}/assistant/conversations/${conversationId}/messages`,
+        data: {
+          content,
+          assistantId,
+        },
+      });
+    },
+    async upsertDocument({
+      document, destination,
+    }) {
+      return this._makeRequest({
+        method: "POST",
+        path: `/w/${this.$auth.workspace_id}/data_sources/${destination}/documents/${document.documentId}`,
+        data: {
+          text: document,
+        },
+      });
     },
   },
-};
+};

components/dust/package.json

@@ -12,4 +12,4 @@
   "publishConfig": {
     "access": "public"
   }
-}
+}