@llumiverse/common
Version:
Public types, enums and options used by Llumiverse API.
52 lines • 2.2 kB
TypeScript
import { ModelOptionsInfo, ModelOptions } from "../types.js";
export type VertexAIOptions = ImagenOptions | VertexAIClaudeOptions;
export declare enum ImagenTaskType {
TEXT_IMAGE = "TEXT_IMAGE",
EDIT_MODE_INPAINT_REMOVAL = "EDIT_MODE_INPAINT_REMOVAL",
EDIT_MODE_INPAINT_INSERTION = "EDIT_MODE_INPAINT_INSERTION",
EDIT_MODE_BGSWAP = "EDIT_MODE_BGSWAP",
EDIT_MODE_OUTPAINT = "EDIT_MODE_OUTPAINT",
CUSTOMIZATION_SUBJECT = "CUSTOMIZATION_SUBJECT",
CUSTOMIZATION_STYLE = "CUSTOMIZATION_STYLE",
CUSTOMIZATION_CONTROLLED = "CUSTOMIZATION_CONTROLLED",
CUSTOMIZATION_INSTRUCT = "CUSTOMIZATION_INSTRUCT"
}
export declare enum ImagenMaskMode {
MASK_MODE_USER_PROVIDED = "MASK_MODE_USER_PROVIDED",
MASK_MODE_BACKGROUND = "MASK_MODE_BACKGROUND",
MASK_MODE_FOREGROUND = "MASK_MODE_FOREGROUND",
MASK_MODE_SEMANTIC = "MASK_MODE_SEMANTIC"
}
export interface ImagenOptions {
_option_id: "vertexai-imagen";
number_of_images?: number;
seed?: number;
person_generation?: "dont_allow" | "allow_adults" | "allow_all";
safety_setting?: "block_none" | "block_only_high" | "block_medium_and_above" | "block_low_and_above";
image_file_type?: "image/jpeg" | "image/png";
jpeg_compression_quality?: number;
aspect_ratio?: "1:1" | "4:3" | "3:4" | "16:9" | "9:16";
add_watermark?: boolean;
enhance_prompt?: boolean;
edit_mode?: ImagenTaskType;
guidance_scale?: number;
edit_steps?: number;
mask_mode?: ImagenMaskMode;
mask_dilation?: number;
mask_class?: number[];
controlType: "CONTROL_TYPE_FACE_MESH" | "CONTROL_TYPE_CANNY" | "CONTROL_TYPE_SCRIBBLE";
controlImageComputation?: boolean;
subjectType: "SUBJECT_TYPE_PERSON" | "SUBJECT_TYPE_ANIMAL" | "SUBJECT_TYPE_PRODUCT" | "SUBJECT_TYPE_DEFAULT";
}
export interface VertexAIClaudeOptions {
_option_id: "vertexai-claude";
max_tokens?: number;
temperature?: number;
top_p?: number;
top_k?: number;
stop_sequence?: string[];
thinking_mode?: boolean;
thinking_budget_tokens?: number;
}
export declare function getVertexAiOptions(model: string, option?: ModelOptions): ModelOptionsInfo;
//# sourceMappingURL=vertexai.d.ts.map