pcm-octopus/pb/octopus.proto

syntax = "proto3";

package octopus;
option go_package = "/octopus";


message resourceReq{
  string platform =1;
}

message cpResp{
  float pOpsAtFp16 = 1;
}

message giResp{
  int32 cpuCoreNum = 1;
  int32 memoryInGib = 2;
}

/******************Algorithm Start*************************/
message GetAlgorithmReq{
  string platform =1;
  string algorithmId = 2;
  string version = 3;
}

message GetAlgorithmResp{
  bool success =1;
  PayloadGetAlgorithm payload =2;
  Error error = 3;
}

message PayloadGetAlgorithm{
  Algorithms algorithm = 1;
  repeated VersionAccesses versionAccesses = 2;
}

message VersionAccesses{
  string algorithmId = 1;
  string spaceId = 2;
  string version = 3;
}

message DownloadCompressReq{
  string platform =1;
  string algorithmId = 2;
  string version = 3;
}

message DownloadCompressResp{
  bool success =1;
  PayloadDownloadCompress payload =2;
  Error error = 3;
}

message PayloadDownloadCompress{
  int64 compressAt = 1;
}

message DownloadAlgorithmReq{
  string platform =1;
  string algorithmId = 2;
  string version = 3;
  int64 compressAt = 4;
  string domain = 5;
}

message DownloadAlgorithmResp{
  bool success =1;
  PayloadDownloadAlgorithm payload =2;
  Error error = 3;
}

message PayloadDownloadAlgorithm{
  string downloadUrl = 1;
}

message AlgorithmUrlReq {
  string platform =1;
  string url = 2;
}

message AlgorithmUrlResp {
  string algorithm = 1;
}

message UploadAlgorithmReq{
  string platform =1;
  string algorithmId = 2;
  string version = 3;
  UploadAlgorithmParam params = 4;
}

message UploadAlgorithmParam{
  string domain = 1;
  string fileName = 2;
}

message UploadAlgorithmResp{
  bool success =1;
  PayloadUploadAlgorithm payload =2;
  Error error = 3;
}

message PayloadUploadAlgorithm{
  string uploadUrl = 1;
}

message UploadAlgorithmConfirmReq{
  string platform =1;
  string algorithmId = 2;
  string version = 3;
  UploadAlgorithmConfirmParam params = 4;
}

message UploadAlgorithmConfirmParam{
  string fileName = 1;
}

message UploadAlgorithmConfirmResp{
  bool success =1;
  PayloadUploadAlgorithmConfirm payload =2;
  Error error = 3;
}

message PayloadUploadAlgorithmConfirm{
  int64 updatedAt = 1;
}

message GetAlgorithmListReq{
  string platform =1;
  string algorithmId = 2;
  int32 pageIndex =3;
  int32 pageSize =4;
}

message GetAlgorithmListResp{
  bool success =1;
  PayloadAlgorithmList payload =2;
  Error error = 3;
}

message PayloadAlgorithmList{
  int32 totalSize =1;
  repeated AlgorithmDetail algorithms = 2;
}

message AlgorithmDetail{
  Algorithms algorithmDetail = 1;
  bool isShared =2;
}

message GetMyAlgorithmListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetMyAlgorithmListResp{
  bool success =1;
  PayloadMyAlgorithmList payload =2;
  Error error = 3;
}

message PayloadMyAlgorithmList{
  int32 totalSize =1;
  repeated Algorithms algorithms = 2;
}

message Algorithms{
  string algorithmId =1;
  string algorithmVersion = 2;
  string spaceId = 3;
  string spaceName = 4;
  string userId = 5;
  string userName=6;
  string algorithmName=7;
  string modelName=8;
  int32 fileStatus=9;
  int32 latestCompressed=10;
  string algorithmDescript=11;
  string path=12;
  bool isPrefab=13;
  int64 createdAt=14;
  string applyId=15;
  string applyName=16;
  string frameworkId=17;
  string frameworkName=18;
}

message GetAlgorithmApplyListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetAlgorithmApplyListResp{
  bool success =1;
  PayloadGetAlgorithmApplyList payload =2;
  Error error = 3;
}

