41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
|
package inspire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"apps/backend/internal/middleware"
|
||
|
|
"apps/backend/internal/response"
|
||
|
|
"apps/backend/internal/svc"
|
||
|
|
"apps/backend/internal/types"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type InspireChatLogic struct {
|
||
|
|
logx.Logger
|
||
|
|
ctx context.Context
|
||
|
|
svcCtx *svc.ServiceContext
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewInspireChatLogic(ctx context.Context, svcCtx *svc.ServiceContext) *InspireChatLogic {
|
||
|
|
return &InspireChatLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *InspireChatLogic) InspireChat(req *types.InspireChatReq) (*types.InspireChatData, error) {
|
||
|
|
|
||
|
|
if l.svcCtx.Inspire == nil {
|
||
|
|
return nil, response.Biz(503, 503001, "inspire not configured")
|
||
|
|
}
|
||
|
|
uid, ok := middleware.UIDFrom(l.ctx)
|
||
|
|
if !ok {
|
||
|
|
return nil, response.Biz(401, 401001, "missing authorization")
|
||
|
|
}
|
||
|
|
out, err := l.svcCtx.Inspire.Chat(l.ctx, uid, req.Message, req.PinnedIds, req.Mode, req.PersonaId, req.SessionId, req.Material)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
sess := types.SessionFromDomain(out.Session)
|
||
|
|
return &types.InspireChatData{Session: sess, Messages: sess.Messages}, nil
|
||
|
|
|
||
|
|
}
|