mirror of
https://github.com/APIParkLab/APIPark.git
synced 2026-06-14 20:41:15 +08:00
Liujian
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package ai
|
|
|
|
import (
|
|
"github.com/APIParkLab/APIPark/module/ai"
|
|
ai_dto "github.com/APIParkLab/APIPark/module/ai/dto"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
var (
|
|
_ IProviderController = (*imlProviderController)(nil)
|
|
)
|
|
|
|
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) Providers(ctx *gin.Context) ([]*ai_dto.ProviderItem, error) {
|
|
return i.module.Providers(ctx)
|
|
}
|
|
|
|
func (i *imlProviderController) Provider(ctx *gin.Context, id string) (*ai_dto.Provider, error) {
|
|
return i.module.Provider(ctx, id)
|
|
}
|
|
|
|
func (i *imlProviderController) LLMs(ctx *gin.Context, driver string) ([]*ai_dto.LLMItem, *ai_dto.ProviderItem, error) {
|
|
return i.module.LLMs(ctx, driver)
|
|
}
|
|
|
|
func (i *imlProviderController) Enable(ctx *gin.Context, id string) error {
|
|
return i.module.UpdateProviderStatus(ctx, id, true)
|
|
}
|
|
|
|
func (i *imlProviderController) Disable(ctx *gin.Context, id string) error {
|
|
return i.module.UpdateProviderStatus(ctx, id, false)
|
|
}
|
|
|
|
func (i *imlProviderController) UpdateProviderConfig(ctx *gin.Context, id string, input *ai_dto.UpdateConfig) error {
|
|
return i.module.UpdateProviderConfig(ctx, id, input)
|
|
}
|
|
|
|
func (i *imlProviderController) UpdateProviderDefaultLLM(ctx *gin.Context, id string, input *ai_dto.UpdateLLM) error {
|
|
return i.module.UpdateProviderDefaultLLM(ctx, id, input)
|
|
}
|