message PayloadGetAlgorithmApplyList{
  int32 totalSize = 1;
  repeated Lables lables =2;
}

message GetAlgorithmFrameworkListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetAlgorithmFrameworkListResp{
  bool success =1;
  PayloadAlgorithmFrameworkList payload =2;
  Error error = 3;
}

message PayloadAlgorithmFrameworkList{
  int32 totalSize = 1;
  repeated Lables lables =2;
}

message Lables{
  string id = 1;
  string lableDesc = 2;
  int32 sourceType = 3;
}

message DeleteMyAlgorithmReq{
  string platform =1;
  string algorithmId=2;
}

message DeleteMyAlgorithmResp{
  bool success =1;
  PayloadDeleteMyAlgorithm payload =2;
  Error error = 3;
}

message PayloadDeleteMyAlgorithm{
  int64 deletedAt=1;
}

message CreateMyAlgorithmReq{
  string platform =1;
  CreateMyAlgorithm createMyAlgorithm=2;
}

message CreateMyAlgorithmResp{
  bool success =1;
  PayloadCreateMyAlgorithm payload =2;
  Error error = 3;
}

message CreateMyAlgorithm{
  string algorithmDescript = 1;
  string algorithmName =2;
  string applyId =3;
  string frameworkId =4;
  bool isEmpty =5;
  string modelName=6;
}

message PayloadCreateMyAlgorithm{
  string algorithmId =1;
  int64 createdAt =2;
  string version =3;
}

/******************Algorithm End*************************/

/******************DatasetService Start*************************/
message GetDatasetVersionListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
  string datasetId = 4;
}

message GetDatasetVersionListResp{
  bool success =1;
  PayloadGetDatasetVersion payload =2;
  Error error = 3;
}

message PayloadGetDatasetVersion{
  int32 totalSize =1;
  repeated DatasetVersion versions = 2;
}

message DatasetVersion{
  int64 createdAt = 1;
  string datasetId = 2;
  string desc = 3;
  bool shared = 4;
  int32 status = 5;
  int32 updatedAt = 6;
  string version = 7;
}

message CreateDataSetReq{
  string platform =1;
  CreateDataSet  createDataSet=2;
}

message CreateDataSetResp{
  bool success =1;
  PayloadCreateDataSet payload =2;
  Error error = 3;
}

message CreateDataSet{
  repeated string applyIds = 1;
  string desc = 2;
  string name = 3;
  string typeId = 4;
}

message PayloadCreateDataSet{
  string id = 1;
  string version = 2;
}

message GetMyDatasetListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetMyDatasetListResp{
  bool success =1;
  PayloadMyDatasetList payload =2;
  Error error = 3;
}

message PayloadMyDatasetList{
  int32 totalSize =1;
  repeated Datasets datasets = 2;
}

message Datasets{
  int64 createdAt =1;
  int64 updatedAt=2;
  string id=3;
  int32 sourceType=4;
  string name=5;
  string typeId=6;
  string typeDesc=7;
  repeated Applies applies=8;
  string desc=9;
  string latestVersion=10;
  string userName = 11;
}
message Applies{
  string id=1;
  string desc=2;
}

message DeleteDataSetReq{
  string platform =1;
  string id=2;
}

message DeleteDataSetResp{
  bool success =1;
  PayloadDeleteDataSet payload =2;
  Error error = 3;
}

message PayloadDeleteDataSet{
  int64 deletedAt = 1;
}

message UploadDataSetReq{
  string platform =1;
  string datasetId=2;
  string version=3;
  UploadDataSetParam params = 4;
}

message UploadDataSetParam{
  string domain = 1;
  string fileName = 2;
}

message UploadDataSetResp{
  bool success =1;
  PayloadUploadDataSet payload =2;
  Error error = 3;
}

message PayloadUploadDataSet{
  string uploadUrl = 1;
}

message UploadDataSetConfirmReq{
  string platform =1;
  string datasetId=2;
  string version=3;
  UploadDataSetConfirmParam params = 4;
}

message UploadDataSetConfirmParam{
  string fileName = 1;
}

