fit:封装数据集http接口

This commit is contained in:
qiwang 2023-03-02 09:01:49 +08:00
parent 1167cb476a
commit 51b0c62a8a
8 changed files with 10198 additions and 4696 deletions

View File

@ -7,3 +7,36 @@ info(
email: "1364512070@qq.com"
)
type DataSet {
DatasetId string `json:"datasetId"`
DataFormat string `json:"dataFormat"`
DatasetFormat int32 `json:"datasetFormat"`
DatasetName string `json:"datasetName"`
DatasetType string `json:"datasetType"`
ImportData bool `json:"importData"`
DataSources []DataSources `json:"DataSources"`
}
type DataSources {
DataPath string `json:"dataPath"`
DataType string `json:"dataType"`
}
type (
listDataSetReq {
ProjectId string `json:"projectId"`
}
listDataSetResp {
Code int32 `json:"code"`
Msg string `json:"msg"`
RecordCount int32 `json:"recordCount"`
DataSet []DataSet `json:"jobs"`
}
)
service AICore-api {
@handler listDataSetHandler
get /listDataSet (listDataSetReq) returns (listDataSetResp)
}

View File

@ -34,30 +34,60 @@ func (s *ModelArtsServer) GetDatasetList(ctx context.Context, in *modelarts.Data
return l.GetDatasetList(in)
}
// creat task
func (s *ModelArtsServer) CreateTask(ctx context.Context, in *modelarts.ImportTaskData) (*modelarts.ImprotTaskDataReq, error) {
// creat task 创建导入任务
func (s *ModelArtsServer) CreateTask(ctx context.Context, in *modelarts.ImportTaskDataReq) (*modelarts.ImportTaskDataResp, error) {
l := logic.NewCreateTaskLogic(ctx, s.svcCtx)
return l.CreateTask(in)
}
// get taskList
func (s *ModelArtsServer) GetImportTaskList(ctx context.Context, in *modelarts.ListImportTasks) (*modelarts.ReturnListImportTasks, error) {
// get taskList 查询数据集导入任务列表
func (s *ModelArtsServer) GetImportTaskList(ctx context.Context, in *modelarts.ListImportTasksReq) (*modelarts.ListImportTasksResp, error) {
l := logic.NewGetImportTaskListLogic(ctx, s.svcCtx)
return l.GetImportTaskList(in)
}
// get ListTrainingJobs
// ListTrainingJobs 查询训练作业列表
func (s *ModelArtsServer) GetListTrainingJobs(ctx context.Context, in *modelarts.ListTrainingJobsreq) (*modelarts.ListTrainingJobsresp, error) {
l := logic.NewGetListTrainingJobsLogic(ctx, s.svcCtx)
return l.GetListTrainingJobs(in)
}
// CreateTrainingJob 创建训练作业
func (s *ModelArtsServer) CreateTrainingJob(ctx context.Context, in *modelarts.CreateTrainingJobreq) (*modelarts.CreateTrainingJobresp, error) {
func (s *ModelArtsServer) CreateTrainingJob(ctx context.Context, in *modelarts.CreateTrainingJobReq) (*modelarts.CreateTrainingJobResp, error) {
l := logic.NewCreateTrainingJobLogic(ctx, s.svcCtx)
return l.CreateTrainingJob(in)
}
// DeleteTrainingJobConfig 删除训练作业
func (s *ModelArtsServer) DeleteTrainingJob(ctx context.Context, in *modelarts.DeleteTrainingJobReq) (*modelarts.DeleteTrainingJobResp, error) {
l := logic.NewDeleteTrainingJobLogic(ctx, s.svcCtx)
return l.DeleteTrainingJob(in)
}
// CreateTrainingJobConfig 创建训练作业参数
func (s *ModelArtsServer) CreateTrainingJobConfig(ctx context.Context, in *modelarts.CreateTrainingJobConfigReq) (*modelarts.CreateTrainingJobConfigResp, error) {
l := logic.NewCreateTrainingJobConfigLogic(ctx, s.svcCtx)
return l.CreateTrainingJobConfig(in)
}
// DeleteTrainingJobConfig 删除训练作业参数
func (s *ModelArtsServer) DeleteTrainingJobConfig(ctx context.Context, in *modelarts.DeleteTrainingJobConfigReq) (*modelarts.DeleteTrainingJobConfigResp, error) {
l := logic.NewDeleteTrainingJobConfigLogic(ctx, s.svcCtx)
return l.DeleteTrainingJobConfig(in)
}
// ListTrainingJobConfig 查询训练作业参数
func (s *ModelArtsServer) ListTrainingJobConfig(ctx context.Context, in *modelarts.ListTrainingJobConfigReq) (*modelarts.ListTrainingJobConfigResp, error) {
l := logic.NewListTrainingJobConfigLogic(ctx, s.svcCtx)
return l.ListTrainingJobConfig(in)
}
// CreateAlgorithm 创建算法
func (s *ModelArtsServer) CreateAlgorithm(ctx context.Context, in *modelarts.CreateAlgorithmReq) (*modelarts.CreateAlgorithmResp, error) {
l := logic.NewCreateAlgorithmLogic(ctx, s.svcCtx)
return l.CreateAlgorithm(in)
}
// export task
func (s *ModelArtsServer) ExportTask(ctx context.Context, in *modelarts.ExportTaskReq) (*modelarts.ExportTaskResp, error) {
l := logic.NewExportTaskLogic(ctx, s.svcCtx)
@ -69,6 +99,11 @@ func (s *ModelArtsServer) GetExportTasksOfDataset(ctx context.Context, in *model
return l.GetExportTasksOfDataset(in)
}
func (s *ModelArtsServer) GetExportTaskStatusOfDataset(ctx context.Context, in *modelarts.GetExportTaskStatusOfDatasetReq) (*modelarts.GetExportTaskStatusOfDatasetResp, error) {
l := logic.NewGetExportTaskStatusOfDatasetLogic(ctx, s.svcCtx)
return l.GetExportTaskStatusOfDataset(in)
}
// processor task
func (s *ModelArtsServer) CreateProcessorTask(ctx context.Context, in *modelarts.CreateProcessorTaskReq) (*modelarts.CreateProcessorTaskResp, error) {
l := logic.NewCreateProcessorTaskLogic(ctx, s.svcCtx)
@ -102,7 +137,7 @@ func (s *ModelArtsServer) ShowModels(ctx context.Context, in *modelarts.ShowMode
}
// service management
func (s *ModelArtsServer) CreateService(ctx context.Context, in *modelarts.CreatServiceReq) (*modelarts.CreatServiceResp, error) {
func (s *ModelArtsServer) CreateService(ctx context.Context, in *modelarts.CreateServiceReq) (*modelarts.CreateServiceResp, error) {
l := logic.NewCreateServiceLogic(ctx, s.svcCtx)
return l.CreateService(in)
}
@ -111,3 +146,18 @@ func (s *ModelArtsServer) ListServices(ctx context.Context, in *modelarts.ListSe
l := logic.NewListServicesLogic(ctx, s.svcCtx)
return l.ListServices(in)
}
func (s *ModelArtsServer) ShowService(ctx context.Context, in *modelarts.ShowServiceReq) (*modelarts.ShowServiceResp, error) {
l := logic.NewShowServiceLogic(ctx, s.svcCtx)
return l.ShowService(in)
}
func (s *ModelArtsServer) DeleteService(ctx context.Context, in *modelarts.DeleteServiceReq) (*modelarts.DeleteServiceResp, error) {
l := logic.NewDeleteServiceLogic(ctx, s.svcCtx)
return l.DeleteService(in)
}
func (s *ModelArtsServer) ListClusters(ctx context.Context, in *modelarts.ListClustersReq) (*modelarts.ListClustersResp, error) {
l := logic.NewListClustersLogic(ctx, s.svcCtx)
return l.ListClusters(in)
}

View File

@ -1,6 +1,8 @@
package svc
import "PCM/adaptor/AIComputing/modelarts/rpc/internal/config"
import (
"PCM/adaptor/AIComputing/modelarts/rpc/internal/config"
)
type ServiceContext struct {
Config config.Config

File diff suppressed because it is too large Load Diff

View File

@ -22,32 +22,46 @@ const _ = grpc.SupportPackageIsVersion7
//
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
type ModelArtsClient interface {
// get modelarts Token
//get modelarts Token
GetToken(ctx context.Context, in *TokenReq, opts ...grpc.CallOption) (*TokenResp, error)
// get modelarts Token
//get modelarts Token
GetDatasetList(ctx context.Context, in *DatasetReq, opts ...grpc.CallOption) (*DatasetResp, error)
// creat task
CreateTask(ctx context.Context, in *ImportTaskData, opts ...grpc.CallOption) (*ImprotTaskDataReq, error)
// get taskList
GetImportTaskList(ctx context.Context, in *ListImportTasks, opts ...grpc.CallOption) (*ReturnListImportTasks, error)
// get ListTrainingJobs
//creat task 创建导入任务
CreateTask(ctx context.Context, in *ImportTaskDataReq, opts ...grpc.CallOption) (*ImportTaskDataResp, error)
//get taskList 查询数据集导入任务列表
GetImportTaskList(ctx context.Context, in *ListImportTasksReq, opts ...grpc.CallOption) (*ListImportTasksResp, error)
// ListTrainingJobs 查询训练作业列表
GetListTrainingJobs(ctx context.Context, in *ListTrainingJobsreq, opts ...grpc.CallOption) (*ListTrainingJobsresp, error)
// CreateTrainingJob 创建训练作业
CreateTrainingJob(ctx context.Context, in *CreateTrainingJobreq, opts ...grpc.CallOption) (*CreateTrainingJobresp, error)
// CreateTrainingJob 创建训练作业
CreateTrainingJob(ctx context.Context, in *CreateTrainingJobReq, opts ...grpc.CallOption) (*CreateTrainingJobResp, error)
// DeleteTrainingJobConfig 删除训练作业
DeleteTrainingJob(ctx context.Context, in *DeleteTrainingJobReq, opts ...grpc.CallOption) (*DeleteTrainingJobResp, error)
// CreateTrainingJobConfig 创建训练作业参数
CreateTrainingJobConfig(ctx context.Context, in *CreateTrainingJobConfigReq, opts ...grpc.CallOption) (*CreateTrainingJobConfigResp, error)
// DeleteTrainingJobConfig 删除训练作业参数
DeleteTrainingJobConfig(ctx context.Context, in *DeleteTrainingJobConfigReq, opts ...grpc.CallOption) (*DeleteTrainingJobConfigResp, error)
// ListTrainingJobConfig 查询训练作业参数
ListTrainingJobConfig(ctx context.Context, in *ListTrainingJobConfigReq, opts ...grpc.CallOption) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error)
//export task
ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error)
GetExportTasksOfDataset(ctx context.Context, in *GetExportTasksOfDatasetReq, opts ...grpc.CallOption) (*GetExportTasksOfDatasetResp, error)
// processor task
GetExportTaskStatusOfDataset(ctx context.Context, in *GetExportTaskStatusOfDatasetReq, opts ...grpc.CallOption) (*GetExportTaskStatusOfDatasetResp, error)
//processor task
CreateProcessorTask(ctx context.Context, in *CreateProcessorTaskReq, opts ...grpc.CallOption) (*CreateProcessorTaskResp, error)
DescribeProcessorTask(ctx context.Context, in *DescribeProcessorTaskReq, opts ...grpc.CallOption) (*DescribeProcessorTaskResp, error)
// model management
//model management
CreateModel(ctx context.Context, in *CreateModelReq, opts ...grpc.CallOption) (*CreateModelResp, error)
DeleteModel(ctx context.Context, in *DeleteModelReq, opts ...grpc.CallOption) (*DeleteModelResp, error)
ListModels(ctx context.Context, in *ListModelReq, opts ...grpc.CallOption) (*ListModelResp, error)
ShowModels(ctx context.Context, in *ShowModelReq, opts ...grpc.CallOption) (*ShowModelResp, error)
// service management
CreateService(ctx context.Context, in *CreatServiceReq, opts ...grpc.CallOption) (*CreatServiceResp, error)
//service management
CreateService(ctx context.Context, in *CreateServiceReq, opts ...grpc.CallOption) (*CreateServiceResp, error)
ListServices(ctx context.Context, in *ListServicesReq, opts ...grpc.CallOption) (*ListServicesResp, error)
ShowService(ctx context.Context, in *ShowServiceReq, opts ...grpc.CallOption) (*ShowServiceResp, error)
DeleteService(ctx context.Context, in *DeleteServiceReq, opts ...grpc.CallOption) (*DeleteServiceResp, error)
ListClusters(ctx context.Context, in *ListClustersReq, opts ...grpc.CallOption) (*ListClustersResp, error)
}
type modelArtsClient struct {
@ -76,8 +90,8 @@ func (c *modelArtsClient) GetDatasetList(ctx context.Context, in *DatasetReq, op
return out, nil
}
func (c *modelArtsClient) CreateTask(ctx context.Context, in *ImportTaskData, opts ...grpc.CallOption) (*ImprotTaskDataReq, error) {
out := new(ImprotTaskDataReq)
func (c *modelArtsClient) CreateTask(ctx context.Context, in *ImportTaskDataReq, opts ...grpc.CallOption) (*ImportTaskDataResp, error) {
out := new(ImportTaskDataResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/createTask", in, out, opts...)
if err != nil {
return nil, err
@ -85,8 +99,8 @@ func (c *modelArtsClient) CreateTask(ctx context.Context, in *ImportTaskData, op
return out, nil
}
func (c *modelArtsClient) GetImportTaskList(ctx context.Context, in *ListImportTasks, opts ...grpc.CallOption) (*ReturnListImportTasks, error) {
out := new(ReturnListImportTasks)
func (c *modelArtsClient) GetImportTaskList(ctx context.Context, in *ListImportTasksReq, opts ...grpc.CallOption) (*ListImportTasksResp, error) {
out := new(ListImportTasksResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/GetImportTaskList", in, out, opts...)
if err != nil {
return nil, err
@ -103,8 +117,8 @@ func (c *modelArtsClient) GetListTrainingJobs(ctx context.Context, in *ListTrain
return out, nil
}
func (c *modelArtsClient) CreateTrainingJob(ctx context.Context, in *CreateTrainingJobreq, opts ...grpc.CallOption) (*CreateTrainingJobresp, error) {
out := new(CreateTrainingJobresp)
func (c *modelArtsClient) CreateTrainingJob(ctx context.Context, in *CreateTrainingJobReq, opts ...grpc.CallOption) (*CreateTrainingJobResp, error) {
out := new(CreateTrainingJobResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/CreateTrainingJob", in, out, opts...)
if err != nil {
return nil, err
@ -112,6 +126,51 @@ func (c *modelArtsClient) CreateTrainingJob(ctx context.Context, in *CreateTrain
return out, nil
}
func (c *modelArtsClient) DeleteTrainingJob(ctx context.Context, in *DeleteTrainingJobReq, opts ...grpc.CallOption) (*DeleteTrainingJobResp, error) {
out := new(DeleteTrainingJobResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/DeleteTrainingJob", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) CreateTrainingJobConfig(ctx context.Context, in *CreateTrainingJobConfigReq, opts ...grpc.CallOption) (*CreateTrainingJobConfigResp, error) {
out := new(CreateTrainingJobConfigResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/CreateTrainingJobConfig", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) DeleteTrainingJobConfig(ctx context.Context, in *DeleteTrainingJobConfigReq, opts ...grpc.CallOption) (*DeleteTrainingJobConfigResp, error) {
out := new(DeleteTrainingJobConfigResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/DeleteTrainingJobConfig", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) ListTrainingJobConfig(ctx context.Context, in *ListTrainingJobConfigReq, opts ...grpc.CallOption) (*ListTrainingJobConfigResp, error) {
out := new(ListTrainingJobConfigResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ListTrainingJobConfig", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error) {
out := new(CreateAlgorithmResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/CreateAlgorithm", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error) {
out := new(ExportTaskResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ExportTask", in, out, opts...)
@ -130,6 +189,15 @@ func (c *modelArtsClient) GetExportTasksOfDataset(ctx context.Context, in *GetEx
return out, nil
}
func (c *modelArtsClient) GetExportTaskStatusOfDataset(ctx context.Context, in *GetExportTaskStatusOfDatasetReq, opts ...grpc.CallOption) (*GetExportTaskStatusOfDatasetResp, error) {
out := new(GetExportTaskStatusOfDatasetResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/GetExportTaskStatusOfDataset", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) CreateProcessorTask(ctx context.Context, in *CreateProcessorTaskReq, opts ...grpc.CallOption) (*CreateProcessorTaskResp, error) {
out := new(CreateProcessorTaskResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/CreateProcessorTask", in, out, opts...)
@ -184,8 +252,8 @@ func (c *modelArtsClient) ShowModels(ctx context.Context, in *ShowModelReq, opts
return out, nil
}
func (c *modelArtsClient) CreateService(ctx context.Context, in *CreatServiceReq, opts ...grpc.CallOption) (*CreatServiceResp, error) {
out := new(CreatServiceResp)
func (c *modelArtsClient) CreateService(ctx context.Context, in *CreateServiceReq, opts ...grpc.CallOption) (*CreateServiceResp, error) {
out := new(CreateServiceResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/CreateService", in, out, opts...)
if err != nil {
return nil, err
@ -202,36 +270,77 @@ func (c *modelArtsClient) ListServices(ctx context.Context, in *ListServicesReq,
return out, nil
}
func (c *modelArtsClient) ShowService(ctx context.Context, in *ShowServiceReq, opts ...grpc.CallOption) (*ShowServiceResp, error) {
out := new(ShowServiceResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ShowService", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) DeleteService(ctx context.Context, in *DeleteServiceReq, opts ...grpc.CallOption) (*DeleteServiceResp, error) {
out := new(DeleteServiceResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/DeleteService", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *modelArtsClient) ListClusters(ctx context.Context, in *ListClustersReq, opts ...grpc.CallOption) (*ListClustersResp, error) {
out := new(ListClustersResp)
err := c.cc.Invoke(ctx, "/modelarts.ModelArts/ListClusters", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// ModelArtsServer is the server API for ModelArts service.
// All implementations must embed UnimplementedModelArtsServer
// for forward compatibility
type ModelArtsServer interface {
// get modelarts Token
//get modelarts Token
GetToken(context.Context, *TokenReq) (*TokenResp, error)
// get modelarts Token
//get modelarts Token
GetDatasetList(context.Context, *DatasetReq) (*DatasetResp, error)
// creat task
CreateTask(context.Context, *ImportTaskData) (*ImprotTaskDataReq, error)
// get taskList
GetImportTaskList(context.Context, *ListImportTasks) (*ReturnListImportTasks, error)
// get ListTrainingJobs
//creat task 创建导入任务
CreateTask(context.Context, *ImportTaskDataReq) (*ImportTaskDataResp, error)
//get taskList 查询数据集导入任务列表
GetImportTaskList(context.Context, *ListImportTasksReq) (*ListImportTasksResp, error)
// ListTrainingJobs 查询训练作业列表
GetListTrainingJobs(context.Context, *ListTrainingJobsreq) (*ListTrainingJobsresp, error)
// CreateTrainingJob 创建训练作业
CreateTrainingJob(context.Context, *CreateTrainingJobreq) (*CreateTrainingJobresp, error)
// CreateTrainingJob 创建训练作业
CreateTrainingJob(context.Context, *CreateTrainingJobReq) (*CreateTrainingJobResp, error)
// DeleteTrainingJobConfig 删除训练作业
DeleteTrainingJob(context.Context, *DeleteTrainingJobReq) (*DeleteTrainingJobResp, error)
// CreateTrainingJobConfig 创建训练作业参数
CreateTrainingJobConfig(context.Context, *CreateTrainingJobConfigReq) (*CreateTrainingJobConfigResp, error)
// DeleteTrainingJobConfig 删除训练作业参数
DeleteTrainingJobConfig(context.Context, *DeleteTrainingJobConfigReq) (*DeleteTrainingJobConfigResp, error)
// ListTrainingJobConfig 查询训练作业参数
ListTrainingJobConfig(context.Context, *ListTrainingJobConfigReq) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(context.Context, *CreateAlgorithmReq) (*CreateAlgorithmResp, error)
//export task
ExportTask(context.Context, *ExportTaskReq) (*ExportTaskResp, error)
GetExportTasksOfDataset(context.Context, *GetExportTasksOfDatasetReq) (*GetExportTasksOfDatasetResp, error)
// processor task
GetExportTaskStatusOfDataset(context.Context, *GetExportTaskStatusOfDatasetReq) (*GetExportTaskStatusOfDatasetResp, error)
//processor task
CreateProcessorTask(context.Context, *CreateProcessorTaskReq) (*CreateProcessorTaskResp, error)
DescribeProcessorTask(context.Context, *DescribeProcessorTaskReq) (*DescribeProcessorTaskResp, error)
// model management
//model management
CreateModel(context.Context, *CreateModelReq) (*CreateModelResp, error)
DeleteModel(context.Context, *DeleteModelReq) (*DeleteModelResp, error)
ListModels(context.Context, *ListModelReq) (*ListModelResp, error)
ShowModels(context.Context, *ShowModelReq) (*ShowModelResp, error)
// service management
CreateService(context.Context, *CreatServiceReq) (*CreatServiceResp, error)
//service management
CreateService(context.Context, *CreateServiceReq) (*CreateServiceResp, error)
ListServices(context.Context, *ListServicesReq) (*ListServicesResp, error)
ShowService(context.Context, *ShowServiceReq) (*ShowServiceResp, error)
DeleteService(context.Context, *DeleteServiceReq) (*DeleteServiceResp, error)
ListClusters(context.Context, *ListClustersReq) (*ListClustersResp, error)
mustEmbedUnimplementedModelArtsServer()
}
@ -245,24 +354,42 @@ func (UnimplementedModelArtsServer) GetToken(context.Context, *TokenReq) (*Token
func (UnimplementedModelArtsServer) GetDatasetList(context.Context, *DatasetReq) (*DatasetResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetDatasetList not implemented")
}
func (UnimplementedModelArtsServer) CreateTask(context.Context, *ImportTaskData) (*ImprotTaskDataReq, error) {
func (UnimplementedModelArtsServer) CreateTask(context.Context, *ImportTaskDataReq) (*ImportTaskDataResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateTask not implemented")
}
func (UnimplementedModelArtsServer) GetImportTaskList(context.Context, *ListImportTasks) (*ReturnListImportTasks, error) {
func (UnimplementedModelArtsServer) GetImportTaskList(context.Context, *ListImportTasksReq) (*ListImportTasksResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetImportTaskList not implemented")
}
func (UnimplementedModelArtsServer) GetListTrainingJobs(context.Context, *ListTrainingJobsreq) (*ListTrainingJobsresp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetListTrainingJobs not implemented")
}
func (UnimplementedModelArtsServer) CreateTrainingJob(context.Context, *CreateTrainingJobreq) (*CreateTrainingJobresp, error) {
func (UnimplementedModelArtsServer) CreateTrainingJob(context.Context, *CreateTrainingJobReq) (*CreateTrainingJobResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateTrainingJob not implemented")
}
func (UnimplementedModelArtsServer) DeleteTrainingJob(context.Context, *DeleteTrainingJobReq) (*DeleteTrainingJobResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteTrainingJob not implemented")
}
func (UnimplementedModelArtsServer) CreateTrainingJobConfig(context.Context, *CreateTrainingJobConfigReq) (*CreateTrainingJobConfigResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateTrainingJobConfig not implemented")
}
func (UnimplementedModelArtsServer) DeleteTrainingJobConfig(context.Context, *DeleteTrainingJobConfigReq) (*DeleteTrainingJobConfigResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteTrainingJobConfig not implemented")
}
func (UnimplementedModelArtsServer) ListTrainingJobConfig(context.Context, *ListTrainingJobConfigReq) (*ListTrainingJobConfigResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ListTrainingJobConfig not implemented")
}
func (UnimplementedModelArtsServer) CreateAlgorithm(context.Context, *CreateAlgorithmReq) (*CreateAlgorithmResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateAlgorithm not implemented")
}
func (UnimplementedModelArtsServer) ExportTask(context.Context, *ExportTaskReq) (*ExportTaskResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ExportTask not implemented")
}
func (UnimplementedModelArtsServer) GetExportTasksOfDataset(context.Context, *GetExportTasksOfDatasetReq) (*GetExportTasksOfDatasetResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetExportTasksOfDataset not implemented")
}
func (UnimplementedModelArtsServer) GetExportTaskStatusOfDataset(context.Context, *GetExportTaskStatusOfDatasetReq) (*GetExportTaskStatusOfDatasetResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetExportTaskStatusOfDataset not implemented")
}
func (UnimplementedModelArtsServer) CreateProcessorTask(context.Context, *CreateProcessorTaskReq) (*CreateProcessorTaskResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateProcessorTask not implemented")
}
@ -281,12 +408,21 @@ func (UnimplementedModelArtsServer) ListModels(context.Context, *ListModelReq) (
func (UnimplementedModelArtsServer) ShowModels(context.Context, *ShowModelReq) (*ShowModelResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ShowModels not implemented")
}
func (UnimplementedModelArtsServer) CreateService(context.Context, *CreatServiceReq) (*CreatServiceResp, error) {
func (UnimplementedModelArtsServer) CreateService(context.Context, *CreateServiceReq) (*CreateServiceResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method CreateService not implemented")
}
func (UnimplementedModelArtsServer) ListServices(context.Context, *ListServicesReq) (*ListServicesResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ListServices not implemented")
}
func (UnimplementedModelArtsServer) ShowService(context.Context, *ShowServiceReq) (*ShowServiceResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ShowService not implemented")
}
func (UnimplementedModelArtsServer) DeleteService(context.Context, *DeleteServiceReq) (*DeleteServiceResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method DeleteService not implemented")
}
func (UnimplementedModelArtsServer) ListClusters(context.Context, *ListClustersReq) (*ListClustersResp, error) {
return nil, status.Errorf(codes.Unimplemented, "method ListClusters not implemented")
}
func (UnimplementedModelArtsServer) mustEmbedUnimplementedModelArtsServer() {}
// UnsafeModelArtsServer may be embedded to opt out of forward compatibility for this service.
@ -337,7 +473,7 @@ func _ModelArts_GetDatasetList_Handler(srv interface{}, ctx context.Context, dec
}
func _ModelArts_CreateTask_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ImportTaskData)
in := new(ImportTaskDataReq)
if err := dec(in); err != nil {
return nil, err
}
@ -349,13 +485,13 @@ func _ModelArts_CreateTask_Handler(srv interface{}, ctx context.Context, dec fun
FullMethod: "/modelarts.ModelArts/createTask",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).CreateTask(ctx, req.(*ImportTaskData))
return srv.(ModelArtsServer).CreateTask(ctx, req.(*ImportTaskDataReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_GetImportTaskList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ListImportTasks)
in := new(ListImportTasksReq)
if err := dec(in); err != nil {
return nil, err
}
@ -367,7 +503,7 @@ func _ModelArts_GetImportTaskList_Handler(srv interface{}, ctx context.Context,
FullMethod: "/modelarts.ModelArts/GetImportTaskList",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).GetImportTaskList(ctx, req.(*ListImportTasks))
return srv.(ModelArtsServer).GetImportTaskList(ctx, req.(*ListImportTasksReq))
}
return interceptor(ctx, in, info, handler)
}
@ -391,7 +527,7 @@ func _ModelArts_GetListTrainingJobs_Handler(srv interface{}, ctx context.Context
}
func _ModelArts_CreateTrainingJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateTrainingJobreq)
in := new(CreateTrainingJobReq)
if err := dec(in); err != nil {
return nil, err
}
@ -403,7 +539,97 @@ func _ModelArts_CreateTrainingJob_Handler(srv interface{}, ctx context.Context,
FullMethod: "/modelarts.ModelArts/CreateTrainingJob",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).CreateTrainingJob(ctx, req.(*CreateTrainingJobreq))
return srv.(ModelArtsServer).CreateTrainingJob(ctx, req.(*CreateTrainingJobReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_DeleteTrainingJob_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteTrainingJobReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).DeleteTrainingJob(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/DeleteTrainingJob",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).DeleteTrainingJob(ctx, req.(*DeleteTrainingJobReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_CreateTrainingJobConfig_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateTrainingJobConfigReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).CreateTrainingJobConfig(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/CreateTrainingJobConfig",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).CreateTrainingJobConfig(ctx, req.(*CreateTrainingJobConfigReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_DeleteTrainingJobConfig_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteTrainingJobConfigReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).DeleteTrainingJobConfig(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/DeleteTrainingJobConfig",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).DeleteTrainingJobConfig(ctx, req.(*DeleteTrainingJobConfigReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_ListTrainingJobConfig_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ListTrainingJobConfigReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).ListTrainingJobConfig(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/ListTrainingJobConfig",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).ListTrainingJobConfig(ctx, req.(*ListTrainingJobConfigReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_CreateAlgorithm_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateAlgorithmReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).CreateAlgorithm(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/CreateAlgorithm",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).CreateAlgorithm(ctx, req.(*CreateAlgorithmReq))
}
return interceptor(ctx, in, info, handler)
}
@ -444,6 +670,24 @@ func _ModelArts_GetExportTasksOfDataset_Handler(srv interface{}, ctx context.Con
return interceptor(ctx, in, info, handler)
}
func _ModelArts_GetExportTaskStatusOfDataset_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(GetExportTaskStatusOfDatasetReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).GetExportTaskStatusOfDataset(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/GetExportTaskStatusOfDataset",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).GetExportTaskStatusOfDataset(ctx, req.(*GetExportTaskStatusOfDatasetReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_CreateProcessorTask_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreateProcessorTaskReq)
if err := dec(in); err != nil {
@ -553,7 +797,7 @@ func _ModelArts_ShowModels_Handler(srv interface{}, ctx context.Context, dec fun
}
func _ModelArts_CreateService_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(CreatServiceReq)
in := new(CreateServiceReq)
if err := dec(in); err != nil {
return nil, err
}
@ -565,7 +809,7 @@ func _ModelArts_CreateService_Handler(srv interface{}, ctx context.Context, dec
FullMethod: "/modelarts.ModelArts/CreateService",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).CreateService(ctx, req.(*CreatServiceReq))
return srv.(ModelArtsServer).CreateService(ctx, req.(*CreateServiceReq))
}
return interceptor(ctx, in, info, handler)
}
@ -588,6 +832,60 @@ func _ModelArts_ListServices_Handler(srv interface{}, ctx context.Context, dec f
return interceptor(ctx, in, info, handler)
}
func _ModelArts_ShowService_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ShowServiceReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).ShowService(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/ShowService",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).ShowService(ctx, req.(*ShowServiceReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_DeleteService_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(DeleteServiceReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).DeleteService(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/DeleteService",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).DeleteService(ctx, req.(*DeleteServiceReq))
}
return interceptor(ctx, in, info, handler)
}
func _ModelArts_ListClusters_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ListClustersReq)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ModelArtsServer).ListClusters(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/modelarts.ModelArts/ListClusters",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ModelArtsServer).ListClusters(ctx, req.(*ListClustersReq))
}
return interceptor(ctx, in, info, handler)
}
// ModelArts_ServiceDesc is the grpc.ServiceDesc for ModelArts service.
// It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy)
@ -619,6 +917,26 @@ var ModelArts_ServiceDesc = grpc.ServiceDesc{
MethodName: "CreateTrainingJob",
Handler: _ModelArts_CreateTrainingJob_Handler,
},
{
MethodName: "DeleteTrainingJob",
Handler: _ModelArts_DeleteTrainingJob_Handler,
},
{
MethodName: "CreateTrainingJobConfig",
Handler: _ModelArts_CreateTrainingJobConfig_Handler,
},
{
MethodName: "DeleteTrainingJobConfig",
Handler: _ModelArts_DeleteTrainingJobConfig_Handler,
},
{
MethodName: "ListTrainingJobConfig",
Handler: _ModelArts_ListTrainingJobConfig_Handler,
},
{
MethodName: "CreateAlgorithm",
Handler: _ModelArts_CreateAlgorithm_Handler,
},
{
MethodName: "ExportTask",
Handler: _ModelArts_ExportTask_Handler,
@ -627,6 +945,10 @@ var ModelArts_ServiceDesc = grpc.ServiceDesc{
MethodName: "GetExportTasksOfDataset",
Handler: _ModelArts_GetExportTasksOfDataset_Handler,
},
{
MethodName: "GetExportTaskStatusOfDataset",
Handler: _ModelArts_GetExportTaskStatusOfDataset_Handler,
},
{
MethodName: "CreateProcessorTask",
Handler: _ModelArts_CreateProcessorTask_Handler,
@ -659,6 +981,18 @@ var ModelArts_ServiceDesc = grpc.ServiceDesc{
MethodName: "ListServices",
Handler: _ModelArts_ListServices_Handler,
},
{
MethodName: "ShowService",
Handler: _ModelArts_ShowService_Handler,
},
{
MethodName: "DeleteService",
Handler: _ModelArts_DeleteService_Handler,
},
{
MethodName: "ListClusters",
Handler: _ModelArts_ListClusters_Handler,
},
},
Streams: []grpc.StreamDesc{},
Metadata: "modelarts.proto",

View File

@ -13,148 +13,178 @@ import (
)
type (
AlgoConfigs = modelarts.AlgoConfigs
Algorithm = modelarts.Algorithm
Algorithms = modelarts.Algorithms
AnnotationFormatConfig = modelarts.AnnotationFormatConfig
AnnotationsS = modelarts.AnnotationsS
Auth = modelarts.Auth
AutoSearch = modelarts.AutoSearch
AutoSearchAlgoConfigParameter = modelarts.AutoSearchAlgoConfigParameter
Billing = modelarts.Billing
ConfigResponse = modelarts.ConfigResponse
Constraint = modelarts.Constraint
ConstraintS = modelarts.ConstraintS
Cpu = modelarts.Cpu
CreateModelReq = modelarts.CreateModelReq
CreateModelRequestInferParams = modelarts.CreateModelRequestInferParams
CreateModelRequestModelApis = modelarts.CreateModelRequestModelApis
CreateModelRequestTemplateInput = modelarts.CreateModelRequestTemplateInput
CreateModelResp = modelarts.CreateModelResp
CreateProcessorTaskReq = modelarts.CreateProcessorTaskReq
CreateProcessorTaskResp = modelarts.CreateProcessorTaskResp
CreateServiceReq = modelarts.CreateServiceReq
CreateServiceResp = modelarts.CreateServiceResp
CreateTrainingJobConfigReq = modelarts.CreateTrainingJobConfigReq
CreateTrainingJobConfigResp = modelarts.CreateTrainingJobConfigResp
CreateTrainingJobReq = modelarts.CreateTrainingJobReq
CreateTrainingJobResp = modelarts.CreateTrainingJobResp
CustomSpec = modelarts.CustomSpec
DataSource = modelarts.DataSource
DataSources = modelarts.DataSources
Dataset = modelarts.Dataset
DatasetReq = modelarts.DatasetReq
DatasetResp = modelarts.DatasetResp
Datasets = modelarts.Datasets
DeleteModelReq = modelarts.DeleteModelReq
DeleteModelResp = modelarts.DeleteModelResp
DeleteModelResponseFailedList = modelarts.DeleteModelResponseFailedList
DeleteServiceReq = modelarts.DeleteServiceReq
DeleteServiceResp = modelarts.DeleteServiceResp
DeleteTrainingJobConfigReq = modelarts.DeleteTrainingJobConfigReq
DeleteTrainingJobConfigResp = modelarts.DeleteTrainingJobConfigResp
DeleteTrainingJobReq = modelarts.DeleteTrainingJobReq
DeleteTrainingJobResp = modelarts.DeleteTrainingJobResp
DescribeProcessorTaskReq = modelarts.DescribeProcessorTaskReq
DescribeProcessorTaskResp = modelarts.DescribeProcessorTaskResp
Disk = modelarts.Disk
Domain = modelarts.Domain
Engine = modelarts.Engine
Engines = modelarts.Engines
ExportParams = modelarts.ExportParams
ExportTaskReq = modelarts.ExportTaskReq
ExportTaskResp = modelarts.ExportTaskResp
ExportTaskStatus = modelarts.ExportTaskStatus
FileStatistics = modelarts.FileStatistics
FlavorDetail = modelarts.FlavorDetail
FlavorInfo = modelarts.FlavorInfo
FlavorResponse = modelarts.FlavorResponse
GetExportTasksOfDatasetReq = modelarts.GetExportTasksOfDatasetReq
GetExportTasksOfDatasetResp = modelarts.GetExportTasksOfDatasetResp
Gpu = modelarts.Gpu
GuideDoc = modelarts.GuideDoc
I18NDescription = modelarts.I18NDescription
Identity = modelarts.Identity
ImportTaskData = modelarts.ImportTaskData
ImportTasks = modelarts.ImportTasks
ImprotTaskDataReq = modelarts.ImprotTaskDataReq
Input = modelarts.Input
InputDataInfo = modelarts.InputDataInfo
Inputs = modelarts.Inputs
JobAlgorithmResponse = modelarts.JobAlgorithmResponse
JobMetadata = modelarts.JobMetadata
JobResponse = modelarts.JobResponse
ListImportTasks = modelarts.ListImportTasks
ListModelReq = modelarts.ListModelReq
ListModelResp = modelarts.ListModelResp
ListServices = modelarts.ListServices
ListServicesReq = modelarts.ListServicesReq
ListServicesResp = modelarts.ListServicesResp
ListTrainingJobConfigReq = modelarts.ListTrainingJobConfigReq
ListTrainingJobConfigResp = modelarts.ListTrainingJobConfigResp
ListTrainingJobsreq = modelarts.ListTrainingJobsreq
ListTrainingJobsresp = modelarts.ListTrainingJobsresp
LogExportPath = modelarts.LogExportPath
Memory = modelarts.Memory
Metadata = modelarts.Metadata
MetadataS = modelarts.MetadataS
Methods = modelarts.Methods
ModelDependencies = modelarts.ModelDependencies
ModelHealth = modelarts.ModelHealth
ModelInOutputParams = modelarts.ModelInOutputParams
ModelListItem = modelarts.ModelListItem
ModelParamsInfo = modelarts.ModelParamsInfo
ModelSpecification = modelarts.ModelSpecification
Nfs = modelarts.Nfs
Npu = modelarts.Npu
Obs = modelarts.Obs
Obs1 = modelarts.Obs1
OperatorParam = modelarts.OperatorParam
Output = modelarts.Output
Outputs = modelarts.Outputs
Packages = modelarts.Packages
Parameter = modelarts.Parameter
ParameterS = modelarts.ParameterS
Parameters = modelarts.Parameters
Password = modelarts.Password
Policies = modelarts.Policies
ProcessorDataSource = modelarts.ProcessorDataSource
Project = modelarts.Project
QueryServiceConfig = modelarts.QueryServiceConfig
Remote = modelarts.Remote
RemoteConstraint = modelarts.RemoteConstraint
Resource = modelarts.Resource
ResourceS = modelarts.ResourceS
ReturnListImportTasks = modelarts.ReturnListImportTasks
RewardAttrs = modelarts.RewardAttrs
Scheduler = modelarts.Scheduler
SchemaMaps = modelarts.SchemaMaps
Scope = modelarts.Scope
SearchCondition = modelarts.SearchCondition
SearchLabel = modelarts.SearchLabel
SearchLabels = modelarts.SearchLabels
SearchParams = modelarts.SearchParams
SearchProp = modelarts.SearchProp
ServiceConfig = modelarts.ServiceConfig
ShowModelReq = modelarts.ShowModelReq
ShowModelResp = modelarts.ShowModelResp
ShowServiceReq = modelarts.ShowServiceReq
ShowServiceResp = modelarts.ShowServiceResp
SourceInfo = modelarts.SourceInfo
Spec = modelarts.Spec
Spec1 = modelarts.Spec1
Status = modelarts.Status
TaskResponse = modelarts.TaskResponse
TaskStatuses = modelarts.TaskStatuses
Template = modelarts.Template
TemplateParam = modelarts.TemplateParam
TokenReq = modelarts.TokenReq
TokenResp = modelarts.TokenResp
TrainingExperimentReference = modelarts.TrainingExperimentReference
User = modelarts.User
Volumes = modelarts.Volumes
Weigou = modelarts.Weigou
WorkPath = modelarts.WorkPath
AdvancedConfigAl = modelarts.AdvancedConfigAl
AlgoConfigs = modelarts.AlgoConfigs
Algorithm = modelarts.Algorithm
Algorithms = modelarts.Algorithms
AnnotationFormatConfig = modelarts.AnnotationFormatConfig
AnnotationsS = modelarts.AnnotationsS
AttributesAlRq = modelarts.AttributesAlRq
Auth = modelarts.Auth
AutoSearch = modelarts.AutoSearch
AutoSearchAlgoConfigParameter = modelarts.AutoSearchAlgoConfigParameter
Billing = modelarts.Billing
Children = modelarts.Children
Cluster = modelarts.Cluster
ClusterNode = modelarts.ClusterNode
CodeTree = modelarts.CodeTree
ConfigResponse = modelarts.ConfigResponse
Constraint = modelarts.Constraint
ConstraintAlRp = modelarts.ConstraintAlRp
ConstraintAlRq = modelarts.ConstraintAlRq
ConstraintS = modelarts.ConstraintS
Cpu = modelarts.Cpu
CreateAlgorithmReq = modelarts.CreateAlgorithmReq
CreateAlgorithmResp = modelarts.CreateAlgorithmResp
CreateModelReq = modelarts.CreateModelReq
CreateModelRequestInferParams = modelarts.CreateModelRequestInferParams
CreateModelRequestModelApis = modelarts.CreateModelRequestModelApis
CreateModelRequestTemplateInput = modelarts.CreateModelRequestTemplateInput
CreateModelResp = modelarts.CreateModelResp
CreateProcessorTaskReq = modelarts.CreateProcessorTaskReq
CreateProcessorTaskResp = modelarts.CreateProcessorTaskResp
CreateServiceReq = modelarts.CreateServiceReq
CreateServiceResp = modelarts.CreateServiceResp
CreateTrainingJobConfigReq = modelarts.CreateTrainingJobConfigReq
CreateTrainingJobConfigResp = modelarts.CreateTrainingJobConfigResp
CreateTrainingJobReq = modelarts.CreateTrainingJobReq
CreateTrainingJobResp = modelarts.CreateTrainingJobResp
CustomSpec = modelarts.CustomSpec
DataSource = modelarts.DataSource
DataSources = modelarts.DataSources
Dataset = modelarts.Dataset
DatasetReq = modelarts.DatasetReq
DatasetResp = modelarts.DatasetResp
Datasets = modelarts.Datasets
DeleteModelReq = modelarts.DeleteModelReq
DeleteModelResp = modelarts.DeleteModelResp
DeleteModelResponseFailedList = modelarts.DeleteModelResponseFailedList
DeleteServiceReq = modelarts.DeleteServiceReq
DeleteServiceResp = modelarts.DeleteServiceResp
DeleteTrainingJobConfigReq = modelarts.DeleteTrainingJobConfigReq
DeleteTrainingJobConfigResp = modelarts.DeleteTrainingJobConfigResp
DeleteTrainingJobReq = modelarts.DeleteTrainingJobReq
DeleteTrainingJobResp = modelarts.DeleteTrainingJobResp
DescribeProcessorTaskReq = modelarts.DescribeProcessorTaskReq
DescribeProcessorTaskResp = modelarts.DescribeProcessorTaskResp
Disk = modelarts.Disk
Domain = modelarts.Domain
Engine = modelarts.Engine
EngineAlRp = modelarts.EngineAlRp
EngineAlRq = modelarts.EngineAlRq
Engines = modelarts.Engines
ExportParams = modelarts.ExportParams
ExportTaskReq = modelarts.ExportTaskReq
ExportTaskResp = modelarts.ExportTaskResp
ExportTaskStatus = modelarts.ExportTaskStatus
FileStatistics = modelarts.FileStatistics
FlavorDetail = modelarts.FlavorDetail
FlavorInfo = modelarts.FlavorInfo
FlavorResponse = modelarts.FlavorResponse
GetExportTaskStatusOfDatasetReq = modelarts.GetExportTaskStatusOfDatasetReq
GetExportTaskStatusOfDatasetResp = modelarts.GetExportTaskStatusOfDatasetResp
GetExportTasksOfDatasetReq = modelarts.GetExportTasksOfDatasetReq
GetExportTasksOfDatasetResp = modelarts.GetExportTasksOfDatasetResp
Gpu = modelarts.Gpu
GuideDoc = modelarts.GuideDoc
I18NDescription = modelarts.I18NDescription
Identity = modelarts.Identity
ImageInfo = modelarts.ImageInfo
ImportTaskDataReq = modelarts.ImportTaskDataReq
ImportTaskDataResp = modelarts.ImportTaskDataResp
ImportTasks = modelarts.ImportTasks
Input = modelarts.Input
InputDataInfo = modelarts.InputDataInfo
Inputs = modelarts.Inputs
InputsAlRp = modelarts.InputsAlRp
InputsAlRq = modelarts.InputsAlRq
JobAlgorithmResponse = modelarts.JobAlgorithmResponse
JobConfigAl = modelarts.JobConfigAl
JobMetadata = modelarts.JobMetadata
JobResponse = modelarts.JobResponse
ListClustersReq = modelarts.ListClustersReq
ListClustersResp = modelarts.ListClustersResp
ListImportTasksReq = modelarts.ListImportTasksReq
ListImportTasksResp = modelarts.ListImportTasksResp
ListModelReq = modelarts.ListModelReq
ListModelResp = modelarts.ListModelResp
ListServices = modelarts.ListServices
ListServicesReq = modelarts.ListServicesReq
ListServicesResp = modelarts.ListServicesResp
ListTrainingJobConfigReq = modelarts.ListTrainingJobConfigReq
ListTrainingJobConfigResp = modelarts.ListTrainingJobConfigResp
ListTrainingJobsreq = modelarts.ListTrainingJobsreq
ListTrainingJobsresp = modelarts.ListTrainingJobsresp
LogExportPath = modelarts.LogExportPath
Memory = modelarts.Memory
Metadata = modelarts.Metadata
MetadataAlRp = modelarts.MetadataAlRp
MetadataAlRq = modelarts.MetadataAlRq
MetadataS = modelarts.MetadataS
Methods = modelarts.Methods
ModelDependencies = modelarts.ModelDependencies
ModelHealth = modelarts.ModelHealth
ModelInOutputParams = modelarts.ModelInOutputParams
ModelListItem = modelarts.ModelListItem
ModelParamsInfo = modelarts.ModelParamsInfo
ModelSpecification = modelarts.ModelSpecification
Nfs = modelarts.Nfs
Npu = modelarts.Npu
Obs = modelarts.Obs
Obs1 = modelarts.Obs1
OperatorParam = modelarts.OperatorParam
Output = modelarts.Output
Outputs = modelarts.Outputs
OutputsAl = modelarts.OutputsAl
OutputsAlRp = modelarts.OutputsAlRp
Packages = modelarts.Packages
Parameter = modelarts.Parameter
ParameterS = modelarts.ParameterS
Parameters = modelarts.Parameters
ParametersAlRp = modelarts.ParametersAlRp
ParametersAlRq = modelarts.ParametersAlRq
Password = modelarts.Password
Policies = modelarts.Policies
ProcessorDataSource = modelarts.ProcessorDataSource
Project = modelarts.Project
QueryServiceConfig = modelarts.QueryServiceConfig
Remote = modelarts.Remote
RemoteConstraint = modelarts.RemoteConstraint
RemoteConstraints = modelarts.RemoteConstraints
Resource = modelarts.Resource
ResourceRequirements = modelarts.ResourceRequirements
ResourceS = modelarts.ResourceS
RewardAttrs = modelarts.RewardAttrs
Scheduler = modelarts.Scheduler
SchemaMaps = modelarts.SchemaMaps
Scope = modelarts.Scope
SearchCondition = modelarts.SearchCondition
SearchLabel = modelarts.SearchLabel
SearchLabels = modelarts.SearchLabels
SearchParams = modelarts.SearchParams
SearchProp = modelarts.SearchProp
ServiceConfig = modelarts.ServiceConfig
ShareInfoAlRp = modelarts.ShareInfoAlRp
ShowModelReq = modelarts.ShowModelReq
ShowModelResp = modelarts.ShowModelResp
ShowServiceReq = modelarts.ShowServiceReq
ShowServiceResp = modelarts.ShowServiceResp
SourceInfo = modelarts.SourceInfo
Spec = modelarts.Spec
Spec1 = modelarts.Spec1
Status = modelarts.Status
TagsAlRp = modelarts.TagsAlRp
TaskResponse = modelarts.TaskResponse
TaskStatuses = modelarts.TaskStatuses
Template = modelarts.Template
TemplateParam = modelarts.TemplateParam
TokenReq = modelarts.TokenReq
TokenResp = modelarts.TokenResp
TrainingExperimentReference = modelarts.TrainingExperimentReference
User = modelarts.User
Volumes = modelarts.Volumes
Weigou = modelarts.Weigou
WorkPath = modelarts.WorkPath
ModelArts interface {
// get modelarts Token
@ -162,9 +192,9 @@ type (
// get modelarts Token
GetDatasetList(ctx context.Context, in *DatasetReq, opts ...grpc.CallOption) (*DatasetResp, error)
// creat task 创建导入任务
CreateTask(ctx context.Context, in *ImportTaskData, opts ...grpc.CallOption) (*ImprotTaskDataReq, error)
CreateTask(ctx context.Context, in *ImportTaskDataReq, opts ...grpc.CallOption) (*ImportTaskDataResp, error)
// get taskList 查询数据集导入任务列表
GetImportTaskList(ctx context.Context, in *ListImportTasks, opts ...grpc.CallOption) (*ReturnListImportTasks, error)
GetImportTaskList(ctx context.Context, in *ListImportTasksReq, opts ...grpc.CallOption) (*ListImportTasksResp, error)
// ListTrainingJobs 查询训练作业列表
GetListTrainingJobs(ctx context.Context, in *ListTrainingJobsreq, opts ...grpc.CallOption) (*ListTrainingJobsresp, error)
// CreateTrainingJob 创建训练作业
@ -177,9 +207,12 @@ type (
DeleteTrainingJobConfig(ctx context.Context, in *DeleteTrainingJobConfigReq, opts ...grpc.CallOption) (*DeleteTrainingJobConfigResp, error)
// ListTrainingJobConfig 查询训练作业参数
ListTrainingJobConfig(ctx context.Context, in *ListTrainingJobConfigReq, opts ...grpc.CallOption) (*ListTrainingJobConfigResp, error)
// CreateAlgorithm 创建算法
CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error)
// export task
ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error)
GetExportTasksOfDataset(ctx context.Context, in *GetExportTasksOfDatasetReq, opts ...grpc.CallOption) (*GetExportTasksOfDatasetResp, error)
GetExportTaskStatusOfDataset(ctx context.Context, in *GetExportTaskStatusOfDatasetReq, opts ...grpc.CallOption) (*GetExportTaskStatusOfDatasetResp, error)
// processor task
CreateProcessorTask(ctx context.Context, in *CreateProcessorTaskReq, opts ...grpc.CallOption) (*CreateProcessorTaskResp, error)
DescribeProcessorTask(ctx context.Context, in *DescribeProcessorTaskReq, opts ...grpc.CallOption) (*DescribeProcessorTaskResp, error)
@ -193,6 +226,7 @@ type (
ListServices(ctx context.Context, in *ListServicesReq, opts ...grpc.CallOption) (*ListServicesResp, error)
ShowService(ctx context.Context, in *ShowServiceReq, opts ...grpc.CallOption) (*ShowServiceResp, error)
DeleteService(ctx context.Context, in *DeleteServiceReq, opts ...grpc.CallOption) (*DeleteServiceResp, error)
ListClusters(ctx context.Context, in *ListClustersReq, opts ...grpc.CallOption) (*ListClustersResp, error)
}
defaultModelArts struct {
@ -219,13 +253,13 @@ func (m *defaultModelArts) GetDatasetList(ctx context.Context, in *DatasetReq, o
}
// creat task 创建导入任务
func (m *defaultModelArts) CreateTask(ctx context.Context, in *ImportTaskData, opts ...grpc.CallOption) (*ImprotTaskDataReq, error) {
func (m *defaultModelArts) CreateTask(ctx context.Context, in *ImportTaskDataReq, opts ...grpc.CallOption) (*ImportTaskDataResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.CreateTask(ctx, in, opts...)
}
// get taskList 查询数据集导入任务列表
func (m *defaultModelArts) GetImportTaskList(ctx context.Context, in *ListImportTasks, opts ...grpc.CallOption) (*ReturnListImportTasks, error) {
func (m *defaultModelArts) GetImportTaskList(ctx context.Context, in *ListImportTasksReq, opts ...grpc.CallOption) (*ListImportTasksResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.GetImportTaskList(ctx, in, opts...)
}
@ -266,6 +300,12 @@ func (m *defaultModelArts) ListTrainingJobConfig(ctx context.Context, in *ListTr
return client.ListTrainingJobConfig(ctx, in, opts...)
}
// CreateAlgorithm 创建算法
func (m *defaultModelArts) CreateAlgorithm(ctx context.Context, in *CreateAlgorithmReq, opts ...grpc.CallOption) (*CreateAlgorithmResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.CreateAlgorithm(ctx, in, opts...)
}
// export task
func (m *defaultModelArts) ExportTask(ctx context.Context, in *ExportTaskReq, opts ...grpc.CallOption) (*ExportTaskResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
@ -277,6 +317,11 @@ func (m *defaultModelArts) GetExportTasksOfDataset(ctx context.Context, in *GetE
return client.GetExportTasksOfDataset(ctx, in, opts...)
}
func (m *defaultModelArts) GetExportTaskStatusOfDataset(ctx context.Context, in *GetExportTaskStatusOfDatasetReq, opts ...grpc.CallOption) (*GetExportTaskStatusOfDatasetResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.GetExportTaskStatusOfDataset(ctx, in, opts...)
}
// processor task
func (m *defaultModelArts) CreateProcessorTask(ctx context.Context, in *CreateProcessorTaskReq, opts ...grpc.CallOption) (*CreateProcessorTaskResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
@ -329,3 +374,8 @@ func (m *defaultModelArts) DeleteService(ctx context.Context, in *DeleteServiceR
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.DeleteService(ctx, in, opts...)
}
func (m *defaultModelArts) ListClusters(ctx context.Context, in *ListClustersReq, opts ...grpc.CallOption) (*ListClustersResp, error) {
client := modelarts.NewModelArtsClient(m.cli.Conn())
return client.ListClusters(ctx, in, opts...)
}

View File

@ -54,8 +54,8 @@ message datasetReq{
}
message datasetResp{
string total_number = 1;
datasets datasets =2;
string total_number = 1; // @gotags: copier:"TotalNumber"
datasets datasets =2; // @gotags: copier:"AcctTime"
}
message datasets{
@ -961,34 +961,7 @@ message GetExportTaskStatusOfDatasetResp{
/******************List Services End*************************/
/******************find datasetList start*************************/
message datasetReq{
string project_id = 1;
}
message datasetResp{
string total_number = 1;
datasets datasets =2;
}
message datasets{
string dataset_id =1;
string data_format =2;
DataSources data_sources =3;
int32 dataset_format =4;
string dataset_name =5;
int32 dataset_type =6;
bool import_data =7;
}
message DataSources{
string data_path =1;
int32 data_type =2;
}
/******************find datasetList end*************************/
/******************Create Processor Task Start*************************/
message CreateProcessorTaskReq{

2
go.mod
View File

@ -4,6 +4,7 @@ go 1.19
require (
github.com/bitly/go-simplejson v0.5.0
github.com/grpc-ecosystem/grpc-gateway/v2 v2.7.0
github.com/jinzhu/copier v0.3.5
github.com/pkg/errors v0.9.1
github.com/shopspring/decimal v1.3.1
@ -38,7 +39,6 @@ require (
github.com/google/pprof v0.0.0-20211214055906-6f57359322fd // indirect
github.com/google/uuid v1.3.0 // indirect
github.com/googleapis/gnostic v0.5.5 // indirect
github.com/grpc-ecosystem/grpc-gateway/v2 v2.7.0 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/mattn/go-colorable v0.1.9 // indirect
github.com/mattn/go-isatty v0.0.14 // indirect