|
|
|
@@ -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 |
|
|
|
} |