annzee
/
PCM
forked from JCCE/PCM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新导入任务参数名

This commit is contained in:
lee 2023-03-01 09:54:28 +08:00
parent d903110666
commit 9e0e7b97b0
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
adaptor/AIComputing/modelarts/rpc/internal/logic/createtasklogic.go
View File

@ -28,8 +28,8 @@ func NewCreateTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Create
}
// creat task 创建导入任务
func (l *CreateTaskLogic) CreateTask(in *modelarts.ImportTaskData) (*modelarts.ImprotTaskDataReq, error) {
var resp modelarts.ImprotTaskDataReq
func (l *CreateTaskLogic) CreateTask(in *modelarts.ImportTaskDataReq) (*modelarts.ImportTaskDataResp, error) {
var resp modelarts.ImportTaskDataResp
url := "https://modelarts.cn-north-4.myhuaweicloud.com/v2/" + in.ProjectId + "/datasets/" + in.DatasetId + "/import-tasks"
reqByte, err := json.Marshal(in)

4
adaptor/AIComputing/modelarts/rpc/internal/logic/getimporttasklistlogic.go
View File

@ -30,12 +30,12 @@ func NewGetImportTaskListLogic(ctx context.Context, svcCtx *svc.ServiceContext)
}
// find taskList 查询数据集导入任务列表
func (l *GetImportTaskListLogic) GetImportTaskList(in *modelarts.ListImportTasks) (*modelarts.ReturnListImportTasks, error) {
func (l *GetImportTaskListLogic) GetImportTaskList(in *modelarts.ListImportTasksReq) (*modelarts.ListImportTasksResp, error) {
datasetId := in.DatasetId
projectId := in.ProjectId
limit := strconv.Itoa(int(in.Limit))
offset := strconv.Itoa(int(in.Offset))
var resp modelarts.ReturnListImportTasks
var resp modelarts.ListImportTasksResp
token := common.GetToken()
//c := http.Client{Timeout: time.Duration(3) * time.Second}
reqUrl, err := tool.HttpClient(tool.GET, "https://modelarts.cn-north-4.myhuaweicloud.com/v2/"+projectId+"/datasets/"+datasetId+"/import-tasks?"+limit+"&"+offset, strings.NewReader(``), token)

12
adaptor/AIComputing/modelarts/rpc/pb/modelarts.proto
View File

@ -51,14 +51,14 @@ message TokenResp{
/******************ImportTask start*************************/
//数据集导入创建
message ImportTaskData{
message ImportTaskDataReq{
string dataset_id = 1;
string project_id = 2;
string import_path = 3;
}
message ImprotTaskDataReq{
message ImportTaskDataResp{
string task_id =1;
}
/******************ImportTask end*************************/
@ -66,7 +66,7 @@ message ImprotTaskDataReq{
/******************ListImportTasks Start*************************/
//查询数据集导入列表
message ListImportTasks{
message ListImportTasksReq{
string dataset_id = 1; // @gotags: copier:"Dataset_id"
string project_id = 2; // @gotags: copier:"Project_id"
int32 limit = 3; // @gotags: copier:"Limit"
@ -74,7 +74,7 @@ message ListImportTasks{
}
message ReturnListImportTasks{
message ListImportTasksResp{
uint32 total_count = 1; //@gotags: copier:"Total_count"
repeated Import_tasks import_tasks = 2; //@gotags: copier:"Import_tasks"
}
@ -1183,9 +1183,9 @@ service ModelArts {
rpc GetDatasetList(datasetReq) returns (datasetResp);
//creat task 创建导入任务
rpc createTask(ImportTaskData) returns(ImprotTaskDataReq);
rpc createTask(ImportTaskDataReq) returns(ImportTaskDataResp);
//get taskList 查询数据集导入任务列表
rpc GetImportTaskList(ListImportTasks) returns (ReturnListImportTasks);
rpc GetImportTaskList(ListImportTasksReq) returns (ListImportTasksResp);
// ListTrainingJobs 查询训练作业列表
rpc GetListTrainingJobs(ListTrainingJobsreq) returns (ListTrainingJobsresp);
// CreateTrainingJob 创建训练作业