41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
|
package settings
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"apps/backend/internal/middleware"
|
||
|
|
"apps/backend/internal/module/member/domain"
|
||
|
|
"apps/backend/internal/svc"
|
||
|
|
"apps/backend/internal/types"
|
||
|
|
|
||
|
|
"github.com/zeromicro/go-zero/core/logx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type GetAiLogic struct {
|
||
|
|
logx.Logger
|
||
|
|
ctx context.Context
|
||
|
|
svcCtx *svc.ServiceContext
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGetAiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAiLogic {
|
||
|
|
return &GetAiLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (l *GetAiLogic) GetAi() (resp *types.AiSettingsData, err error) {
|
||
|
|
uid, _ := middleware.UIDFrom(l.ctx)
|
||
|
|
st, err := l.svcCtx.Members.GetSettings(l.ctx, uid)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
if st == nil {
|
||
|
|
st = domain.DefaultSettings(uid)
|
||
|
|
}
|
||
|
|
return &types.AiSettingsData{
|
||
|
|
Provider: st.Provider, Model: st.Model,
|
||
|
|
ResearchProvider: st.Provider, ResearchModel: st.Model,
|
||
|
|
ApiKeyConfigured: st.APIKeyConfigured,
|
||
|
|
ResearchApiKeyConfigured: st.ResearchAPIKeyConfigured || st.APIKeyConfigured,
|
||
|
|
PlatformKeyAvailable: l.svcCtx.Config.Platform.AIKey != "",
|
||
|
|
}, nil
|
||
|
|
}
|