From 6dad3539457c8baffedbb73a47a370184deeefc5 Mon Sep 17 00:00:00 2001
From: qiwang <1364512070@qq.com>
Date: Sat, 14 Sep 2024 17:01:21 +0800
Subject: [PATCH] fix: add modelarts of ai CheckModelExistence

---
 internal/storeLink/modelarts.go | 29 ++++++++++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go
index acc624d5..d718e8cb 100644
--- a/internal/storeLink/modelarts.go
+++ b/internal/storeLink/modelarts.go
@@ -812,10 +812,37 @@ func (m *ModelArtsLink) CheckModelExistence(ctx context.Context, name string, mt
 		ModelName: name,
 		ModelType: mtype,
 	}
-	err := m.GetModelId(ctx, ifoption)
+	err := m.CheckImageExist(ctx, ifoption)
 	if err != nil {
 		return false
 	}
 
 	return true
 }
+
+func (m *ModelArtsLink) CheckImageExist(ctx context.Context, option *option.InferOption) error {
+	req := &modelarts.ListImagesReq{
+		Limit:  m.pageSize,
+		Offset: m.pageIndex,
+	}
+	ListImageResp, err := m.modelArtsRpc.ListImages(ctx, req)
+	if err != nil {
+		return err
+	}
+	var modelName string
+	if ListImageResp.Code == 200 {
+		//return errors.New("failed to get ModelId")
+		for _, ListImage := range ListImageResp.Data {
+			if option.ModelName == "ChatGLM-6B" {
+				modelName = "chatglm-6b"
+			} else {
+				modelName = option.ModelName
+			}
+
+			if ListImage.Name == modelName {
+				return nil
+			}
+		}
+	}
+	return errors.New("failed to find Image ")
+}