Browse Source

fix octopus imageUrls bugs

Former-commit-id: 80d1702160
pull/243/head
tzwang 1 year ago
parent
commit
2cd7f793b4
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      api/internal/storeLink/octopus.go

+ 5
- 1
api/internal/storeLink/octopus.go View File

@@ -884,7 +884,7 @@ func (o *OctopusLink) GetImageInferUrl(ctx context.Context, option *option.Infer


var imageUrls []*collector.ImageInferUrl var imageUrls []*collector.ImageInferUrl
for _, notebook := range list.Payload.GetNotebooks() { 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) url := strings.Replace(notebook.Tasks[0].Url, FORWARD_SLASH, "", -1)
names := strings.Split(notebook.AlgorithmName, UNDERSCORE) names := strings.Split(notebook.AlgorithmName, UNDERSCORE)
imageUrl := &collector.ImageInferUrl{ imageUrl := &collector.ImageInferUrl{
@@ -896,5 +896,9 @@ func (o *OctopusLink) GetImageInferUrl(ctx context.Context, option *option.Infer
continue continue
} }
} }

if len(imageUrls) == 0 {
return nil, errors.New("no infer url available")
}
return imageUrls, nil return imageUrls, nil
} }

Loading…
Cancel
Save