Added actions

Author: lcaresia

components/x_ai/actions/get-model/get-model.mjs

@@ -0,0 +1,29 @@
+import app from "../../x_ai.app.mjs";
+
+export default {
+  key: "x_ai-get-model",
+  name: "Get Model",
+  description: "List all language and embedding models available. [See the documentation](https://docs.x.ai/api/endpoints#get-model)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    model: {
+      propDefinition: [
+        app,
+        "model"
+      ]
+    },
+  },
+
+  async run({ $ }) {
+    const response = await this.app.getModel({
+      $,
+      model: this.model,
+    });
+
+    $.export("$summary", `Successfully retrieved the '${this.model}' model`);
+
+    return response;
+  },
+};

components/x_ai/actions/post-chat-completion/post-chat-completion.mjs

@@ -0,0 +1,40 @@
+import app from "../../x_ai.app.mjs";
+
+export default {
+  key: "x_ai-post-chat-completion",
+  name: "Post Chat Completion",
+  description: "Create a language model response for a chat conversation. [See the documentation](https://docs.x.ai/api/endpoints#chat-completions)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    model: {
+      propDefinition: [
+        app,
+        "model"
+      ]
+    },
+    message: {
+      propDefinition: [
+        app,
+        "message"
+      ]
+    },
+  },
+
+  async run({ $ }) {
+    const response = await this.app.postChatCompletion({
+      $,
+      data: {
+        model: this.model,
+        messages: [
+          { role: "user", content: this.message },
+        ],
+      },
+    });
+
+    $.export("$summary", `Successfully sent message to the model '${this.model}'`);
+
+    return response;
+  },
+};

components/x_ai/actions/post-completion/post-completion.mjs

@@ -0,0 +1,38 @@
+import app from "../../x_ai.app.mjs";
+
+export default {
+  key: "x_ai-post-completion",
+  name: "Post Completion",
+  description: "Create a language model response for a given prompt. [See the documentation](https://docs.x.ai/api/endpoints#completions)",
+  version: "0.0.1",
+  type: "action",
+  props: {
+    app,
+    model: {
+      propDefinition: [
+        app,
+        "model"
+      ]
+    },
+    prompt: {
+      propDefinition: [
+        app,
+        "prompt"
+      ]
+    },
+  },
+
+  async run({ $ }) {
+    const response = await this.app.postCompletion({
+      $,
+      data: {
+        model: this.model,
+        prompt: this.prompt,
+      },
+    });
+
+    $.export("$summary", `Successfully sent prompt to the model '${this.model}'`);
+
+    return response;
+  },
+};

components/x_ai/package.json

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

components/x_ai/x_ai.app.mjs

@@ -1,11 +1,77 @@
+import { axios } from "@pipedream/platform";
+
 export default {
   type: "app",
   app: "x_ai",
-  propDefinitions: {},
+  propDefinitions: {
+    model: {
+      type: "string",
+      label: "Model",
+      description: "Specifies the model to be used for the request",
+      async options() {
+        const response = await this.listModels();
+        const modelsIds = response.data;
+        return modelsIds.map(({ id }) => ({
+          value: id
+        }));
+      }
+    },
+    prompt: {
+      type: "string",
+      label: "Prompt",
+      description: "Prompt for the request",
+    },
+    message: {
+      type: "string",
+      label: "Message",
+      description: "Message for the chat completion",
+    },
+  },
   methods: {
-    // this.$auth contains connected account data
-    authKeys() {
-      console.log(Object.keys(this.$auth));
+    _baseUrl() {
+      return `https://api.x.ai`;
+    },
+    async _makeRequest(opts = {}) {
+      const {
+        $ = this,
+        path,
+        headers,
+        ...otherOpts
+      } = opts;
+      return axios($, {
+        ...otherOpts,
+        url: this._baseUrl() + path,
+        headers: {
+          ...headers,
+          Authorization: `Bearer ${this.$auth.api_key}`,
+        },
+      });
+    },
+    async postChatCompletion(args = {}) {
+      return this._makeRequest({
+        path: "/v1/chat/completions",
+        method: "post",
+        ...args,
+      });
+    },
+    async postCompletion(args = {}) {
+      return this._makeRequest({
+        path: `/v1/completions`,
+        method: "post",
+        ...args,
+      });
+    },
+    async getModel({ model, ...args }) {
+      return this._makeRequest({
+        path: `/v1/models/${model}`,
+        ...args,
+      });
+    },
+    async listModels(args = {}) {
+      return this._makeRequest({
+        path: "/v1/models",
+        ...args,
+      });
     },
   },
 };