fetch model & add apitype

internal/controller/apikey.go

@@ -1,10 +1,13 @@
 package controller
 
 import (
+	"bytes"
+	"encoding/json"
 	"net/http"
 	"opencatd-open/internal/consts"
 	"opencatd-open/internal/dto"
 	"opencatd-open/internal/model"
+	"opencatd-open/internal/utils"
 	"strconv"
 	"strings"
 
@@ -18,13 +21,22 @@ func (a Api) CreateApiKey(c *gin.Context) {
 		dto.Fail(c, 403, "Permission denied")
 		return
 	}
-	req := new(model.ApiKey)
-	err := c.ShouldBind(&req)
+	newkey := new(model.ApiKey)
+	err := c.ShouldBind(newkey)
 	if err != nil {
 		dto.Fail(c, 400, err.Error())
 	}
+	if slice.Contain([]string{"openai", "azure", "claude"}, *newkey.ApiType) {
+		sma, err := utils.FetchKeyModel(a.db, newkey)
+		if err == nil {
+			newkey.SupportModelsArray = sma
+			var buf = new(bytes.Buffer)
+			json.NewEncoder(buf).Encode(sma) //nolint:errcheck
+			newkey.SupportModels = utils.ToPtr(buf.String())
+		}
+	}
 
-	err = a.keyService.CreateApiKey(c, req)
+	err = a.keyService.CreateApiKey(c, newkey)
 	if err != nil {
 		dto.Fail(c, 400, err.Error())
 	} else {
@@ -79,6 +91,10 @@ func (a Api) ListApiKey(c *gin.Context) {
 			}
 			str = str[:slen]
 			key.ApiKey = &str
+
+			var sma []string
+			json.NewDecoder(strings.NewReader(*key.SupportModels)).Decode(&sma) //nolint:errcheck
+			key.SupportModelsArray = sma
 		}
 		dto.Success(c, gin.H{
 			"total": total,