@civitai/client 0.1.1-beta.0

package/dist/generated/types.gen.js

@@ -0,0 +1,3 @@
+"use strict";
+// This file is auto-generated by @hey-api/openapi-ts
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export * from './generated';
+export * from './utils';
+export * from './client/CivitaiClient';

package/dist/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./generated"), exports);
+tslib_1.__exportStar(require("./utils"), exports);
+tslib_1.__exportStar(require("./client/CivitaiClient"), exports);

package/dist/utils/Air.d.ts

@@ -0,0 +1,19 @@
+type AirProps = {
+  /** Type of the ecosystem (sd1, sd2, sdxl) */
+  ecosystem: string;
+  /** Type of the resource (model, lora, embedding, hypernet) */
+  type: string;
+  /**  Supported network source */
+  source: string;
+  /** Id of the resource from the source */
+  id: string;
+  version?: string;
+  /** The format of the model (safetensor, ckpt, diffuser, tensor rt) optional */
+  format?: string;
+};
+/** https://github.com/civitai/civitai/wiki/AIR-%E2%80%90-Uniform-Resource-Names-for-AI */
+export declare abstract class Air {
+  static parse(identifier: string): AirProps;
+  static stringify({ ecosystem, type, source, id, version, format }: AirProps): string;
+}
+export {};

package/dist/utils/Air.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Air = void 0;
+const regex = /^(?:urn:)?(?:air:)?(?:(?<ecosystem>[a-zA-Z0-9_\-\/]+):)?(?:(?<type>[a-zA-Z0-9_\-\/]+):)?(?<source>[a-zA-Z0-9_\-\/]+):(?<id>[a-zA-Z0-9_\-\/]+)(?:@(?<version>[a-zA-Z0-9_\-]+))?(?:\.(?<format>[a-zA-Z0-9_\-]+))?$/i;
+/** https://github.com/civitai/civitai/wiki/AIR-%E2%80%90-Uniform-Resource-Names-for-AI */
+class Air {
+    static parse(identifier) {
+        const match = regex.exec(identifier);
+        if (!match) {
+            throw new Error(`Invalid identifier: ${identifier}`);
+        }
+        return match.groups;
+    }
+    static stringify({ ecosystem, type, source, id, version, format }) {
+        return `urn:air:${ecosystem}:${type}:${source}:${id}${version ? `@${version}` : ''}${format ? `:${format}` : ''}`;
+    }
+}
+exports.Air = Air;

package/dist/utils/CivitaiClientError.js

@@ -0,0 +1,4 @@
+"use strict";
+// export class ApiClientError extends Error {
+//   constructor() {}
+// }

package/dist/utils/TimeSpan.d.ts

@@ -0,0 +1,45 @@
+export declare class TimeSpan {
+  #private;
+  /**
+   *
+   * @param timeSpanString format "days:hours:minutes:seconds.ticks" ("00:00:00:00.000000")
+   */
+  constructor(timeSpanString: string);
+  constructor(ticks: number);
+  constructor(hours: number, minutes: number, seconds: number);
+  constructor(days: number, hours: number, minutes: number, seconds: number);
+  constructor(days: number, hours: number, minutes: number, seconds: number, milliseconds: number);
+  constructor(
+    days: number,
+    hours: number,
+    minutes: number,
+    seconds: number,
+    milliseconds: number,
+    ticks: number
+  );
+  addTicks: (value: number) => number;
+  subtractTicks: (value: number) => number;
+  addMilliseconds: (value: number) => number;
+  subtracktMilliseconds: (value: number) => number;
+  addSeconds: (value: number) => number;
+  subtractSeconds: (value: number) => number;
+  addMinutes: (value: number) => number;
+  subtractMinutes: (value: number) => number;
+  addHours: (value: number) => number;
+  subtractHours: (value: number) => number;
+  addDays: (value: number) => number;
+  subtractDays: (value: number) => number;
+  get days(): number;
+  get hours(): number;
+  get minutes(): number;
+  get seconds(): number;
+  get milliseconds(): number;
+  get ticks(): number;
+  get totalMilliseconds(): number;
+  get totalMinutes(): number;
+  get totalSeconds(): number;
+  get totalHours(): number;
+  get totalDays(): number;
+  get totalTicks(): number;
+  toString: () => string;
+}

