You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.6 KiB
61 lines
1.6 KiB
// Code scaffolded by goctl. Safe to edit.
|
|
// goctl 1.9.2
|
|
|
|
package ai
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/youruser/base/internal/svc"
|
|
"github.com/youruser/base/internal/types"
|
|
"github.com/youruser/base/model"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type AiProviderUpdateLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
// 更新AI平台
|
|
func NewAiProviderUpdateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AiProviderUpdateLogic {
|
|
return &AiProviderUpdateLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *AiProviderUpdateLogic) AiProviderUpdate(req *types.AIProviderUpdateRequest) (resp *types.AIProviderInfo, err error) {
|
|
provider, err := model.AIProviderFindOne(l.ctx, l.svcCtx.DB, req.Id)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("AI平台不存在: %v", err)
|
|
}
|
|
|
|
provider.DisplayName = req.DisplayName
|
|
provider.BaseUrl = req.BaseUrl
|
|
provider.SdkType = req.SdkType
|
|
provider.Protocol = req.Protocol
|
|
provider.IsActive = req.IsActive
|
|
provider.SortOrder = req.SortOrder
|
|
|
|
if err := model.AIProviderUpdate(l.ctx, l.svcCtx.DB, provider); err != nil {
|
|
return nil, fmt.Errorf("更新AI平台失败: %v", err)
|
|
}
|
|
|
|
return &types.AIProviderInfo{
|
|
Id: provider.Id,
|
|
Name: provider.Name,
|
|
DisplayName: provider.DisplayName,
|
|
BaseUrl: provider.BaseUrl,
|
|
SdkType: provider.SdkType,
|
|
Protocol: provider.Protocol,
|
|
IsActive: provider.IsActive,
|
|
SortOrder: provider.SortOrder,
|
|
CreatedAt: provider.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
UpdatedAt: provider.UpdatedAt.Format("2006-01-02 15:04:05"),
|
|
}, nil
|
|
}
|
|
|