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

proto更新

This commit is contained in:
lee 2023-02-27 17:08:41 +08:00
parent 37e3630ae4
commit ec5f2d3eea
3 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

File diff suppressed because one or more lines are too long

2
adaptor/AIComputing/modelarts/rpc/internal/server/modelartsserver.go
View File

@ -29,7 +29,7 @@ func (s *ModelArtsServer) GetToken(ctx context.Context, in *modelarts.TokenReq)
}
// creat task
func (s *ModelArtsServer) CreateTask(ctx context.Context, in *modelarts.ImportTaskData) (*modelarts.ExportTaskData, error) {
func (s *ModelArtsServer) CreateTask(ctx context.Context, in *modelarts.ImportTaskData) (*modelarts.ImprotTaskDataReq, error) {
l := logic.NewCreateTaskLogic(ctx, s.svcCtx)
return l.CreateTask(in)
}

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

@ -53,13 +53,11 @@ message TokenResp{
message ImportTaskData{
string dataset_id = 1;
string project_id = 2;
BodyTaskData bodyTaskData = 3;
}
message BodyTaskData{
string import_path =1;
string import_path = 3;
}
message ExportTaskData{
message ImprotTaskDataReq{
string task_id =1;
}
/******************createTaskData end*************************/
@ -298,7 +296,7 @@ service ModelArts {
rpc GetToken(TokenReq) returns (TokenResp);
//creat task
rpc createTask(ImportTaskData) returns(ExportTaskData);
rpc createTask(ImportTaskData) returns(ImprotTaskDataReq);
//find taskList
rpc GetDatasetList(ListImportTasks) returns (ReturnListImportTasks);