From c0b64237badf595a62625ec1e50bf71ab01f7575 Mon Sep 17 00:00:00 2001 From: tzwang Date: Tue, 25 Jun 2024 21:33:28 +0800 Subject: [PATCH] updated textinfer api Former-commit-id: 0ae4e41fae1f4bd8ee8d8c6185cd43e38c3d6068 --- api/internal/logic/inference/imageinferencelogic.go | 4 ++++ api/internal/storeLink/octopus.go | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/api/internal/logic/inference/imageinferencelogic.go b/api/internal/logic/inference/imageinferencelogic.go index 48aaf97a..25036805 100644 --- a/api/internal/logic/inference/imageinferencelogic.go +++ b/api/internal/logic/inference/imageinferencelogic.go @@ -8,6 +8,7 @@ import ( "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/scheduler/schedulers/option" "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/scheduler/service/collector" "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/scheduler/strategy" + "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/storeLink" "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/svc" "gitlink.org.cn/JointCloud/pcm-coordinator/api/internal/types" "gitlink.org.cn/JointCloud/pcm-coordinator/pkg/constants" @@ -183,6 +184,9 @@ func infer(opt *option.InferOption, clusters []*strategy.AssignedCluster, ts []s c := cluster go func() { imageUrls, err := collectorMap[c.ClusterId].GetInferUrl(ctx, opt) + for i, _ := range imageUrls { + imageUrls[i].Url = imageUrls[i].Url + storeLink.FORWARD_SLASH + "image" + } if err != nil { wg.Done() return diff --git a/api/internal/storeLink/octopus.go b/api/internal/storeLink/octopus.go index 9c605345..ed706c0a 100644 --- a/api/internal/storeLink/octopus.go +++ b/api/internal/storeLink/octopus.go @@ -888,7 +888,7 @@ func (o *OctopusLink) GetInferUrl(ctx context.Context, option *option.InferOptio url := strings.Replace(notebook.Tasks[0].Url, FORWARD_SLASH, "", -1) names := strings.Split(notebook.AlgorithmName, UNDERSCORE) imageUrl := &collector.InferUrl{ - Url: DOMAIN + url + FORWARD_SLASH + "image", + Url: DOMAIN + url, Card: names[2], } imageUrls = append(imageUrls, imageUrl)