From 2cd7f793b4d869cf202268877762bd7aafb7c1eb Mon Sep 17 00:00:00 2001 From: tzwang Date: Tue, 25 Jun 2024 15:53:12 +0800 Subject: [PATCH] fix octopus imageUrls bugs Former-commit-id: 80d170216054490c90601bb94085620e1edd70ec --- api/internal/storeLink/octopus.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/api/internal/storeLink/octopus.go b/api/internal/storeLink/octopus.go index 6219f86a..a902cf93 100644 --- a/api/internal/storeLink/octopus.go +++ b/api/internal/storeLink/octopus.go @@ -884,7 +884,7 @@ func (o *OctopusLink) GetImageInferUrl(ctx context.Context, option *option.Infer var imageUrls []*collector.ImageInferUrl for _, notebook := range list.Payload.GetNotebooks() { - if strings.Contains(notebook.AlgorithmName, option.ModelName) { + if strings.Contains(notebook.AlgorithmName, option.ModelName) && notebook.Status == "running" { url := strings.Replace(notebook.Tasks[0].Url, FORWARD_SLASH, "", -1) names := strings.Split(notebook.AlgorithmName, UNDERSCORE) imageUrl := &collector.ImageInferUrl{ @@ -896,5 +896,9 @@ func (o *OctopusLink) GetImageInferUrl(ctx context.Context, option *option.Infer continue } } + + if len(imageUrls) == 0 { + return nil, errors.New("no infer url available") + } return imageUrls, nil }