koishi-plugin-hhs-huatu 0.0.1

package/lib/config.d.ts

@@ -0,0 +1,188 @@
+import { Computed, Dict, Schema, Session } from 'koishi';
+import { Size } from './utils';
+export declare const modelMap: {
+    readonly safe: "safe-diffusion";
+    readonly nai: "nai-diffusion";
+    readonly furry: "nai-diffusion-furry-3";
+    readonly 'nai-v3': "nai-diffusion-3";
+    readonly 'nai-v4-curated-preview': "nai-diffusion-4-curated-preview";
+};
+export declare const orientMap: {
+    readonly landscape: {
+        readonly height: 832;
+        readonly width: 1216;
+    };
+    readonly portrait: {
+        readonly height: 1216;
+        readonly width: 832;
+    };
+    readonly square: {
+        readonly height: 1024;
+        readonly width: 1024;
+    };
+};
+export declare const hordeModels: string[];
+type Model = keyof typeof modelMap;
+type Orient = keyof typeof orientMap;
+export declare const models: Model[];
+export declare const orients: Orient[];
+export declare namespace scheduler {
+    const nai: readonly ["native", "karras", "exponential", "polyexponential"];
+    const nai4: readonly ["karras", "exponential", "polyexponential"];
+    const sd: readonly ["Automatic", "Uniform", "Karras", "Exponential", "Polyexponential", "SGM Uniform"];
+    const horde: readonly ["karras"];
+    const comfyUI: readonly ["normal", "karras", "exponential", "sgm_uniform", "simple", "ddim_uniform"];
+}
+export declare namespace sampler {
+    const nai: {
+        k_euler_a: string;
+        k_euler: string;
+        k_lms: string;
+        ddim: string;
+        plms: string;
+    };
+    const nai3: {
+        k_euler: string;
+        k_euler_a: string;
+        k_dpmpp_2s_ancestral: string;
+        k_dpmpp_2m: string;
+        k_dpmpp_sde: string;
+        ddim_v3: string;
+    };
+    const nai4: {
+        k_euler: string;
+        k_euler_a: string;
+        k_dpmpp_2s_ancestral: string;
+        k_dpmpp_2m_sde: string;
+        k_dpmpp_2m: string;
+        k_dpmpp_sde: string;
+    };
+    const sd: Dict<string>;
+    const horde: {
+        k_lms: string;
+        k_heun: string;
+        k_euler: string;
+        k_euler_a: string;
+        k_dpm_2: string;
+        k_dpm_2_a: string;
+        k_dpm_fast: string;
+        k_dpm_adaptive: string;
+        k_dpmpp_2m: string;
+        k_dpmpp_2s_a: string;
+        k_dpmpp_sde: string;
+        dpmsolver: string;
+        lcm: string;
+        DDIM: string;
+    };
+    const comfyui: {
+        euler: string;
+        euler_ancestral: string;
+        heun: string;
+        heunpp2: string;
+        dpm_2: string;
+        dpm_2_ancestral: string;
+        lms: string;
+        dpm_fast: string;
+        dpm_adaptive: string;
+        dpmpp_2s_ancestral: string;
+        dpmpp_sde: string;
+        dpmpp_sde_gpu: string;
+        dpmpp_2m: string;
+        dpmpp_2m_sde: string;
+        dpmpp_2m_sde_gpu: string;
+        dpmpp_3m_sde: string;
+        dpmpp_3m_sde_gpu: string;
+        ddpm: string;
+        lcm: string;
+        ddim: string;
+        uni_pc: string;
+        uni_pc_bh2: string;
+    };
+    function createSchema(map: Dict<string>): Schema<string, string>;
+    function sd2nai(sampler: string, model: string): string;
+}
+export declare const upscalers: readonly ["None", "Lanczos", "Nearest", "LDSR", "ESRGAN_4x", "R-ESRGAN General 4xV3", "R-ESRGAN General WDN 4xV3", "R-ESRGAN AnimeVideo", "R-ESRGAN 4x+", "R-ESRGAN 4x+ Anime6B", "R-ESRGAN 2x+", "ScuNET GAN", "ScuNET PSNR", "SwinIR 4x"];
+export declare const latentUpscalers: string[];
+export interface Options {
+    enhance: boolean;
+    model: string;
+    resolution: Size;
+    sampler: string;
+    seed: string;
+    steps: number;
+    scale: number;
+    noise: number;
+    strength: number;
+}
+export interface PromptConfig {
+    basePrompt?: Computed<string>;
+    negativePrompt?: Computed<string>;
+    forbidden?: Computed<string>;
+    defaultPromptSw?: boolean;
+    defaultPrompt?: Computed<string>;
+    placement?: Computed<'before' | 'after'>;
+    latinOnly?: Computed<boolean>;
+    translator?: boolean;
+    lowerCase?: boolean;
+    maxWords?: Computed<number>;
+}
+export declare const PromptConfig: Schema<PromptConfig>;
+interface FeatureConfig {
+    anlas?: Computed<boolean>;
+    text?: Computed<boolean>;
+    image?: Computed<boolean>;
+    upscale?: Computed<boolean>;
+}
+interface ParamConfig {
+    model?: Model;
+    sampler?: string;
+    smea?: boolean;
+    smeaDyn?: boolean;
+    scheduler?: string;
+    decrisper?: boolean;
+    upscaler?: string;
+    restoreFaces?: boolean;
+    hiresFix?: boolean;
+    hiresFixUpscaler: string;
+    scale?: Computed<number>;
+    textSteps?: Computed<number>;
+    imageSteps?: Computed<number>;
+    maxSteps?: Computed<number>;
+    strength?: Computed<number>;
+    resolution?: Computed<Orient | Size>;
+    maxResolution?: Computed<number>;
+}
+export interface Config extends PromptConfig, ParamConfig {
+    type: 'token' | 'login' | 'naifu' | 'sd-webui' | 'stable-horde' | 'comfyui';
+    token?: string;
+    email?: string;
+    password?: string;
+    authLv?: Computed<number>;
+    authLvDefault?: Computed<number>;
+    output?: Computed<'minimal' | 'default' | 'verbose'>;
+    features?: FeatureConfig;
+    apiEndpoint?: string;
+    endpoint?: string;
+    headers?: Dict<string>;
+    nsfw?: Computed<'disallow' | 'censor' | 'allow'>;
+    maxIterations?: number;
+    maxRetryCount?: number;
+    requestTimeout?: number;
+    recallTimeout?: number;
+    maxConcurrency?: number;
+    pollInterval?: number;
+    trustedWorkers?: boolean;
+    workflowText2Image?: string;
+    workflowImage2Image?: string;
+    maxQueueSize?: number;
+    maxUserQueueSize?: number;
+    penaltyCooldown?: number;
+}
+export declare const Config: Schema<Config>;
+interface Forbidden {
+    pattern: string;
+    strict: boolean;
+}
+export declare function parseForbidden(input: string): Forbidden[];
+export declare function parseInput(session: Session, input: string, config: Config, override: boolean): string[];
+export {};

