bigmodels

type CloudConfig struct {
	Domain       string `json:"domain"                 required:"true"`
	User         string `json:"user"                   required:"true"`
	Password     string `json:"password"               required:"true"`
	Project      string `json:"project"                required:"true"`
	AuthEndpoint string `json:"auth_endpoint"          required:"true"`
}

type Config struct {
	OBS        OBSConfig   `json:"obs"             required:"true"`
	Cloud      CloudConfig `json:"cloud"           required:"true"`
	WuKong     WuKong      `json:"wukong"          required:"true"`
	Endpoints  Endpoints   `json:"endpoints"       required:"true"`
	Moderation Moderation  `json:"moderation"      required:"true"`

	MaxPictureSizeToDescribe int64 `json:"max_picture_size_to_describe"`
	MaxPictureSizeToVQA      int64 `json:"max_picture_size_to_vqa"`
}

type Endpoints struct {
	VQA              string `json:"vqa"                required:"true"`
	Pangu            string `json:"pangu"              required:"true"`
	LuoJia           string `json:"luojia"             required:"true"`
	WuKong           string `json:"wukong"             required:"true"`
	CodeGeex         string `json:"codegeex"           required:"true"`
	DescPicture      string `json:"desc_picture"       required:"true"`
	SinglePicture    string `json:"single_picture"     required:"true"`
	MultiplePictures string `json:"multiple_pictures"  required:"true"`
}

type Moderation struct {
	Endpoint    string `json:"endpoint"       required:"true"`
	AccessKey   string `json:"access_key"     required:"true"`
	SecretKey   string `json:"secret_key"     required:"true"`
	IAMEndpoint string `json:"iam_endpoint"   required:"true"`
	Region      string `json:"region"         required:"true"`
}

type OBSAuthInfo struct {
	Endpoint  string `json:"endpoint"                  required:"true"`
	AccessKey string `json:"access_key"                required:"true"`
	SecretKey string `json:"secret_key"                required:"true"`
}

type OBSConfig struct {
	OBSAuthInfo

	VQABucket    string `json:"vqa_bucket"             required:"true"`
	LuoJiaBucket string `json:"luo_jia_bucket"         required:"true"`
}

type WuKong struct {
	WuKongSample
	CloudConfig
	OBSAuthInfo

	Bucket string `json:"bucket"             required:"true"`

	// DownloadExpiry specifies the timeout to download a obs file.
	// The unit is second.
	DownloadExpiry int `json:"download_expiry"`
}

type WuKongSample struct {
	SampleId    string `json:"sample_id"     required:"true"`
	SampleNum   int    `json:"sample_num"    required:"true"`
	SampleCount int    `json:"sample_count"  required:"true"`
}