base/backend/internal/logic/ai/aimodelcreatelogic.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 AiModelCreateLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 创建AI模型
func NewAiModelCreateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AiModelCreateLogic {
	return &AiModelCreateLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *AiModelCreateLogic) AiModelCreate(req *types.AIModelCreateRequest) (resp *types.AIModelInfo, err error) {
	aiModel := &model.AIModel{
		ProviderId:     req.ProviderId,
		ModelId:        req.ModelId,
		DisplayName:    req.DisplayName,
		InputPrice:     req.InputPrice,
		OutputPrice:    req.OutputPrice,
		MaxTokens:      req.MaxTokens,
		ContextWindow:  req.ContextWindow,
		SupportsStream: req.SupportsStream,
		SupportsVision: req.SupportsVision,
		IsActive:       true,
	}

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

	// 查询刚创建的记录以获取完整字段
	aiModel, err = model.AIModelFindOne(l.ctx, l.svcCtx.DB, id)
	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
}