finish ai key api

controller/ai-key/controller.go

@@ -0,0 +1,26 @@
+package ai_key
+
+import (
+	"reflect"
+
+	ai_key_dto "github.com/APIParkLab/APIPark/module/ai-key/dto"
+	"github.com/eolinker/go-common/autowire"
+	"github.com/gin-gonic/gin"
+)
+
+type IKeyController interface {
+	Create(ctx *gin.Context, providerId string, input *ai_key_dto.Create) error
+	Edit(ctx *gin.Context, providerId string, id string, input *ai_key_dto.Edit) error
+	Delete(ctx *gin.Context, providerId string, id string) error
+	Get(ctx *gin.Context, providerId string, id string) (*ai_key_dto.Key, error)
+	List(ctx *gin.Context, providerId string, keyword string, page string, pageSize string) ([]*ai_key_dto.Item, int64, error)
+	Enable(ctx *gin.Context, providerId string, id string) error
+	Disable(ctx *gin.Context, providerId string, id string) error
+	Sort(ctx *gin.Context, providerId string, input *ai_key_dto.Sort) error
+}
+
+func init() {
+	autowire.Auto[IKeyController](func() reflect.Value {
+		return reflect.ValueOf(new(imlAIKeyController))
+	})
+}

controller/ai-key/iml.go

@@ -0,0 +1,61 @@
+package ai_key
+
+import (
+	"strconv"
+
+	ai_key "github.com/APIParkLab/APIPark/module/ai-key"
+	ai_key_dto "github.com/APIParkLab/APIPark/module/ai-key/dto"
+	"github.com/gin-gonic/gin"
+)
+
+var _ IKeyController = &imlAIKeyController{}
+
+type imlAIKeyController struct {
+	module ai_key.IKeyModule `autowired:""`
+}
+
+func (i *imlAIKeyController) Enable(ctx *gin.Context, providerId string, id string) error {
+	return i.module.UpdateKeyStatus(ctx, providerId, id, true)
+}
+
+func (i *imlAIKeyController) Disable(ctx *gin.Context, providerId string, id string) error {
+	return i.module.UpdateKeyStatus(ctx, providerId, id, false)
+}
+
+func (i *imlAIKeyController) Create(ctx *gin.Context, providerId string, input *ai_key_dto.Create) error {
+	return i.module.Create(ctx, providerId, input)
+}
+
+func (i *imlAIKeyController) Edit(ctx *gin.Context, providerId string, id string, input *ai_key_dto.Edit) error {
+	return i.module.Edit(ctx, providerId, id, input)
+}
+
+func (i *imlAIKeyController) Delete(ctx *gin.Context, providerId string, id string) error {
+	return i.module.Delete(ctx, providerId, id)
+}
+
+func (i *imlAIKeyController) Get(ctx *gin.Context, providerId string, id string) (*ai_key_dto.Key, error) {
+	return i.module.Get(ctx, providerId, id)
+}
+
+func (i *imlAIKeyController) List(ctx *gin.Context, providerId string, keyword string, page string, pageSize string) ([]*ai_key_dto.Item, int64, error) {
+	p, err := strconv.Atoi(page)
+	if err != nil {
+		if page != "" {
+			return nil, 0, err
+		}
+		p = 1
+	}
+	ps, err := strconv.Atoi(pageSize)
+	if err != nil {
+		if pageSize != "" {
+			return nil, 0, err
+		}
+		ps = 15
+	}
+	return i.module.List(ctx, providerId, keyword, p, ps)
+}
+
+func (i *imlAIKeyController) Sort(ctx *gin.Context, providerId string, input *ai_key_dto.Sort) error {
+	return i.module.Sort(ctx, providerId, input)
+}

controller/ai/controller.go

@@ -1,14 +1,18 @@
 package ai
 
 import (
+	"reflect"
+
+	"github.com/eolinker/go-common/auto"
+
 	ai_dto "github.com/APIParkLab/APIPark/module/ai/dto"
 	"github.com/eolinker/go-common/autowire"
 	"github.com/gin-gonic/gin"
-	"reflect"
 )
 
 type IProviderController interface {
-	Providers(ctx *gin.Context) ([]*ai_dto.ProviderItem, error)
+	ConfiguredProviders(ctx *gin.Context) ([]*ai_dto.ConfiguredProviderItem, *auto.Label, error)
+	UnConfiguredProviders(ctx *gin.Context) ([]*ai_dto.ProviderItem, error)
 	SimpleProviders(ctx *gin.Context) ([]*ai_dto.SimpleProviderItem, error)
 	Provider(ctx *gin.Context, id string) (*ai_dto.Provider, error)
 	LLMs(ctx *gin.Context, driver string) ([]*ai_dto.LLMItem, *ai_dto.ProviderItem, error)

controller/ai/iml.go

@@ -3,6 +3,7 @@ package ai
 import (
 	"github.com/APIParkLab/APIPark/module/ai"
 	ai_dto "github.com/APIParkLab/APIPark/module/ai/dto"
+	"github.com/eolinker/go-common/auto"
 	"github.com/gin-gonic/gin"
 )
 
@@ -14,12 +15,16 @@ type imlProviderController struct {
 	module ai.IProviderModule `autowired:""`
 }
 
-func (i *imlProviderController) SimpleProviders(ctx *gin.Context) ([]*ai_dto.SimpleProviderItem, error) {
-	return i.module.SimpleProviders(ctx)
+func (i *imlProviderController) ConfiguredProviders(ctx *gin.Context) ([]*ai_dto.ConfiguredProviderItem, *auto.Label, error) {
+	return i.module.ConfiguredProviders(ctx)
 }
 
-func (i *imlProviderController) Providers(ctx *gin.Context) ([]*ai_dto.ProviderItem, error) {
-	return i.module.Providers(ctx)
+func (i *imlProviderController) UnConfiguredProviders(ctx *gin.Context) ([]*ai_dto.ProviderItem, error) {
+	return i.module.UnConfiguredProviders(ctx)
+}
+
+func (i *imlProviderController) SimpleProviders(ctx *gin.Context) ([]*ai_dto.SimpleProviderItem, error) {
+	return i.module.SimpleProviders(ctx)
 }
 
 func (i *imlProviderController) Provider(ctx *gin.Context, id string) (*ai_dto.Provider, error) {