36 lines
979 B
Go
36 lines
979 B
Go
package controller
|
|
|
|
import (
|
|
"context"
|
|
|
|
"rag-local/kb/model/dto"
|
|
"rag-local/kb/service"
|
|
)
|
|
|
|
type systemConfig struct{}
|
|
|
|
var SystemConfig = new(systemConfig)
|
|
|
|
func (c *systemConfig) Login(ctx context.Context, req *dto.LoginReq) (res *dto.LoginRes, err error) {
|
|
token, err := service.SystemConfigService.Login(ctx, req.Token)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.LoginRes{Token: token}, nil
|
|
}
|
|
|
|
func (c *systemConfig) GetSettings(ctx context.Context, _ *dto.GetSettingsReq) (*dto.GetSettingsRes, error) {
|
|
size, overlap, err := service.SystemConfigService.GetSettings(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.GetSettingsRes{ChunkSize: size, ChunkOverlap: overlap}, nil
|
|
}
|
|
|
|
func (c *systemConfig) SaveSettings(ctx context.Context, req *dto.SaveSettingsReq) (*dto.SaveSettingsRes, error) {
|
|
if err := service.SystemConfigService.SaveSettings(ctx, req.ChunkSize, req.ChunkOverlap); err != nil {
|
|
return nil, err
|
|
}
|
|
return &dto.SaveSettingsRes{}, nil
|
|
}
|