package/dist/utils/TimeSpan.js

@@ -0,0 +1,101 @@
+"use strict";
+var _TimeSpan_ticks, _TimeSpan_add, _TimeSpan_subtract;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TimeSpan = void 0;
+const tslib_1 = require("tslib");
+const ticksPerMillisecond = 10000, ticksPerSecond = 1000 * ticksPerMillisecond, ticksPerMinute = 60000 * ticksPerMillisecond, ticksPerHour = 3600000 * ticksPerMillisecond, ticksPerDay = 86400000 * ticksPerMillisecond;
+const timeSpanWithDays = /^(\d+):(\d+):(\d+):(\d+).(\d+)?/, timeSpanNoDays = /^(\d+):(\d+):(\d+).(\d+)?/;
+function toDoubleDigitString(n) {
+    return n < 10 ? '0' + n : n;
+}
+class TimeSpan {
+    constructor() {
+        _TimeSpan_ticks.set(this, 0);
+        _TimeSpan_add.set(this, (value, multiplier) => (tslib_1.__classPrivateFieldSet(this, _TimeSpan_ticks, tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") + (multiplier ? value * multiplier : value), "f")));
+        _TimeSpan_subtract.set(this, (value, multiplier) => (tslib_1.__classPrivateFieldSet(this, _TimeSpan_ticks, tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") - (multiplier ? value * multiplier : value), "f")));
+        this.addTicks = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value);
+        this.subtractTicks = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value);
+        this.addMilliseconds = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value, ticksPerMillisecond);
+        this.subtracktMilliseconds = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value, ticksPerMillisecond);
+        this.addSeconds = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value, ticksPerSecond);
+        this.subtractSeconds = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value, ticksPerSecond);
+        this.addMinutes = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value, ticksPerMinute);
+        this.subtractMinutes = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value, ticksPerMinute);
+        this.addHours = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value, ticksPerHour);
+        this.subtractHours = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value, ticksPerHour);
+        this.addDays = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_add, "f").call(this, value, ticksPerDay);
+        this.subtractDays = (value) => tslib_1.__classPrivateFieldGet(this, _TimeSpan_subtract, "f").call(this, value, ticksPerDay);
+        this.toString = () => {
+            return ([this.days, this.hours, this.minutes, this.seconds].map(toDoubleDigitString).join(':') +
+                '.' +
+                toDoubleDigitString(this.ticks));
+        };
+        let days = 0, hours = 0, minutes = 0, seconds = 0, milliseconds = 0, ticks = 0;
+        if (typeof arguments[0] === 'string') {
+            const str = arguments[0];
+            let match;
+            if ((match = str.match(timeSpanWithDays))) {
+                [days, hours, minutes, seconds, ticks] = match.splice(1, 6).map(Number);
+            }
+            else if ((match = str.match(timeSpanNoDays))) {
+                [hours, minutes, seconds, ticks] = match.splice(1, 4).map(Number);
+            }
+        }
+        else if (arguments.length === 1)
+            [ticks] = arguments;
+        else if (arguments.length === 3)
+            [hours, minutes, seconds] = arguments;
+        else if (arguments.length === 4)
+            [days, hours, minutes, seconds] = arguments;
+        else if (arguments.length === 5)
+            [days, hours, minutes, seconds, milliseconds] = arguments;
+        this.addDays(days);
+        this.addHours(hours);
+        this.addMinutes(minutes);
+        this.addSeconds(seconds);
+        this.addMilliseconds(milliseconds);
+        this.addTicks(ticks);
+    }
+    get days() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerDay);
+    }
+    get hours() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerHour) % 24;
+    }
+    get minutes() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerMinute) % 60;
+    }
+    get seconds() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerSecond) % 60;
+    }
+    get milliseconds() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerMillisecond) % 1000;
+    }
+    get ticks() {
+        const dayTicks = this.days * ticksPerDay;
+        const hourTicks = this.hours * ticksPerHour;
+        const minuteTicks = this.minutes * ticksPerMinute;
+        const secondTicks = this.seconds * ticksPerSecond;
+        return tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") - dayTicks - hourTicks - minuteTicks - secondTicks;
+    }
+    get totalMilliseconds() {
+        return Math.floor(tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f") / ticksPerMillisecond);
+    }
+    get totalMinutes() {
+        return Math.floor(this.minutes / ticksPerMinute);
+    }
+    get totalSeconds() {
+        return Math.floor(this.seconds / ticksPerSecond);
+    }
+    get totalHours() {
+        return Math.floor(this.hours / ticksPerHour);
+    }
+    get totalDays() {
+        return Math.floor(this.days / ticksPerDay);
+    }
+    get totalTicks() {
+        return tslib_1.__classPrivateFieldGet(this, _TimeSpan_ticks, "f");
+    }
+}
+exports.TimeSpan = TimeSpan;
+_TimeSpan_ticks = new WeakMap(), _TimeSpan_add = new WeakMap(), _TimeSpan_subtract = new WeakMap();

