diff --git a/api/internal/scheduler/service/aiService.go b/api/internal/scheduler/service/aiService.go index aa8ded01..9fa480c7 100644 --- a/api/internal/scheduler/service/aiService.go +++ b/api/internal/scheduler/service/aiService.go @@ -28,18 +28,18 @@ func InitAiClusterMap(octopusRpc octopusclient.Octopus, modelArtsRpc modelartsse case OCTOPUS: id, _ := strconv.ParseInt(c.Id, 10, 64) octopus := storeLink.NewOctopusLink(octopusRpc, c.Nickname, id) - collectorMap[c.Nickname] = octopus - executorMap[c.Nickname] = octopus + collectorMap[c.Id] = octopus + executorMap[c.Id] = octopus case MODELARTS: id, _ := strconv.ParseInt(c.Id, 10, 64) modelarts := storeLink.NewModelArtsLink(modelArtsRpc, modelArtsImgRpc, c.Nickname, id) - collectorMap[c.Nickname] = modelarts - executorMap[c.Nickname] = modelarts + collectorMap[c.Id] = modelarts + executorMap[c.Id] = modelarts case SHUGUANGAI: id, _ := strconv.ParseInt(c.Id, 10, 64) sgai := storeLink.NewShuguangAi(aCRpc, c.Nickname, id) - collectorMap[c.Nickname] = sgai - executorMap[c.Nickname] = sgai + collectorMap[c.Id] = sgai + executorMap[c.Id] = sgai } }