@xsai/generate-image 0.2.0-beta.2

package/LICENSE.md

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Moeru AI
+
+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/README.md

@@ -0,0 +1 @@
+https://xsai.js.org/docs/packages/generate/image

package/dist/index.d.ts

@@ -0,0 +1,36 @@
+import { CommonRequestOptions } from '@xsai/shared';
+
+interface GenerateImageOptions extends CommonRequestOptions {
+    [key: string]: unknown;
+    /**
+     * The number of images to generate.
+     * @default `1`
+     */
+    n?: number;
+    /** A text description of the desired image(s). */
+    prompt: string;
+    /**
+     * The size of the generated images.
+     * @default `1024x1024`
+     */
+    size?: `${number}x${number}`;
+}
+interface GenerateImageResponse {
+    created: number;
+    data: {
+        b64_json: string;
+        revised_prompt?: string;
+    }[];
+}
+interface GenerateImageResult {
+    image: GenerateImageResultImage;
+    images: GenerateImageResultImage[];
+}
+interface GenerateImageResultImage {
+    base64: string;
+    mimeType: string;
+}
+/** @experimental */
+declare const generateImage: (options: GenerateImageOptions) => Promise<GenerateImageResult>;
+
+export { type GenerateImageOptions, type GenerateImageResponse, type GenerateImageResult, type GenerateImageResultImage, generateImage };

package/dist/index.js

@@ -0,0 +1,35 @@
+import { requestURL, requestHeaders, requestBody, responseJSON } from '@xsai/shared';
+
+const mimeTypes = {
+  "/9j/": "image/jpg",
+  "AAAAIGZ0eXBhdmlm": "image/avif",
+  "iVBORw0KGgo": "image/png",
+  "R0lGOD": "image/gif",
+  "UklGRg==": "image/webp"
+};
+const convertImage = (b64_json) => {
+  const key = Object.keys(mimeTypes).find((prefix) => b64_json.startsWith(prefix));
+  const mimeType = mimeTypes[key ?? "iVBORw0KGgo"];
+  return {
+    // eslint-disable-next-line @masknet/string-no-data-url
+    base64: `data:${mimeType};base64,${b64_json}`,
+    mimeType
+  };
+};
+const generateImage = async (options) => (options.fetch ?? globalThis.fetch)(requestURL("images/generations", options.baseURL), {
+  body: requestBody({
+    ...options,
+    responseFormat: "b64_json"
+  }),
+  headers: requestHeaders({
+    "Content-Type": "application/json",
+    ...options.headers
+  }, options.apiKey),
+  method: "POST",
+  signal: options.abortSignal
+}).then(responseJSON).then(({ data }) => {
+  const images = data.map(({ b64_json }) => convertImage(b64_json));
+  return { image: images[0], images };
+});
+
+export { generateImage };

package/package.json

@@ -0,0 +1,39 @@
+{
+  "name": "@xsai/generate-image",
+  "type": "module",
+  "version": "0.2.0-beta.2",
+  "description": "extra-small AI SDK for Browser, Node.js, Deno, Bun or Edge Runtime.",
+  "author": "Moeru AI",
+  "license": "MIT",
+  "homepage": "https://xsai.js.org",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/moeru-ai/xsai.git",
+    "directory": "packages/generate-image"
+  },
+  "bugs": "https://github.com/moeru-ai/xsai/issues",
+  "keywords": [
+    "xsai",
+    "openai",
+    "ai"
+  ],
+  "sideEffects": false,
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@xsai/shared": ""
+  },
+  "scripts": {
+    "build": "pkgroll"
+  },
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts"
+}