package/dist/utils/createClient.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/utils/createClient.js

@@ -0,0 +1,119 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+// class CivitaiClient extends GeneratedClient implements ICivitaiClient {
+//   constructor(config: ClientConfig, HttpRequest?: HttpRequestConstructor) {
+//     super(
+//       {
+//         BASE:
+//           config.env === 'dev'
+//             ? 'https://orchestration-dev.civitai.com'
+//             : 'https://orchestration-stage.civitai.com', // TODO: Switch to orchestration.civitai.com once prod is back up and running
+//         HEADERS: { Authorization: `Bearer ${config.auth}` },
+//         interceptors: {
+//           request: config?.interceptors?.request ?? new Interceptors(),
+//           response: config?.interceptors?.response ?? new Interceptors(),
+//         },
+//       },
+//       HttpRequest
+//     );
+//   }
+//   public override requests = new CustomRequestsService(this.request);
+// }
+// export class CustomRequestsService extends RequestsService {
+//   constructor(public readonly httpRequest: BaseHttpRequest) {
+//     super(httpRequest);
+//   }
+//   public override submitRequest(
+//     props: Parameters<GeneratedClient['requests']['submitRequest']>[0],
+//     options?: { cleanupOnCancellation?: boolean }
+//   ): CancelablePromise<RequestInfo> {
+//     return new CancelablePromise<RequestInfo>(async (resolve, reject, oncancel) => {
+//       const { wait, include } = props ?? {};
+//       const { cleanupOnCancellation = true } = options ?? {};
+//       let result = await withCancellation(this.submitRequest({ ...props, wait: false }));
+//       if (!result) throw new Error('no result');
+//       try {
+//         while (wait && (result.status === 'Initialized' || result.status === 'Processing')) {
+//           result = await withCancellation(this.getRequest({ requestId: result.id, include, wait }));
+//         }
+//         resolve(result);
+//       } catch (error) {
+//         if (error instanceof CancelError && wait && cleanupOnCancellation) {
+//           console.log('cancelling request');
+//           await this.updateRequest({
+//             requestId: result.id,
+//             requestBody: {
+//               status: 'Canceled',
+//             },
+//           });
+//         }
+//         reject(error);
+//       }
+//       function withCancellation<T>(promise: CancelablePromise<T>) {
+//         oncancel(promise.cancel);
+//         return promise;
+//       }
+//     });
+//   }
+// }
+// export function createCivitaiClient(config: ClientConfig, HttpRequest?: HttpRequestConstructor) {
+//   const client = new GeneratedClient(
+//     {
+//       BASE:
+//         config.base ?? config.env === 'dev'
+//           ? 'https://orchestration-dev.civitai.com'
+//           : 'https://orchestration-stage.civitai.com', // TODO: Switch to orchestration.civitai.com once prod is back up and running
+//       HEADERS: { Authorization: `Bearer ${config.auth}` },
+//       interceptors: {
+//         request: config?.interceptors?.request ?? new Interceptors(),
+//         response: config?.interceptors?.response ?? new Interceptors(),
+//       },
+//     },
+//     HttpRequest
+//   ) as ICivitaiClient;
+//   function trySubmitRequest(
+//     props: Parameters<typeof client.requests.submitRequest>[0],
+//     options?: { cleanupOnCancellation?: boolean }
+//   ) {
+//     return new CancelablePromise<RequestInfo>(async (resolve, reject, oncancel) => {
+//       try {
+//         const { wait, include } = props ?? {};
+//         const { cleanupOnCancellation = true } = options ?? {};
+//         let result = await withCancellation(
+//           client.requests.submitRequest({ ...props, wait: false })
+//         );
+//         if (!result) throw new Error('no result');
+//         try {
+//           while (wait && (result.status === 'Initialized' || result.status === 'Processing')) {
+//             result = await withCancellation(
+//               client.requests.getRequest({ requestId: result.id, include, wait })
+//             );
+//           }
+//           resolve(result);
+//         } catch (error) {
+//           if (error instanceof CancelError && wait && cleanupOnCancellation) {
+//             console.log('cancelling request');
+//             await client.requests.updateRequest({
+//               requestId: result.id,
+//               requestBody: {
+//                 status: 'Canceled',
+//               },
+//             });
+//           }
+//           throw error;
+//         }
+//       } catch (error) {
+//         reject(error);
+//       }
+//       function withCancellation<T>(promise: CancelablePromise<T>) {
+//         oncancel(promise.cancel);
+//         return promise;
+//       }
+//     });
+//   }
+//   const requests = { ...client.requests, submitRequest: trySubmitRequest };
+//   return {
+//     ...client,
+//     requests,
+//   };
+// }

