root
f0ae237da5
Map imageQuality/imageCount and output_format across the HTTP and WebSocket handlers, pass them into the design service calls, and register the layered document generator in the service context. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
41 lines
1003 B
Go
41 lines
1003 B
Go
// Code scaffolded by goctl. Safe to edit.
|
|
// goctl 1.10.1
|
|
|
|
package logic
|
|
|
|
import (
|
|
"context"
|
|
|
|
"img_infinite_canvas/internal/svc"
|
|
"img_infinite_canvas/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GenerateProjectAsyncLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGenerateProjectAsyncLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GenerateProjectAsyncLogic {
|
|
return &GenerateProjectAsyncLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GenerateProjectAsyncLogic) GenerateProjectAsync(req *types.GenerateRequest) (resp *types.ProjectResponse, err error) {
|
|
if err := validateRequiredPrompt(req.Prompt, false); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
project, err := l.svcCtx.DesignService.GenerateAsync(l.ctx, req.Id, req.Prompt, req.Mode, req.ImageModel, req.ImageSize, req.ImageQuality, int(req.ImageCount))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return toProjectResponse(project), nil
|
|
}
|