package public

const (
	CallModeSync   = 0 // 同步调用
	CallModeAsync  = 1 // 异步调用
	CallModeStream = 2 // 流式调用
)

const (
	BuildTypePrompt = 1 //提示词构建
	BuildTypeNode   = 2 //节点构建
	BuildTypeStruct = 3 //结构构建
)

// ModelType 模型类型常量
const (
	ModelTypeInference = 100 // 推理模型

	ModelTypeImage               = 200 // 图片模型
	ImageSubTypeTextToImage      = 201 // 图片模型-文生图
	ImageSubTypeImageToImage     = 202 // 图片模型-图生图
	ImageSubTypeImageEdit        = 203 // 图片模型-图片编辑
	ImageSubTypeImageVariation   = 204 // 图片模型-图片变体
	ImageSubTypeImageTextToImage = 205 // 图片模型-图文生图

	ModelTypeAudio             = 300 // 音频模型
	AudioSubTypeTextToSpeech   = 301 // 音频模型-文生音
	AudioSubTypeSpeechToText   = 302 // 音频模型-音生文
	AudioSubTypeSpeechToSpeech = 303 // 音频模型-音生音

	ModelTypeVector        = 400 // 向量化模型
	VectorSubTypeEmbedding = 401 // 向量化模型-文本嵌入
	VectorSubTypeRerank    = 402 // 向量化模型-重排序

	ModelTypeOmni             = 500 // 全模态模型
	OmniSubTypeTextImageAudio = 501 // 全模态模型-文图音
	OmniSubTypeVision         = 502 // 全模态模型-视觉理解

	ModelTypeVideo               = 600 // 视频模型
	VideoSubTypeTextToVideo      = 601 // 视频模型-文生视频
	VideoSubTypeImageToVideo     = 602 // 视频模型-图生视频
	VideoSubTypeImageTextToVideo = 603 // 视频模型-图文生视频
	VideoSubTypeVideoToVideo     = 604 // 视频模型-视频生视频
)

// ModelTypeName 模型类型名称映射
var ModelTypeName = map[int]string{
	ModelTypeInference: "推理模型",

	ModelTypeImage:               "图片模型",
	ImageSubTypeTextToImage:      "图片模型-文生图",
	ImageSubTypeImageToImage:     "图片模型-图生图",
	ImageSubTypeImageEdit:        "图片模型-图片编辑",
	ImageSubTypeImageVariation:   "图片模型-图片变体",
	ImageSubTypeImageTextToImage: "图片模型-图文生图",

	ModelTypeAudio:             "音频模型",
	AudioSubTypeTextToSpeech:   "音频模型-文生音",
	AudioSubTypeSpeechToText:   "音频模型-音生文",
	AudioSubTypeSpeechToSpeech: "音频模型-音生音",

	ModelTypeVector:        "向量化模型",
	VectorSubTypeEmbedding: "向量化模型-文本嵌入",
	VectorSubTypeRerank:    "向量化模型-重排序",

	ModelTypeOmni:             "全模态模型",
	OmniSubTypeTextImageAudio: "全模态模型-文图音",
	OmniSubTypeVision:         "全模态模型-视觉理解",

	ModelTypeVideo:               "视频模型",
	VideoSubTypeTextToVideo:      "视频模型-文生视频",
	VideoSubTypeImageToVideo:     "视频模型-图生视频",
	VideoSubTypeImageTextToVideo: "视频模型-图文生视频",
	VideoSubTypeVideoToVideo:     "视频模型-视频生视频",
}

// 运营商常量
const (
	OperatorAliyun     = "阿里云百炼"
	OperatorVolcengine = "火山引擎"
	OperatorTencent    = "腾讯云"
	OperatorHuawei     = "华为云"
	OperatorBaidu      = "百度智能云"
	OperatorOpenAI     = "OpenAI"
	OperatorAzure      = "Azure OpenAI"
	OperatorAWS        = "AWS Bedrock"
	OperatorGoogle     = "Google Cloud"
	OperatorDeepSeek   = "DeepSeek"
	OperatorMoonshot   = "Moonshot"
	OperatorZhipu      = "智谱AI"
	OperatorBaichuan   = "百川智能"
	OperatorMinimax    = "MiniMax"
	OperatorXunfei     = "科大讯飞"
	OperatorOthers     = "其他"
)

// OperatorList 运营商列表（供前端下拉框使用）
var OperatorList = []string{
	OperatorAliyun,
	OperatorVolcengine,
	OperatorTencent,
	OperatorHuawei,
	OperatorBaidu,
	OperatorOpenAI,
	OperatorAzure,
	OperatorAWS,
	OperatorGoogle,
	OperatorDeepSeek,
	OperatorMoonshot,
	OperatorZhipu,
	OperatorBaichuan,
	OperatorMinimax,
	OperatorXunfei,
	OperatorOthers,
}