1

kb/controller/model_config_controller.go

@@ -0,0 +1,49 @@
+package controller
+
+import (
+	"context"
+
+	"rag-local/kb/model/dto"
+	"rag-local/kb/model/entity"
+	"rag-local/kb/service"
+)
+
+type modelConfig struct{}
+
+var ModelConfig = new(modelConfig)
+
+func (c *modelConfig) List(ctx context.Context, req *dto.ListModelConfigReq) (res *dto.ListModelConfigRes, err error) {
+	list, err := service.ModelConfigService.List(ctx, req.ModelType)
+	if err != nil {
+		return nil, err
+	}
+	return &dto.ListModelConfigRes{List: list}, nil
+}
+
+func (c *modelConfig) Save(ctx context.Context, req *dto.SaveModelConfigReq) (res *dto.SaveModelConfigRes, err error) {
+	id, err := service.ModelConfigService.Save(ctx, &entity.ModelConfig{
+		Id:          req.Id,
+		Name:        req.Name,
+		ModelType:   req.ModelType,
+		ModelName:   req.ModelName,
+		EndpointUrl: req.EndpointUrl,
+		ApiKey:      req.ApiKey,
+		Dimension:   req.Dimension,
+		Extra:       req.Extra,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &dto.SaveModelConfigRes{Id: id}, nil
+}
+
+func (c *modelConfig) Delete(ctx context.Context, req *dto.DeleteModelConfigReq) (res *dto.DeleteModelConfigRes, err error) {
+	return nil, service.ModelConfigService.Delete(ctx, req.Id)
+}
+
+func (c *modelConfig) Test(ctx context.Context, req *dto.TestModelConfigReq) (res *dto.TestModelConfigRes, err error) {
+	if err := service.ModelConfigService.Test(ctx, req.Id); err != nil {
+		return &dto.TestModelConfigRes{Ok: false, Msg: err.Error()}, nil
+	}
+	return &dto.TestModelConfigRes{Ok: true, Msg: "连接正常"}, nil
+}