60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
|
package compose
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"apps/backend/internal/middleware"
|
|||
|
|
"apps/backend/internal/response"
|
|||
|
|
"apps/backend/internal/svc"
|
|||
|
|
"apps/backend/internal/types"
|
|||
|
|
|
|||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ComposeMimicLogic struct {
|
|||
|
|
logx.Logger
|
|||
|
|
ctx context.Context
|
|||
|
|
svcCtx *svc.ServiceContext
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewComposeMimicLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ComposeMimicLogic {
|
|||
|
|
return &ComposeMimicLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (l *ComposeMimicLogic) ComposeMimic(req *types.ComposeMimicReq) (*types.ComposeMimicData, error) {
|
|||
|
|
uid, ok := middleware.UIDFrom(l.ctx)
|
|||
|
|
if !ok {
|
|||
|
|
return nil, response.Biz(401, 401001, "missing authorization")
|
|||
|
|
}
|
|||
|
|
src := strings.TrimSpace(req.SourceText)
|
|||
|
|
if src == "" {
|
|||
|
|
return nil, response.Biz(400, 400001, "source_text is required")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 優先背景 Job（避免 HTTP 120s 逾時）
|
|||
|
|
if l.svcCtx.Jobs != nil {
|
|||
|
|
lang := "zh-TW"
|
|||
|
|
if v := l.ctx.Value("lang"); v != nil {
|
|||
|
|
if s, ok := v.(string); ok && s != "" {
|
|||
|
|
lang = s
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
j, err := l.svcCtx.Jobs.ScheduleComposeMimic(l.ctx, uid, src, req.PersonaId, req.StructureNotes, lang)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return &types.ComposeMimicData{JobId: j.ID, Async: true}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 無 Jobs（測試／降級）：同步執行
|
|||
|
|
if l.svcCtx.Studio == nil {
|
|||
|
|
return nil, response.Biz(503, 503001, "studio not configured")
|
|||
|
|
}
|
|||
|
|
text, err := l.svcCtx.Studio.Mimic(l.ctx, uid, src, req.PersonaId, req.StructureNotes)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return &types.ComposeMimicData{Text: text, Async: false}, nil
|
|||
|
|
}
|