Merging pull request #14868

* Added requested changes * Added requested changes * Added requested changes
PipedreamHQ · Dec 9, 2024 · 63ac5ef · 63ac5ef
1 parent 426d08a
commit 63ac5ef
Show file tree

Hide file tree

Showing 7 changed files with 344 additions and 17 deletions.
diff --git a/components/x_ai/actions/create-embeddings/create-embeddings.mjs b/components/x_ai/actions/create-embeddings/create-embeddings.mjs
@@ -0,0 +1,57 @@
+import app from "../../x_ai.app.mjs";
+
+export default {
+  key: "x_ai-create-embeddings",
+  name: "Create Embedding",
+  description: "Create an embedding vector representation corresponding to the input text. [See the documentation](https://docs.x.ai/api/endpoints#create-embeddings)",
+  version: "0.0.2",
+  type: "action",
+  props: {
+    app,
+    embeddingModel: {
+      propDefinition: [
+        app,
+        "embeddingModel",
+      ],
+    },
+    input: {
+      propDefinition: [
+        app,
+        "input",
+      ],
+    },
+    dimensions: {
+      propDefinition: [
+        app,
+        "dimensions",
+      ],
+    },
+    encodingFormat: {
+      propDefinition: [
+        app,
+        "encodingFormat",
+      ],
+    },
+    user: {
+      propDefinition: [
+        app,
+        "user",
+      ],
+    },
+  },
+
+  async run({ $ }) {
+    const response = await this.app.createEmbeddings({
+      $,
+      data: {
+        dimensions: this.dimensions,
+        encoding_format: this.encodingFormat,
+        input: this.input,
+        model: this.embeddingModel,
+        user: this.user,
+      },
+    });
+    $.export("$summary", "Successfully created embedding");
+    return response;
+  },
+};
diff --git a/components/x_ai/actions/get-model/get-model.mjs b/components/x_ai/actions/get-model/get-model.mjs
@@ -4,7 +4,7 @@ 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",
+  version: "0.0.2",
   type: "action",
   props: {
     app,
@@ -21,9 +21,7 @@ export default {
       $,
       model: this.model,
     });
-
     $.export("$summary", `Successfully retrieved the '${this.model}' model`);
-
     return response;
   },
 };
diff --git a/components/x_ai/actions/post-chat-completion/post-chat-completion.mjs b/components/x_ai/actions/post-chat-completion/post-chat-completion.mjs
@@ -4,7 +4,7 @@ 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",
+  version: "0.0.2",
   type: "action",
   props: {
     app,
@@ -20,6 +20,66 @@ export default {
         "message",
       ],
     },
+    frequencyPenalty: {
+      propDefinition: [
+        app,
+        "frequencyPenalty",
+      ],
+    },
+    logprobs: {
+      propDefinition: [
+        app,
+        "logprobs",
+      ],
+    },
+    maxTokens: {
+      propDefinition: [
+        app,
+        "maxTokens",
+      ],
+    },
+    n: {
+      propDefinition: [
+        app,
+        "n",
+      ],
+    },
+    presencePenalty: {
+      propDefinition: [
+        app,
+        "presencePenalty",
+      ],
+    },
+    seed: {
+      propDefinition: [
+        app,
+        "seed",
+      ],
+    },
+    stream: {
+      propDefinition: [
+        app,
+        "stream",
+      ],
+    },
+    temperature: {
+      propDefinition: [
+        app,
+        "temperature",
+      ],
+    },
+    topP: {
+      propDefinition: [
+        app,
+        "topP",
+      ],
+    },
+    user: {
+      propDefinition: [
+        app,
+        "user",
+      ],
+    },
   },
 
   async run({ $ }) {
@@ -28,16 +88,24 @@ export default {
       data: {
         model: this.model,
         messages: [
-          {
-            role: "user",
-            content: this.message,
+          {
+            role: "user",
+            content: this.message,
           },
         ],
+        frequency_penalty: Number(this.frequencyPenalty),
+        logprobs: this.logprobs,
+        max_tokens: this.maxTokens,
+        n: this.n,
+        presence_penalty: Number(this.presencePenalty),
+        seed: this.seed,
+        stream: this.stream,
+        temperature: Number(this.temperature),
+        top_p: Number(this.topP),
+        user: this.user,
       },
     });
-
     $.export("$summary", `Successfully sent message to the model '${this.model}'`);
-
     return response;
   },
 };
diff --git a/components/x_ai/actions/post-completion/post-completion.mjs b/components/x_ai/actions/post-completion/post-completion.mjs
@@ -4,7 +4,7 @@ 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",
+  version: "0.0.2",
   type: "action",
   props: {
     app,
@@ -20,6 +20,78 @@ export default {
         "prompt",
       ],
     },
+    echo: {
+      propDefinition: [
+        app,
+        "echo",
+      ],
+    },
+    frequencyPenalty: {
+      propDefinition: [
+        app,
+        "frequencyPenalty",
+      ],
+    },
+    logprobs: {
+      propDefinition: [
+        app,
+        "logprobs",
+      ],
+    },
+    maxTokens: {
+      propDefinition: [
+        app,
+        "maxTokens",
+      ],
+    },
+    n: {
+      propDefinition: [
+        app,
+        "n",
+      ],
+    },
+    presencePenalty: {
+      propDefinition: [
+        app,
+        "presencePenalty",
+      ],
+    },
+    seed: {
+      propDefinition: [
+        app,
+        "seed",
+      ],
+    },
+    stream: {
+      propDefinition: [
+        app,
+        "stream",
+      ],
+    },
+    suffix: {
+      propDefinition: [
+        app,
+        "suffix",
+      ],
+    },
+    temperature: {
+      propDefinition: [
+        app,
+        "temperature",
+      ],
+    },
+    topP: {
+      propDefinition: [
+        app,
+        "topP",
+      ],
+    },
+    user: {
+      propDefinition: [
+        app,
+        "user",
+      ],
+    },
   },
 
   async run({ $ }) {
@@ -28,11 +100,21 @@ export default {
       data: {
         model: this.model,
         prompt: this.prompt,
+        echo: this.echo,
+        frequency_penalty: Number(this.frequencyPenalty),
+        logprobs: this.logprobs,
+        max_tokens: this.maxTokens,
+        n: this.n,
+        presence_penalty: Number(this.presencePenalty),
+        seed: this.seed,
+        stream: this.stream,
+        suffix: this.suffix,
+        temperature: Number(this.temperature),
+        top_p: Number(this.topP),
+        user: this.user,
       },
     });
-
     $.export("$summary", `Successfully sent prompt to the model '${this.model}'`);
-
     return response;
   },
 };
diff --git a/components/x_ai/common/constants.mjs b/components/x_ai/common/constants.mjs
@@ -0,0 +1,6 @@
+export default {
+  ENCODING_FORMATS: [
+    "float",
+    "base64",
+  ],
+};
diff --git a/components/x_ai/package.json b/components/x_ai/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@pipedream/x_ai",
-  "version": "0.1.0",
-  "description": "Pipedream X AI Components",
+  "version": "0.1.1",
+  "description": "Pipedream Chat Data Components",
   "main": "x_ai.app.mjs",
   "keywords": [
     "pipedream",