base/backend/internal/logic/ai/aimodelupdatelogic.go

// 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 AiModelUpdateLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 更新AI模型
func NewAiModelUpdateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AiModelUpdateLogic {
	return &AiModelUpdateLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AiModelUpdateLogic) AiModelUpdate(req *types.AIModelUpdateRequest) (resp *types.AIModelInfo, err error) {
	// 查询现有记录
	aiModel, err := model.AIModelFindOne(l.ctx, l.svcCtx.DB, req.Id)
	if err != nil {
		return nil, fmt.Errorf("查询AI模型失败: %v", err)
	}

	// 更新字段
	aiModel.DisplayName = req.DisplayName
	aiModel.InputPrice = req.InputPrice
	aiModel.OutputPrice = req.OutputPrice
	aiModel.MaxTokens = req.MaxTokens
	aiModel.ContextWindow = req.ContextWindow
	aiModel.SupportsStream = req.SupportsStream
	aiModel.SupportsVision = req.SupportsVision
	aiModel.IsActive = req.IsActive

	err = model.AIModelUpdate(l.ctx, l.svcCtx.DB, aiModel)
	if err != nil {
		return nil, fmt.Errorf("更新AI模型失败: %v", err)
	}

	// 查询供应商名称
	providerName := ""
	provider, err := model.AIProviderFindOne(l.ctx, l.svcCtx.DB, aiModel.ProviderId)
	if err == nil {
		providerName = provider.DisplayName
	}

	resp = &types.AIModelInfo{
		Id:             aiModel.Id,
		ProviderId:     aiModel.ProviderId,
		ProviderName:   providerName,
		ModelId:        aiModel.ModelId,
		DisplayName:    aiModel.DisplayName,
		InputPrice:     aiModel.InputPrice,
		OutputPrice:    aiModel.OutputPrice,
		MaxTokens:      aiModel.MaxTokens,
		ContextWindow:  aiModel.ContextWindow,
		SupportsStream: aiModel.SupportsStream,
		SupportsVision: aiModel.SupportsVision,
	}

	return resp, nil
}