package/lib/index.d.ts

@@ -0,0 +1,10 @@
+import { Context } from 'koishi';
+import { Config } from './config';
+export * from './config';
+export declare const reactive = true;
+export declare const name = "hhs-huatu";
+export declare const inject: {
+    required: string[];
+    optional: string[];
+};
+export declare function apply(ctx: Context, config: Config): void;

package/lib/types.d.ts

@@ -0,0 +1,127 @@
+export interface Perks {
+    maxPriorityActions: number;
+    startPriority: number;
+    contextTokens: number;
+    moduleTrainingSteps: number;
+    unlimitedMaxPriority: boolean;
+    voiceGeneration: boolean;
+    imageGeneration: boolean;
+    unlimitedImageGeneration: boolean;
+    unlimitedImageGenerationLimits: {
+        resolution: number;
+        maxPrompts: number;
+    }[];
+}
+export interface PaymentProcessorData {
+    c: string;
+    n: number;
+    o: string;
+    p: number;
+    r: string;
+    s: string;
+    t: number;
+    u: string;
+}
+export interface TrainingStepsLeft {
+    fixedTrainingStepsLeft: number;
+    purchasedTrainingSteps: number;
+}
+export interface Subscription {
+    tier: number;
+    active: boolean;
+    expiresAt: number;
+    perks: Perks;
+    paymentProcessorData: PaymentProcessorData;
+    trainingStepsLeft: TrainingStepsLeft;
+}
+export interface ImageData {
+    buffer: ArrayBuffer;
+    base64: string;
+    dataUrl: string;
+}
+export declare namespace NovelAI {
+    /** 0.5, 0.5 means make ai choose */
+    interface V4CharacterPromptCenter {
+        x: number;
+        y: number;
+    }
+    interface V4CharacterPrompt {
+        prompt: string;
+        uc: string;
+        center: V4CharacterPromptCenter;
+    }
+    interface V4CharCaption {
+        char_caption: string;
+        centers: V4CharacterPromptCenter[];
+    }
+    interface V4PromptCaption {
+        base_caption: string;
+        char_captions: V4CharCaption[];
+    }
+    interface V4Prompt {
+        caption: V4PromptCaption;
+    }
+    interface V4PromptPositive extends V4Prompt {
+        use_coords: boolean;
+        use_order: boolean;
+    }
+}
+export declare namespace StableDiffusionWebUI {
+    interface Request {
+        prompt: string;
+        negative_prompt?: string;
+        enable_hr?: boolean;
+        denoising_strength?: number;
+        firstphase_width?: number;
+        firstphase_height?: number;
+        styles?: string[];
+        seed?: number;
+        subseed?: number;
+        subseed_strength?: number;
+        seed_resize_from_h?: number;
+        seed_resize_from_w?: number;
+        batch_size?: number;
+        n_iter?: number;
+        steps?: number;
+        cfg_scale?: number;
+        width?: number;
+        height?: number;
+        restore_faces?: boolean;
+        tiling?: boolean;
+        eta?: number;
+        s_churn?: number;
+        s_tmax?: number;
+        s_tmin?: number;
+        s_noise?: number;
+        sampler_index?: string;
+    }
+    interface Response {
+        /** Image list in base64 format */
+        images: string[];
+        parameters: any;
+        info: any;
+    }
+    /**
+     * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/828438b4a190759807f9054932cae3a8b880ddf1/modules/api/models.py#L122
+     */
+    interface ExtraSingleImageRequest {
+        image: string;
+        /** Sets the resize mode: 0 to upscale by upscaling_resize amount, 1 to upscale up to upscaling_resize_h x upscaling_resize_w. */
+        resize_mode?: 0 | 1;
+        show_extras_results?: boolean;
+        gfpgan_visibility?: number;
+        codeformer_visibility?: number;
+        codeformer_weight?: number;
+        upscaling_resize?: number;
+        upscaling_resize_w?: number;
+        upscaling_resize_h?: number;
+        upscaling_crop?: boolean;
+        upscaler_1?: string;
+        upscaler_2?: string;
+        extras_upscaler_2_visibility?: number;
+        upscale_first?: boolean;
+    }
+    interface ExtraSingleImageResponse {
+        image: string;
+    }
+}

