88 lines
3.3 KiB
Go
88 lines
3.3 KiB
Go
package models
|
||
|
||
import "time"
|
||
|
||
// SubmitTaskRequest 提交任务的请求结构体
|
||
type SubmitTaskRequest struct {
|
||
RunConfig RunConfig `json:"run_config"` // 运行配置
|
||
CodeID int `json:"code_id"` // 上传代码返回的ID
|
||
ModelID int `json:"model_id"` // 上传模型返回的ID
|
||
DatasetID int `json:"dataset_id"` // 数据集ID（可选）
|
||
}
|
||
|
||
// UploadCodeRequest 上传代码的请求结构体
|
||
type UploadCodeRequest struct {
|
||
RunConfig RunConfig `json:"run_config"` // 运行配置
|
||
}
|
||
|
||
// InferenceSubmitTaskRequest 推理任务提交请求结构体
|
||
type InferenceSubmitTaskRequest struct {
|
||
Version string `json:"version"` // 版本
|
||
Description string `json:"description"` // 描述
|
||
Model DataResourceConfig `json:"model"` // 模型信息
|
||
SubModel *DataResourceConfig `json:"sub_model"` // 增量模型信息（可选）
|
||
Image ImageInfo `json:"image"` // 镜像信息
|
||
CodeConfig CodeConfig `json:"code_config"` // 代码配置
|
||
ResourceType string `json:"resource_type"` // 资源类型
|
||
Resource RemoteSourceConfig `json:"resource"` // 资源配置
|
||
Command string `json:"command"` // 启动命令
|
||
}
|
||
|
||
type RemoteSourceConfig struct {
|
||
Id int `json:"id"`
|
||
SourceKey string `json:"sourceKey"`
|
||
Type string `json:"type"`
|
||
Name string `json:"name"`
|
||
TotalCount int `json:"totalCount"`
|
||
AvailableCount int `json:"availableCount"`
|
||
ChangeType int `json:"changeType"`
|
||
Status int `json:"status"`
|
||
Region string `json:"region"`
|
||
ClusterID string `json:"clusterId"`
|
||
CostPerUnit int `json:"costPerUnit"`
|
||
CostType string `json:"costType"`
|
||
Tag string `json:"tag"`
|
||
UserId string `json:"userId"`
|
||
CreateTime time.Time `json:"createTime"`
|
||
UpdateTime time.Time `json:"updateTime"`
|
||
BaseResourceSpecs []BaseResourceSpec
|
||
}
|
||
|
||
type BaseResourceSpec struct {
|
||
Id int `json:"id"`
|
||
ResourceSpecId int `json:"resourceSpecId"`
|
||
Type string `json:"type"`
|
||
Name string `json:"name"`
|
||
TotalValue int `json:"totalValue"`
|
||
TotalUnit string `json:"totalUnit"`
|
||
AvailableValue int `json:"availableValue"`
|
||
AvailableUnit string `json:"availableUnit"`
|
||
UserId string `json:"userId"`
|
||
CreateTime time.Time `json:"createTime"`
|
||
UpdateTime time.Time `json:"updateTime"`
|
||
}
|
||
|
||
// ImageInfo 镜像信息
|
||
type ImageInfo struct {
|
||
ImageID int `json:"imageID"`
|
||
Name string `json:"name"`
|
||
CreateTime string `json:"createTime"`
|
||
ClusterImages []ClusterImage `json:"clusterImages"`
|
||
}
|
||
|
||
// ClusterImage 集群镜像信息
|
||
type ClusterImage struct {
|
||
ImageID int `json:"imageID"`
|
||
ClusterID string `json:"clusterID"`
|
||
OriginImageType string `json:"originImageType"`
|
||
OriginImageID string `json:"originImageID"`
|
||
OriginImageName string `json:"originImageName"`
|
||
Cards []Card `json:"cards"`
|
||
}
|
||
|
||
// Card 卡片信息
|
||
type Card struct {
|
||
OriginImageID string `json:"originImageID"`
|
||
Card string `json:"card"`
|
||
}
|