import { ApiProperty } from "@nestjs/swagger";
import { IsEnum, IsIn, IsNotEmpty, IsString } from "class-validator";
import { TAIModel } from "./subscriptions.interface";

export class ChangeModelDto {
	@ApiProperty({ example: 'gpt-3.5' })
	@IsString()
	@IsNotEmpty()
	// @IsIn(AI_MODEL_LIST)
	model: TAIModel
}



export class ChangePlanDto {
	@ApiProperty({ example: 'free' })
	@IsString()
	@IsNotEmpty()
	@IsIn(['free', 'premium'])
	plan: 'free' | 'premium'
}



export class UpdateFeatureSettingsDto {
	@ApiProperty({ example: 'imageGeneration' })
	@IsString()
	@IsNotEmpty()
	@IsEnum(['imageGeneration', 'realtimeSearch'])
	feature: 'imageGeneration' | 'realtimeSearch';
}