| @@ -18,7 +18,7 @@ require ( | |||||
| github.com/prometheus/common v0.61.0 | github.com/prometheus/common v0.61.0 | ||||
| github.com/robfig/cron/v3 v3.0.1 | github.com/robfig/cron/v3 v3.0.1 | ||||
| github.com/zeromicro/go-zero v1.7.4 | github.com/zeromicro/go-zero v1.7.4 | ||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20241202005821-e7089b705a97 | |||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20250107025835-8fc888b1d170 | |||||
| gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4 | gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4 | ||||
| gitlink.org.cn/JointCloud/pcm-modelarts v0.0.0-20241211021152-9771ba5670b7 | gitlink.org.cn/JointCloud/pcm-modelarts v0.0.0-20241211021152-9771ba5670b7 | ||||
| gitlink.org.cn/JointCloud/pcm-octopus v0.0.0-20240817071412-44397870b110 | gitlink.org.cn/JointCloud/pcm-octopus v0.0.0-20240817071412-44397870b110 | ||||
| @@ -524,8 +524,8 @@ github.com/yuin/gopher-lua v1.1.1 h1:kYKnWBjvbNP4XLT3+bPEwAXJx262OhaHDWDVOPjL46M | |||||
| github.com/yuin/gopher-lua v1.1.1/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw= | github.com/yuin/gopher-lua v1.1.1/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw= | ||||
| github.com/zeromicro/go-zero v1.7.4 h1:lyIUsqbpVRzM4NmXu5pRM3XrdRdUuWOkQmHiNmJF0VU= | github.com/zeromicro/go-zero v1.7.4 h1:lyIUsqbpVRzM4NmXu5pRM3XrdRdUuWOkQmHiNmJF0VU= | ||||
| github.com/zeromicro/go-zero v1.7.4/go.mod h1:jmv4hTdUBkDn6kxgI+WrKQw0q6LKxDElGPMfCLOeeEY= | github.com/zeromicro/go-zero v1.7.4/go.mod h1:jmv4hTdUBkDn6kxgI+WrKQw0q6LKxDElGPMfCLOeeEY= | ||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20241202005821-e7089b705a97 h1:m4eVyO1NkPk5Bqsr2htNLPfIpkUeF2GJN3scTV8BgQ4= | |||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20241202005821-e7089b705a97/go.mod h1:3eECiw9O2bIFkkePlloKyLNXiqBAhOxNrDoGaaGseGY= | |||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20250107025835-8fc888b1d170 h1:/n3pl6WuHk6hmkIs0LF5um9WreggL2QuDIOch10ELEI= | |||||
| gitlink.org.cn/JointCloud/pcm-ac v0.0.0-20250107025835-8fc888b1d170/go.mod h1:3eECiw9O2bIFkkePlloKyLNXiqBAhOxNrDoGaaGseGY= | |||||
| gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4 h1:WIs/189lRLNMXF2ui/Wm1+Y55eJ53BVGx+4+gdn9cls= | gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4 h1:WIs/189lRLNMXF2ui/Wm1+Y55eJ53BVGx+4+gdn9cls= | ||||
| gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4/go.mod h1:YbuoRgF9sEVvNJPQtGRjdocX7Du6NBOTLn+GVwqRVjo= | gitlink.org.cn/JointCloud/pcm-hpc v0.0.0-20241125115811-72f3568255a4/go.mod h1:YbuoRgF9sEVvNJPQtGRjdocX7Du6NBOTLn+GVwqRVjo= | ||||
| gitlink.org.cn/JointCloud/pcm-modelarts v0.0.0-20241211021152-9771ba5670b7 h1:sPyw2qIID5vY/eJBY9UiHBzIF/gk0GUhei9sRIADrE4= | gitlink.org.cn/JointCloud/pcm-modelarts v0.0.0-20241211021152-9771ba5670b7 h1:sPyw2qIID5vY/eJBY9UiHBzIF/gk0GUhei9sRIADrE4= | ||||
| @@ -604,7 +604,17 @@ func (s *ShuguangAi) GetTrainingTask(ctx context.Context, taskId string) (*colle | |||||
| } | } | ||||
| func (s *ShuguangAi) Stop(ctx context.Context, id string) error { | func (s *ShuguangAi) Stop(ctx context.Context, id string) error { | ||||
| return errors.New("failed to implement") | |||||
| req := &hpcAC.StopTaskAiReq{ | |||||
| Id: id, | |||||
| } | |||||
| resp, err := s.aCRpc.StopTaskAi(ctx, req) | |||||
| if err != nil { | |||||
| return err | |||||
| } | |||||
| if resp.Code != "0" { | |||||
| return errors.New(resp.Msg) | |||||
| } | |||||
| return nil | |||||
| } | } | ||||
| func (s *ShuguangAi) Execute(ctx context.Context, option *option.AiOption, mode int) (interface{}, error) { | func (s *ShuguangAi) Execute(ctx context.Context, option *option.AiOption, mode int) (interface{}, error) { | ||||
| @@ -115,10 +115,10 @@ var ( | |||||
| ERROR_CONVERT_EMPTY = errors.New("convert empty error") | ERROR_CONVERT_EMPTY = errors.New("convert empty error") | ||||
| StorageTypeMap = map[string]string{ | StorageTypeMap = map[string]string{ | ||||
| OCTOPUS: "JCS", | |||||
| MODELARTS: "API", | |||||
| OPENI: "API", | |||||
| SHUGUANGAI: "API", | |||||
| OCTOPUS: "FTP", | |||||
| MODELARTS: "JCS", | |||||
| OPENI: "URL", | |||||
| SHUGUANGAI: "EFILE", | |||||
| } | } | ||||
| ClusterTypeMap = map[string]string{ | ClusterTypeMap = map[string]string{ | ||||