package/dist/utils/index.d.ts

@@ -0,0 +1,2 @@
+export * from './Air';
+export * from './TimeSpan';

package/dist/utils/index.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./Air"), exports);
+tslib_1.__exportStar(require("./TimeSpan"), exports);

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@civitai/client",
+  "version": "0.1.1-beta.0",
+  "description": "Civitai's javascript client for generating ai content",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "repository": "https://github.com/civitai/civitai-client-javascript",
+  "files": [
+    "dist",
+    "README.md",
+    "dist/index.d.ts",
+    "dist/index.js",
+    "package.json"
+  ],
+  "engines": {
+    "node": ">=18.0.0",
+    "npm": ">=7.19.0",
+    "git": ">=2.11.0",
+    "yarn": ">=1.7.0"
+  },
+  "scripts": {
+    "generate": "openapi-ts",
+    "build": "tsc && eslint --fix dist",
+    "test": "jest",
+    "prepublishOnly": "npm test && npm run build"
+  },
+  "devDependencies": {
+    "@hey-api/openapi-ts": "^0.42.1",
+    "@types/jest": "^29.5.12",
+    "@types/node": "^20.11.19",
+    "@typescript-eslint/eslint-plugin": "^7.7.1",
+    "@typescript-eslint/parser": "^7.7.1",
+    "dotenv": "^16.4.4",
+    "eslint": "^9.1.1",
+    "eslint-config-prettier": "^9.1.0",
+    "eslint-plugin-prettier": "^5.1.3",
+    "jest": "^29.7.0",
+    "prettier": "^3.2.5",
+    "ts-jest": "^29.1.2",
+    "typescript": "^5.3.3"
+  },
+  "dependencies": {
+    "tslib": "^2.6.2"
+  },
+  "publishConfig": {
+    "access": "public"
+  }
+}