@pipedream/openai 0.0.1

package/LICENSE

@@ -0,0 +1,7 @@
+Copyright 2020 Pipedream, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package/actions/common/constants.mjs

@@ -0,0 +1,11 @@
+export default {
+  IMAGE_SIZES: [
+    "256x256",
+    "512x512",
+    "1024x1024",
+  ],
+  RESPONSE_FORMATS: [
+    "url",
+    "b64_json",
+  ],
+};

package/actions/create-image/create-image.mjs

@@ -0,0 +1,55 @@
+import openai from "../../openai.app.mjs";
+import constants from "../common/constants.mjs";
+
+export default {
+  name: "Create Image",
+  version: "0.0.1",
+  key: "openai-create-image",
+  description: "Creates an image given a prompt. [See docs here](https://beta.openai.com/docs/api-reference/images/create)",
+  type: "action",
+  props: {
+    openai,
+    prompt: {
+      label: "Prompt",
+      description: "A text description of the desired image(s). The maximum length is 1000 characters.",
+      type: "string",
+    },
+    n: {
+      label: "N",
+      description: "The number of images to generate. Must be between 1 and 10.",
+      type: "integer",
+      optional: true,
+    },
+    size: {
+      label: "Size",
+      description: "The size of the generated images.",
+      type: "string",
+      optional: true,
+      options: constants.IMAGE_SIZES,
+    },
+    responseFormat: {
+      label: "Response Format",
+      description: "The format in which the generated images are returned.",
+      type: "string",
+      optional: true,
+      options: constants.RESPONSE_FORMATS,
+    },
+  },
+  async run({ $ }) {
+    const response = await this.openai.createImage({
+      $,
+      data: {
+        prompt: this.prompt,
+        n: this.n,
+        size: this.size,
+        response_format: this.responseFormat,
+      },
+    });
+
+    if (response) {
+      $.export("$summary", `Successfully created image with id ${response.id}`);
+    }
+
+    return response;
+  },
+};

package/actions/send-prompt/send-prompt.mjs

@@ -0,0 +1,111 @@
+import openai from "../../openai.app.mjs";
+
+export default {
+  name: "Send Prompt",
+  version: "0.0.1",
+  key: "openai-send-prompt",
+  description: "Creates a completion for the provided prompt and parameters. [See docs here](https://beta.openai.com/docs/api-reference/completions/create)",
+  type: "action",
+  props: {
+    openai,
+    modelId: {
+      propDefinition: [
+        openai,
+        "modelId",
+      ],
+    },
+    prompt: {
+      label: "Prompt",
+      description: "The prompt(s) to generate completions for, encoded as a string, array of strings, array of tokens, or array of token arrays. Note that <|endoftext|> is the document separator that the model sees during training, so if a prompt is not specified the model will generate as if from the beginning of a new document.",
+      type: "string[]",
+      optional: true,
+    },
+    suffix: {
+      label: "Suffix",
+      description: "The suffix that comes after a completion of inserted text.",
+      type: "string",
+      optional: true,
+    },
+    maxTokens: {
+      label: "Max Tokens",
+      description: "The maximum number of [tokens](https://beta.openai.com/tokenizer) to generate in the completion.",
+      type: "integer",
+      optional: true,
+    },
+    temperature: {
+      label: "Temperature",
+      description: "What [sampling temperature](https://towardsdatascience.com/how-to-sample-from-language-models-682bceb97277) to use. Higher values means the model will take more risks. Try 0.9 for more creative applications, and 0 (argmax sampling) for ones with a well-defined answer.",
+      type: "string",
+      optional: true,
+    },
+    topP: {
+      label: "Top P",
+      description: "An alternative to sampling with temperature, called nucleus sampling, where the model considers the results of the tokens with top_p probability mass. So 0.1 means only the tokens comprising the top 10% probability mass are considered. We generally recommend altering this or `temperature` but not both.",
+      type: "string",
+      optional: true,
+    },
+    n: {
+      label: "N",
+      description: "How many completions to generate for each prompt.",
+      type: "string",
+      optional: true,
+    },
+    stop: {
+      label: "Stop",
+      description: "Up to 4 sequences where the API will stop generating further tokens. The returned text will not contain the stop sequence.",
+      type: "string[]",
+      optional: true,
+    },
+    presencePenalty: {
+      label: "Presence Penalty",
+      description: "Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far, increasing the model's likelihood to talk about new topics.",
+      type: "string",
+      optional: true,
+    },
+    frequencyPenalty: {
+      label: "Frequency Penalty",
+      description: "Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.",
+      type: "string",
+      optional: true,
+    },
+    bestOf: {
+      label: "Best Of",
+      description: "Generates best_of completions server-side and returns the \"best\" (the one with the highest log probability per token). Results cannot be streamed.",
+      type: "integer",
+      optional: true,
+    },
+  },
+  async run({ $ }) {
+    const response = await this.openai.sendPrompt({
+      $,
+      data: {
+        model: this.modelId,
+        prompt: this.prompt,
+        max_tokens: this.maxTokens,
+        temperature: this.temperature
+          ? +this.temperature
+          : this.temperature,
+        top_p: this.topP
+          ? +this.topP
+          : this.topP,
+        n: this.n
+          ? +this.n
+          : this.n,
+        stop: this.stop,
+        presence_penalty: this.presencePenalty
+          ? +this.presencePenalty
+          : this.presencePenalty,
+        frequency_penalty: this.frequencyPenalty
+          ? +this.frequencyPenalty
+          : this.frequencyPenalty,
+        best_of: this.bestOf,
+      },
+    });
+
+    if (response) {
+      $.export("$summary", `Successfully sent prompt with id ${response.id}`);
+    }
+
+    return response;
+  },
+};

package/openai.app.mjs

@@ -0,0 +1,57 @@
+import { axios } from "@pipedream/platform";
+
+export default {
+  type: "app",
+  app: "openai",
+  propDefinitions: {
+    modelId: {
+      label: "Model ID",
+      description: "ID of the model to use",
+      type: "string",
+      async options() {
+        const { data: models } = await this.getModels();
+
+        return models.map((model) => model.id);
+      },
+    },
+  },
+  methods: {
+    _apiKey() {
+      return this.$auth.api_key;
+    },
+    _apiUrl() {
+      return "https://api.openai.com/v1";
+    },
+    async _makeRequest({
+      $ = this, path, ...args
+    }) {
+      return axios($, {
+        url: `${this._apiUrl()}${path}`,
+        headers: {
+          Authorization: `Bearer ${this._apiKey()}`,
+        },
+        ...args,
+      });
+    },
+    async getModels(args = {}) {
+      return this._makeRequest({
+        path: "/models",
+        ...args,
+      });
+    },
+    async sendPrompt(args = {}) {
+      return this._makeRequest({
+        path: "/completions",
+        method: "post",
+        ...args,
+      });
+    },
+    async createImage(args = {}) {
+      return this._makeRequest({
+        path: "/images/generations",
+        method: "post",
+        ...args,
+      });
+    },
+  },
+};

package/package.json

@@ -0,0 +1,20 @@
+{
+  "name": "@pipedream/openai",
+  "version": "0.0.1",
+  "description": "Pipedream OpenAI Components",
+  "main": "openai.app.mjs",
+  "keywords": [
+    "pipedream",
+    "openai"
+  ],
+  "homepage": "https://pipedream.com/apps/openai",
+  "author": "Pipedream <support@pipedream.com> (https://pipedream.com/)",
+  "license": "MIT",
+  "gitHead": "e12480b94cc03bed4808ebc6b13e7fdb3a1ba535",
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@pipedream/platform": "^1.2.1"
+  }
+}