From 3d320d4ab06a58f0a2457eb3a0beb2e8d727273f Mon Sep 17 00:00:00 2001 From: tzwang Date: Thu, 11 Apr 2024 18:29:13 +0800 Subject: [PATCH] fix aiClusterMap bugs Former-commit-id: 9a2a5eb7bc32a4977149953a4c556743264e6c4a --- api/internal/scheduler/service/aiService.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) 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 } }