52 lines
1.9 KiB
Go
52 lines
1.9 KiB
Go
package models
|
||
|
||
// 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 ResourceConfig `json:"resource"` // 资源配置
|
||
Command string `json:"command"` // 启动命令
|
||
}
|
||
|
||
// 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"`
|
||
}
|