@lobehub/chat
Version:
Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.
255 lines (229 loc) • 7.04 kB
text/typescript
/* eslint-disable sort-keys-fix/sort-keys-fix, typescript-sort-keys/interface */
import type { Simplify } from 'type-fest';
import { z } from 'zod';
export const MAX_SEED = 2 ** 31 - 1;
/**
* 默认宽高比，当模型不支持原生宽高比时使用
*/
export const DEFAULT_ASPECT_RATIO = '1:1';
export const PRESET_ASPECT_RATIOS = [
DEFAULT_ASPECT_RATIO, // '1:1' - 正方形，最常用
'16:9', // 现代显示器/电视/视频标准
'9:16', // 手机竖屏/短视频
'4:3', // 传统显示器/照片
'3:4', // 传统竖屏照片
'3:2', // 经典照片比例横屏
'2:3', // 经典照片比例竖屏
];
/**
* Image generation and processing configuration constants
*/
export const IMAGE_GENERATION_CONFIG = {
/**
* Maximum cover image size in pixels (longest edge)
* Used for generating cover images from source images
*/
COVER_MAX_SIZE: 256,
/**
* Maximum thumbnail size in pixels (longest edge)
* Used for generating thumbnail images from original images
*/
THUMBNAIL_MAX_SIZE: 512,
} as const;
/**
* Default dimension constraints for image upload auto-setting
* Used when model schema doesn't provide min/max values
*/
export const DEFAULT_DIMENSION_CONSTRAINTS = {
MAX_SIZE: 1024,
MIN_SIZE: 512,
} as const;
export const CHAT_MODEL_IMAGE_GENERATION_PARAMS: ModelParamsSchema = {
imageUrls: {
default: [],
},
prompt: { default: '' },
};
// 定义顶层的元规范 - 平铺结构
export const ModelParamsMetaSchema = z.object({
/**
* Prompt 是唯一一个每个模型都有的参数
*/
prompt: z.object({
default: z.string().optional().default(''),
description: z.string().optional(),
type: z.literal('string').optional(),
}),
imageUrl: z
.object({
default: z.string().nullable().optional(),
description: z.string().optional(),
maxFileSize: z.number().optional(),
type: z.tuple([z.literal('string'), z.literal('null')]).optional(),
})
.optional(),
imageUrls: z
.object({
default: z.array(z.string()),
description: z.string().optional(),
maxCount: z.number().optional(),
/**
* The maximum file size in bytes
*/
maxFileSize: z.number().optional(),
type: z.literal('array').optional(),
})
.optional(),
width: z
.object({
default: z.number(),
description: z.string().optional(),
max: z.number(),
min: z.number(),
step: z.number().optional().default(1),
type: z.literal('number').optional(),
})
.optional(),
/**
* samplerName is not requires by all i2i providers
*/
samplerName: z
.object({
default: z.string(),
description: z.string().optional(),
enum: z.array(z.string()).optional(),
type: z.literal('string').optional(),
})
.optional(),
/**
* scheduler is not requires by all i2i providers
*/
scheduler: z
.object({
default: z.string(),
description: z.string().optional(),
enum: z.array(z.string()).optional(),
type: z.literal('string').optional(),
})
.optional(),
height: z
.object({
default: z.number(),
description: z.string().optional(),
max: z.number(),
min: z.number(),
step: z.number().optional().default(1),
type: z.literal('number').optional(),
})
.optional(),
size: z
.object({
default: z.string(),
description: z.string().optional(),
enum: z.array(z.string()),
type: z.literal('string').optional(),
})
.optional(),
aspectRatio: z
.object({
default: z.string(),
description: z.string().optional(),
enum: z.array(z.string()),
type: z.literal('string').optional(),
})
.optional(),
cfg: z
.object({
default: z.number(),
description: z.string().optional(),
max: z.number(),
min: z.number(),
step: z.number(),
type: z.literal('number').optional(),
})
.optional(),
/**
* strength/denoise is optional for t2i but must be used for i2i
*/
strength: z
.object({
default: z.number(),
description: z.string().optional(),
max: z.number().optional().default(1),
min: z.number().optional().default(0),
step: z.number().optional().default(0.05),
type: z.literal('number').optional(),
})
.optional(),
steps: z
.object({
default: z.number(),
description: z.string().optional(),
max: z.number(),
min: z.number(),
step: z.number().optional().default(1),
type: z.literal('number').optional(),
})
.optional(),
quality: z
.object({
default: z.string(),
description: z.string().optional(),
enum: z.array(z.string()),
type: z.literal('string').optional(),
})
.optional(),
seed: z
.object({
default: z.number().nullable().default(null),
description: z.string().optional(),
max: z.number().optional().default(MAX_SEED),
min: z.number().optional().default(0),
type: z.tuple([z.literal('number'), z.literal('null')]).optional(),
})
.optional(),
});
// 导出推断出的类型，供定义对象使用
export type ModelParamsSchema = z.input<typeof ModelParamsMetaSchema>;
export type ModelParamsOutputSchema = z.output<typeof ModelParamsMetaSchema>;
export type ModelParamsKeys = Simplify<keyof ModelParamsOutputSchema>;
type TypeMapping<T> = T extends 'string'
? string
: T extends 'number'
? number
: T extends ['number', 'null']
? number | null
: T extends ['string', 'null']
? string | null
: T extends 'string'
? string
: T extends 'boolean'
? boolean
: never;
type TypeType<K extends ModelParamsKeys> = NonNullable<ModelParamsOutputSchema[K]>['type'];
type DefaultType<K extends ModelParamsKeys> = NonNullable<ModelParamsOutputSchema[K]>['default'];
type _StandardImageGenerationParameters<P extends ModelParamsKeys = ModelParamsKeys> = {
[]: NonNullable<TypeType<key>> extends 'array'
? DefaultType<key>
: TypeMapping<TypeType<key>>;
};
export type RuntimeImageGenParams = Pick<_StandardImageGenerationParameters, 'prompt'> &
Partial<Omit<_StandardImageGenerationParameters, 'prompt'>>;
export type RuntimeImageGenParamsKeys = keyof RuntimeImageGenParams;
export type RuntimeImageGenParamsValue = RuntimeImageGenParams[RuntimeImageGenParamsKeys];
// 验证函数
export function validateModelParamsSchema(paramsSchema: unknown): ModelParamsOutputSchema {
return ModelParamsMetaSchema.parse(paramsSchema);
}
/**
* 从参数定义对象提取默认值
*/
export function extractDefaultValues(paramsSchema: ModelParamsSchema) {
// 部分默认值从 ModelParamsMetaSchema 中获取
const schemaWithDefault = ModelParamsMetaSchema.parse(paramsSchema);
return Object.fromEntries(
Object.entries(schemaWithDefault).map(([key, value]) => {
return [key, value.default];
}),
) as RuntimeImageGenParams;
}