package/lib/utils.d.ts

@@ -0,0 +1,26 @@
+import { Context, Dict } from 'koishi';
+import { ImageData } from './types';
+export declare function project(object: {}, mapping: {}): {};
+export interface Size {
+    width: number;
+    height: number;
+}
+export declare function getImageSize(buffer: ArrayBuffer): Size;
+export declare function download(ctx: Context, url: string, headers?: {}): Promise<ImageData>;
+export declare function calcAccessKey(email: string, password: string): Promise<string>;
+export declare function calcEncryptionKey(email: string, password: string): Promise<string>;
+export declare class NetworkError extends Error {
+    params: {};
+    constructor(message: string, params?: {});
+    static catch: (mapping: Dict<string>) => (e: any) => never;
+}
+export declare function login(ctx: Context): Promise<string>;
+export declare function closestMultiple(num: number, mult?: number): number;
+export interface Size {
+    width: number;
+    height: number;
+    /** Indicate whether this resolution is pre-defined or customized */
+    custom?: boolean;
+}
+export declare function resizeInput(size: Size): Size;
+export declare function forceDataPrefix(url: string, mime?: string): string;

package/package.json

@@ -0,0 +1,20 @@
+{
+  "name": "koishi-plugin-hhs-huatu",
+  "description": "novelai更新了模型尝试一下",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "^4.18.3"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-hhs-huatu
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-hhs-huatu?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-hhs-huatu)
+
+对novelaibot简单修改，修复使用novelai时1，不能更换-furry模型。2新增排队逻辑。增加单个用户多个排队处罚cd机制。其他的没什么改变。