From 92428bcc069308d9fdc8d7a41893fe62c042858f Mon Sep 17 00:00:00 2001 From: tzwang Date: Thu, 24 Oct 2024 16:14:29 +0800 Subject: [PATCH] fix getadepterByModel bugs Former-commit-id: 1dc4ccd87fc22d7df3ebc427caede9aa69a82a0b --- .../logic/inference/getadaptersbymodellogic.go | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/internal/logic/inference/getadaptersbymodellogic.go b/internal/logic/inference/getadaptersbymodellogic.go index fe6a3c2b..fed1abde 100644 --- a/internal/logic/inference/getadaptersbymodellogic.go +++ b/internal/logic/inference/getadaptersbymodellogic.go @@ -38,7 +38,19 @@ func (l *GetAdaptersByModelLogic) GetAdaptersByModel(req *types.GetAdaptersByMod } for _, cluster := range clusters.List { - exist := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id][cluster.Id].CheckModelExistence(l.ctx, req.ModelName, req.ModelType) + + cmap, found := l.svcCtx.Scheduler.AiService.InferenceAdapterMap[adapter.Id] + if !found { + continue + } + + iCluster, found := cmap[cluster.Id] + if !found { + continue + } + + exist := iCluster.CheckModelExistence(l.ctx, req.ModelName, req.ModelType) + if exist { c := &types.ClusterAvail{ ClusterId: cluster.Id,