92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
|
// Code scaffolded by goctl. Safe to edit.
|
||
|
|
// goctl 1.10.1
|
||
|
|
|
||
|
|
package brand
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
app "haixun-backend/internal/library/errors"
|
||
|
|
"haixun-backend/internal/library/errors/code"
|
||
|
|
libkg "haixun-backend/internal/library/knowledge"
|
||
|
|
"haixun-backend/internal/library/placement"
|
||
|
|
jobdom "haixun-backend/internal/model/job/domain/usecase"
|
||
|
|
"haixun-backend/internal/svc"
|
||
|
|
"haixun-backend/internal/types"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ExpandKnowledgeGraphLogic struct {
|
||
|
|
logx.Logger
|
||
|
|
ctx context.Context
|
||
|
|
svcCtx *svc.ServiceContext
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewExpandKnowledgeGraphLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ExpandKnowledgeGraphLogic {
|
||
|
|
return &ExpandKnowledgeGraphLogic{
|
||
|
|
Logger: logx.WithContext(ctx),
|
||
|
|
ctx: ctx,
|
||
|
|
svcCtx: svcCtx,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *ExpandKnowledgeGraphLogic) ExpandKnowledgeGraph(req *types.ExpandKnowledgeGraphHandlerReq) (resp *types.ExpandKnowledgeGraphData, err error) {
|
||
|
|
tenantID, uid, err := actorFrom(l.ctx)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
seed := strings.TrimSpace(req.SeedQuery)
|
||
|
|
supplemental := req.Supplemental
|
||
|
|
if seed == "" {
|
||
|
|
return nil, app.For(code.Brand).InputMissingRequired("seed_query is required")
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := l.svcCtx.Brand.Get(l.ctx, tenantID, uid, req.ID); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
research, err := l.svcCtx.Placement.ResearchSettings(l.ctx, tenantID, uid)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
expandStrategy := placement.EffectiveExpandStrategy(research)
|
||
|
|
if supplemental && placement.WebSearchAvailable(research) {
|
||
|
|
expandStrategy = libkg.ExpandStrategyBrave
|
||
|
|
}
|
||
|
|
memberCtx, err := l.svcCtx.ThreadsAccount.ResolveMemberPlacementContext(l.ctx, tenantID, uid, research)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
payload := map[string]any{
|
||
|
|
"brand_id": req.ID,
|
||
|
|
"seed_query": seed,
|
||
|
|
"supplemental": supplemental,
|
||
|
|
"regenerate_map": req.RegenerateMap,
|
||
|
|
"expand_strategy": expandStrategy.String(),
|
||
|
|
}
|
||
|
|
for key, value := range memberCtx.PayloadFields() {
|
||
|
|
payload[key] = value
|
||
|
|
}
|
||
|
|
|
||
|
|
run, err := l.svcCtx.Job.CreateRun(l.ctx, jobdom.CreateRunRequest{
|
||
|
|
TemplateType: "expand-graph",
|
||
|
|
Scope: "brand",
|
||
|
|
ScopeID: req.ID,
|
||
|
|
Payload: payload,
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
message := "研究地圖產生中，完成後可檢視延伸知識與參考連結"
|
||
|
|
|
||
|
|
return &types.ExpandKnowledgeGraphData{
|
||
|
|
JobID: run.ID.Hex(),
|
||
|
|
Status: string(run.Status),
|
||
|
|
Message: message,
|
||
|
|
}, nil
|
||
|
|
}
|