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 // 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) l := logic.NewCreateTaskLogic(ctx, s.svcCtx)
return l.CreateTask(in) return l.CreateTask(in)
} }

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

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