message UploadDataSetConfirmResp{
  bool success =1;
  PayloadUploadDataSetConfirm payload =2;
  Error error = 3;
}

message PayloadUploadDataSetConfirm{
  int64 updatedAt = 1;
}

message CreateDataSetVersionReq{
  string platform =1;
  string datasetId=2;
  CreateDataSetVersionParam params =3;
}

message CreateDataSetVersionParam{
  string desc =1;
}

message CreateDataSetVersionResp{
  bool success =1;
  PayloadCreateDataSetVersion payload =2;
  Error error = 3;
}

message PayloadCreateDataSetVersion {
  string datasetId = 1;
  string version = 2;
}

message DeleteDataSetVersionReq{
  string platform =1;
  string datasetId=2;
  string version = 3;
}

message DeleteDataSetVersionResp{
  bool success =1;
  PayloadDeleteDataSetVersion payload =2;
  Error error = 3;
}

message PayloadDeleteDataSetVersion{
  int64 deletedAt = 1;
}

message GetDatasetApplyListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetDatasetApplyListResp{
  bool success =1;
  PayloadGetDatasetApplyList payload =2;
  Error error = 3;
}

message PayloadGetDatasetApplyList{
  int32 totalSize = 1;
  repeated Lables lables =2;
}

message GetDatasetTypeListRep{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetDatasetTypeListResp{
  bool success =1;
  PayloadGetDatasetTypeList payload =2;
  Error error = 3;
}

message PayloadGetDatasetTypeList{
  int32 totalSize = 1;
  repeated Lables lables =2;
}


/******************DatasetService End*************************/

/******************ModelDeployService Start*************************/
message CreateModelDeployReq{
  string platform =1;
  CreateModelDeployParam params = 2;
}

message CreateModelDeployParam{
  string desc = 1;
  string domain = 2;
  string modelFrame = 3;
  string modelId = 4;
  string modelVersion = 5;
  string name = 6;
  string resourcePool = 7;
  string resourceSpecId = 8;
  string resourceType = 9;
  string serviceType = 10;
}

message CreateModelDeployResp{
  bool success =1;
  PayloadCreateModelDeploy payload =2;
  Error error = 3;
}

message PayloadCreateModelDeploy{
  string message = 1;
  string serviceId = 2;
  string serviceUrl = 3;
}

message GetModelDeployListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetModelDeployListResp{
  bool success =1;
  PayloadGetModelDeployList payload =2;
  Error error = 3;
}

message PayloadGetModelDeployList{
  int32 totalSize = 1;
  repeated DepInfo depInfos = 2;
}

message DepInfo{
  int64 completedAt = 1;
  int64 createdAt = 2;
  string desc = 3;
  string id = 4;
  string modelFrame = 5;
  string modelId = 6;
  string modelName = 7;
  string modelVersion = 8;
  string name = 9;
  int64 runSec = 10;
  string serviceUrl = 11;
  int64 startedAt = 12;
  string status = 13;
  int64 updatedAt = 14;
  string userId = 15;
  string workspaceId = 16;
}

message GetModelDeployReq{
  string platform =1;
  string id =2;
}

message GetModelDeployResp{
  bool success =1;
  PayloadGetModelDeploy payload =2;
  Error error = 3;
}

message PayloadGetModelDeploy{
  DepInfo depInfo = 1;
}

message GetModelDeployEventReq{
  string platform =1;
  string id = 2;
  bool isMain = 3;
  int32 pageIndex =4;
  int32 pageSize =5;
}

message GetModelDeployEventResp{
  bool success =1;
  PayloadGetModelDeployEvent payload =2;
  Error error = 3;
}

message PayloadGetModelDeployEvent{
  int32 totalSize =1;
  repeated DepEvent depEvents = 2;
}

message DepEvent{
  string message = 1;
  string name = 2;
  string reason = 3;
  string timestamp = 4;
}

message StopModelDeployReq{
  string platform =1;
  string id = 2;
}

message StopModelDeployResp{
  bool success =1;
  PayloadStopModelDeploy payload =2;
  Error error = 3;
}

message PayloadStopModelDeploy{
  int64 stoppedAt = 1;
}

message DeleteModelDeployReq{
  string platform =1;
  repeated string jobIds = 2;
}

message DeleteModelDeployResp{
  bool success =1;
  PayloadDeleteModelDeploy payload =2;
  Error error = 3;
}

message PayloadDeleteModelDeploy{
  int64 deletedAt = 1;
}

message InferModelDeployReq{
  string platform =1;
}

message InferModelDeployResp{
  bool success =1;
  PayloadInferModelDeploy payload =2;
  Error error = 3;
}

message PayloadInferModelDeploy{
  int64 stoppedAt = 1;
}

/******************ModelDeployService End*************************/

/******************Develop Start*************************/
message CreateNotebookReq{
  string platform =1;
  CreateNotebookParam params = 2;
}

message CreateNotebookParam{
  string algorithmId = 1;
  string algorithmVersion = 2;
  string command = 3;
  string datasetId = 4;
  string datasetVersion = 5;
  string desc = 6;
  map<string, string> envs = 7;
  string imageId = 8;
  string imageUrl = 9;
  repeated Mounts mounts = 10;
  string name = 11;
  string resourcePool = 12;
  string resourceSpecId = 13;
  int32 taskNumber = 14;
}

message CreateNotebookResp{
  bool success =1;
  PayloadCreateNotebook payload =2;
  Error error = 3;
}

message PayloadCreateNotebook{
  string id = 1;
}

message GetNotebookReq{
  string platform =1;
  string id =2;
}

message GetNotebookResp{
  bool success =1;
  PayloadGetNotebook payload =2;
  Error error = 3;
}

message PayloadGetNotebook{
  Notebook notebook = 1;
}

message DeleteNotebookReq{
  string platform =1;
  string id =2;
}

message DeleteNotebookResp{
  bool success =1;
  PayloadDeleteNotebook payload =2;
  Error error = 3;
}

message PayloadDeleteNotebook{
  string id = 1;
}

message GetNotebookListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetNotebookListResp{
  bool success =1;
  PayloadNotebookList payload =2;
  Error error = 3;
}

message PayloadNotebookList{
  int32 totalSize =1;
  repeated Notebook notebooks = 2;
}

message Notebook{
  int64 createdAt =1;
  int64 updatedAt =2;
  string id = 3;
  string userId =4;
  string workspaceId=5;
  string name=6;
  string desc=7;
  string imageId=8;
  string imageName=9;
  string algorithmId=10;
  string algorithmVersion=11;
  string algorithmName=12;
  string resourceSpecId=13;
  string resourceSpecName=14;
  string status=15;
  string datasetId=16;
  string datasetVersion=17;
  string datasetName=18;
  int32 resourceSpecPrice=19;
  string notebookJobId=20;
  string imageVersion=21;
  repeated Tasks tasks=22;
  string imageUrl=23;
}

message Tasks{
  string url =1;
  string name= 2;
}

message StartNotebookReq{
  string platform =1;
  string id =2;
}

message StartNotebookResp{
  bool success =1;
  PayloadStartNotebook payload =2;
  Error error = 3;
}

message PayloadStartNotebook{
  string id = 1;
}

message StopNotebookReq{
  string platform =1;
  string id =2;
}

message StopNotebookResp{
  bool success =1;
  PayloadStopNotebook payload =2;
  Error error = 3;
}

message PayloadStopNotebook{
  string id = 1;
}


/******************Develop End*************************/

/******************ImageService Start*************************/
message GetUserImageListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetUserImageListResp{
  bool success =1;
  PayloadUserImageList payload =2;
  Error error = 3;
}

message GetPresetImageListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetPresetImageListResp{
  bool success =1;
  PayloadGetPresetImageList payload =2;
  Error error = 3;
}

message PayloadUserImageList{
  int32 totalSize =1;
  repeated Images images =2;
}

message PayloadGetPresetImageList{
  int32 totalSize =1;
  repeated Image images =2;
}

message Images{
  bool isShared =1;
  Image image =2;
}

message Image{
  string id =1;
  string imageName =2;
  string imageDesc = 3;
  string imageAddr = 4;
  int32 sourceType = 5;
  string spaceId = 6;
  string userId = 7;
  int64 createdAt = 8;
  int64 updatedAt = 9;
  int32 imageStatus = 10;
  string username = 11;
  string imageVersion = 12;
  string imageFullAddr = 13;
}

message DeleteImageReq{
  string platform =1;
  string imageId=2;
}

message DeleteImageResp{
  bool success =1;
  PayloadDeleteImage payload =2;
  Error error = 3;
}

message PayloadDeleteImage{
  int64 deletedAt=1;
}

message CreateImageReq{
  string platform =1;
  CreateImage createImage =2;
}

message CreateImage{
  string imageAddr=2;
  string imageDesc=3;
  string imageName=4;
  string imageVersion=5; //镜像标签
  int32 sourceType=6; //1 = 上传， 2 = 远程
}

message CreateImageResp{
  bool success =1;
  PayloadCreateImage payload =2;
  Error error = 3;
}

message PayloadCreateImage{
  string imageId = 1;
  int64 createdAt=2;
}

message UploadImageReq{
  string platform =1;
  string imageId=2;
  UploadImageParam params = 3;
}

message UploadImageParam{
  string domain = 1;
  string fileName = 2;
}

message UploadImageResp{
  bool success =1;
  PayloadUploadImage payload =2;
  Error error = 3;
}

message PayloadUploadImage{
  string uploadUrl = 1;
  Headers headers = 2;
}

message Headers{
  string additionalProp1=1;
  string additionalProp2=2;
  string additionalProp3=3;
}

message UploadImageConfirmReq{
  string platform =1;
  string imageId=2;
}

message UploadImageConfirmResp{
  bool success =1;
  PayloadUploadImageConfirm payload =2;
  Error error = 3;
}

message PayloadUploadImageConfirm{
  int64 updatedAt = 1;
}

/******************ImageService End*************************/

/******************Model Start*************************/
message GetModelVersionListReq{
  string platform =1;
  string modelId = 2;
  int32 pageIndex = 3;
  int32 pageSize =4;
}

message GetModelVersionListResp{
  bool success =1;
  PayloadGetModelVersionList payload =2;
  Error error = 3;
}

message PayloadGetModelVersionList{
  int32 totalSize = 1;
  repeated ModelVersion modelVersions = 2;
}

message ModelVersion{
  bool isShared = 1;
  VersionDetail versionDetail = 2;
}

message VersionDetail{
  int64 createdAt = 1;
  string descript = 2;
  int32 fileStatus = 3;
  string modelId = 4;
  string version = 5;
}

message DeleteMyModelReq{
  string platform =1;
  string modelId = 2;
}

message DeleteMyModelResp{
  bool success =1;
  PayloadDeleteMyModel payload =2;
  Error error = 3;
}

message PayloadDeleteMyModel{
  int64 deletedAt = 1;
}

message DeleteModelVersionReq{
  string platform =1;
  string modelId = 2;
  string version = 3;
}

message DeleteModelVersionResp{
  bool success =1;
  PayloadDeleteModelVersion payload =2;
  Error error = 3;
}

message PayloadDeleteModelVersion{
  int64 deletedAt = 1;
}

message DownloadModelVersionReq{
  string platform =1;
  string modelId = 2;
  string version = 3;
  string domain = 4;
}

message DownloadModelVersionResp{
  bool success =1;
  PayloadDownloadModelVersion payload =2;
  Error error = 3;
}

message PayloadDownloadModelVersion{
  string downloadUrl = 1;
}

message GetMyModelListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetMyModelListResp{
  bool success =1;
  PayloadGetMyModelList payload =2;
  Error error = 3;
}

message PayloadGetMyModelList{
  int32 totalSize = 1;
  repeated Model models = 2;
}

message Model{
  string algorithmName = 1;
  string algorithmVersion = 2;
  int64 createdAt = 3;
  string frameWorkId = 4;
  string frameWorkName = 5;
  string latestVersion = 6;
  string modelDescript = 7;
  string modelId = 8;
  string modelName = 9;
  string spaceId = 10;
  string spaceName = 11;
  string userName = 12;
}
/******************Model End*************************/

/******************TrainJobService Start*************************/
message GetTrainJobReq{
  string platform =1;
  string id =2;
}

message GetTrainJobResp{
  bool success =1;
  PayloadGetTrainJob payload =2;
  Error error = 3;
}

message PayloadGetTrainJob{
  TrainJob trainJob = 1;
}

message DeleteTrainJobReq{
  string platform =1;
  repeated string jobIds = 2;
}

message DeleteTrainJobResp{
  bool success =1;
  PayloadDeleteTrainJob payload =2;
  Error error = 3;
}

message PayloadDeleteTrainJob{
  int64 deletedAt = 1;
}

message StopTrainJobReq{
  string platform =1;
  string id =2;
}

message StopTrainJobResp{
  bool success =1;
  PayloadStopTrainJob payload =2;
  Error error = 3;
}

message PayloadStopTrainJob{
  int64 stoppedAt = 1;
}

message GetTrainJobEventReq{
  string platform =1;
  string id =2;
  int32 pageIndex =3;
  int32 pageSize =4;
}

message GetTrainJobEventResp{
  bool success =1;
  PayloadGetTrainJobEvent payload =2;
  Error error = 3;
}

message PayloadGetTrainJobEvent{
  int32 totalSize = 1;
  repeated JobEvent jobEvents = 2;
}

message JobEvent{
  string message = 1;
  string name = 2;
  string reason = 3;
  string timestamp = 4;
}

message GetTrainJobMetricReq{
  string platform =1;
  string id =2;
  int64 start = 3;
  int32 size = 4;
  int32 step = 5;
}

message GetTrainJobMetricResp{
  bool success =1;
  PayloadGetTrainJobMetric payload =2;
  Error error = 3;
}

message PayloadGetTrainJobMetric{
  repeated int64 cpuUsage = 1;
  repeated int64 gpuMemUsage = 2;
  repeated int64 gpuUtil = 3;
  repeated int64 memUsage = 4;
}

message GetTrainJobListReq{
  string platform =1;
  int32 pageIndex =2;
  int32 pageSize =3;
}

message GetTrainJobListResp{
  bool success =1;
  PayloadGetTrainJobList payload =2;
  Error error = 3;
}

message PayloadGetTrainJobList{
  int32 totalSize = 1;
  repeated TrainJob trainJobs = 2;
}

message TrainJob{
  string algorithmId = 1;
  string algorithmName = 2;
  string algorithmVersion = 3;
  int64 completedAt = 4;
  repeated Config config = 5;
  int64 createdAt = 6;
  string dataSetId = 7;
  string dataSetName = 8;
  string dataSetVersion = 9;
  string desc = 10;
  string id = 11;
  string imageId = 12;
  string imageName = 13;
  string imageUrl = 14;
  string imageVersion = 15;
  bool isDistributed = 16;
  string name = 17;
  string resourcePool = 18;
  int64 runSec = 19;
  int64 startedAt = 20;
  string status = 21;
  int64 updatedAt = 22;
  string userId = 23;
  string workspaceId = 24;
}

message CreateTrainJobReq{
  string platform =1;
  CreateTrainJobParam params = 2;
}

message CreateTrainJobResp{
  bool success =1;
  PayloadCreateTrainJob payload =2;
  Error error = 3;
}

message CreateTrainJobParam{
  string algorithmId = 1;
  string algorithmVersion = 2;
  repeated Config config = 3;
  string dataSetId = 4;
  string dataSetVersion = 5;
  string desc = 6;
  string imageId = 7;
  string imageUrl = 8;
  bool isDistributed = 9;
  repeated Mounts mounts = 10;
  string name = 11;
  string resourcePool = 12;
}

message Config{
  string command = 1;
  map<string, string> envs = 2;
  bool isMainRole = 3;
  int32 minFailedTaskCount = 4;
  int32 minSucceededTaskCount =5;
  string name = 6;
  repeated Parameters parameters = 7;
  repeated ReplicaStates replicaStates = 8;
  string resourceSpecId = 9;
  string resourceSpecName = 10;
  int32 resourceSpecPrice = 11;
  string subTaskState = 12;
  int32 taskNumber = 13;
}

message Parameters{
  string key = 1;
  string value = 2;
}

message ReplicaStates{
  string key = 1;
  string state = 2;
}

message Mounts{
  string containerPath = 1;
  Nfs nfs = 2;
  TrainJobOctopus octopus = 3;
  bool readOnly = 4;
}

message PayloadCreateTrainJob{
  string jobId = 1;
}

message Nfs{
  string path = 1;
  string server = 2;
}

message TrainJobOctopus{
  string bucket = 1;
  string object = 2;
}

message GetTrainJobLogReq{
  string platform =1;
  string taskId = 2;
  string taskNum = 3;
  int32 num = 4;

}

message GetTrainJobLogResp{
  string content = 1;
}

/******************TrainJobService End*************************/

/******************ResourceSpecService Start*************************/
message GetResourceSpecsReq{
  string platform =1;
  string resourcePool = 2;
}

message GetResourceSpecsResp{
  bool success =1;
  repeated ResourceSpecs TrainResourceSpecs =2;
  Error error = 3;
}

message ResourceSpecs{
  string id = 1;
  string name = 2;
  double price = 3;
}

message Error{
  int32 code =1;
  int32 subcode =2;
  string message =3;
  string subMessage =4;
}
/******************ResourceSpecService End*************************/

/******************Billing Start*************************/
message GetUserBalanceReq{
  string platform =1;
}

message GetUserBalanceResp{
  bool success =1;
  PayloadBillingUser payload =2;
  Error error = 3;
}

message PayloadBillingUser{
  BillingUser billingUser =1;
}

message BillingUser{
  int64 createdAt = 1;
  int64 updatedAt = 2;
  float amount = 3;
}
/******************Billing End*************************/


service Octopus {

  rpc GetComputingPower(resourceReq) returns (cpResp);

  rpc GetGeneralInfo(resourceReq) returns (giResp);

  //Algorithm
  rpc GetMyAlgorithmList(GetMyAlgorithmListReq) returns (GetMyAlgorithmListResp);
  rpc GetAlgorithmList(GetAlgorithmListReq) returns (GetAlgorithmListResp); //查询算法版本列表
  rpc GetAlgorithm(GetAlgorithmReq) returns (GetAlgorithmResp); //查询算法版本详情
  rpc GetAlgorithmApplyList(GetAlgorithmApplyListReq) returns (GetAlgorithmApplyListResp);
  rpc GetAlgorithmFrameworkList(GetAlgorithmFrameworkListReq) returns (GetAlgorithmFrameworkListResp);
  rpc DeleteMyAlgorithm(DeleteMyAlgorithmReq) returns (DeleteMyAlgorithmResp);
  rpc CreateMyAlgorithm(CreateMyAlgorithmReq) returns (CreateMyAlgorithmResp);
  rpc DownloadAlgorithm(DownloadAlgorithmReq) returns (DownloadAlgorithmResp); //下载算法版本
  rpc DownloadCompress(DownloadCompressReq) returns (DownloadCompressResp);
  rpc DownloadAlgorithmUrl(AlgorithmUrlReq) returns (AlgorithmUrlResp);
  rpc UploadAlgorithm(UploadAlgorithmReq) returns (UploadAlgorithmResp);
  rpc UploadAlgorithmConfirm(UploadAlgorithmConfirmReq) returns (UploadAlgorithmConfirmResp);


  //DatasetService
  rpc GetMyDatasetList(GetMyDatasetListReq) returns (GetMyDatasetListResp);
  rpc GetDatasetVersionList(GetDatasetVersionListReq) returns (GetDatasetVersionListResp);
  rpc CreateDataSet(CreateDataSetReq) returns (CreateDataSetResp);
  rpc DeleteDataSet(DeleteDataSetReq) returns (DeleteDataSetResp);
  rpc UploadDataSet(UploadDataSetReq) returns (UploadDataSetResp);
  rpc UploadDataSetConfirm(UploadDataSetConfirmReq) returns (UploadDataSetConfirmResp);
  rpc CreateDataSetVersion(CreateDataSetVersionReq) returns (CreateDataSetVersionResp); //创建版本
  rpc DeleteDataSetVersion(DeleteDataSetVersionReq) returns (DeleteDataSetVersionResp); //删除版本
  rpc GetDatasetApplyList(GetDatasetApplyListReq) returns (GetDatasetApplyListResp); //查询数据集用途列表
  rpc GetDatasetTypeList(GetDatasetTypeListRep) returns (GetDatasetTypeListResp); //查询数据集类型列表


  //ModelDeployService
  rpc CreateModelDeploy(CreateModelDeployReq) returns (CreateModelDeployResp);
  rpc GetModelDeployList(GetModelDeployListReq) returns (GetModelDeployListResp);
  rpc GetModelDeploy(GetModelDeployReq) returns (GetModelDeployResp);
  rpc GetModelDeployEvent(GetModelDeployEventReq) returns (GetModelDeployEventResp);
  rpc StopModelDeploy(StopModelDeployReq) returns (StopModelDeployResp);
  rpc DeleteModelDeploy(DeleteModelDeployReq) returns (DeleteModelDeployResp);
  rpc InferModelDeploy(InferModelDeployReq) returns (InferModelDeployResp);


  //Develop
  rpc GetNotebookList(GetNotebookListReq) returns (GetNotebookListResp);
  rpc GetNotebook(GetNotebookReq) returns (GetNotebookResp);
  rpc DeleteNotebook(DeleteNotebookReq) returns (DeleteNotebookResp);
  rpc CreateNotebook(CreateNotebookReq) returns (CreateNotebookResp);
  rpc StartNotebook(StartNotebookReq) returns (StartNotebookResp);
  rpc StopNotebook(StopNotebookReq) returns (StopNotebookResp);


  //ImageService
  rpc GetUserImageList(GetUserImageListReq) returns (GetUserImageListResp);
  rpc GetPresetImageList(GetPresetImageListReq) returns (GetPresetImageListResp);
  rpc CreateImage(CreateImageReq) returns (CreateImageResp);
  rpc DeleteImage(DeleteImageReq) returns (DeleteImageResp);
  rpc UploadImage(UploadImageReq) returns (UploadImageResp);
  rpc UploadImageConfirm(UploadImageConfirmReq) returns (UploadImageConfirmResp);


  //Model
  rpc GetMyModelList(GetMyModelListReq) returns (GetMyModelListResp);
  rpc GetModelVersionList(GetModelVersionListReq) returns (GetModelVersionListResp);
  rpc DeleteMyModel(DeleteMyModelReq) returns (DeleteMyModelResp);
  rpc DeleteModelVersion(DeleteModelVersionReq) returns (DeleteModelVersionResp);
  rpc DownloadModelVersion(DownloadModelVersionReq) returns (DownloadModelVersionResp);


  //TrainJobService
  rpc CreateTrainJob(CreateTrainJobReq) returns (CreateTrainJobResp);
  rpc GetTrainJobList(GetTrainJobListReq) returns (GetTrainJobListResp);
  rpc GetTrainJob(GetTrainJobReq) returns (GetTrainJobResp);
  rpc DeleteTrainJob(DeleteTrainJobReq) returns (DeleteTrainJobResp);
  rpc StopTrainJob(StopTrainJobReq) returns (StopTrainJobResp);
  rpc GetTrainJobEvent(GetTrainJobEventReq) returns (GetTrainJobEventResp);
  rpc GetTrainJobMetric(GetTrainJobMetricReq) returns (GetTrainJobMetricResp);
  rpc GetTrainJobLog(GetTrainJobLogReq) returns (GetTrainJobLogResp);

  //ResourceSpecService
  rpc GetResourceSpecs(GetResourceSpecsReq) returns (GetResourceSpecsResp);

  //Billing
  rpc GetUserBalance(GetUserBalanceReq) returns (GetUserBalanceResp);


}