parent
6507272923
commit
3f57e45619
@ -0,0 +1,22 @@
|
|||||||
|
package web
|
||||||
|
|
||||||
|
import (
|
||||||
|
"goskeleton/app/global/consts"
|
||||||
|
"goskeleton/app/service/ai_model_cli"
|
||||||
|
"goskeleton/app/utils/response"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DocRefine struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ai 模型进行文档优化
|
||||||
|
|
||||||
|
func (u *DocRefine) DocRefine(context *gin.Context) {
|
||||||
|
if r, recogWords := ai_model_cli.RequestQianFan(context); r {
|
||||||
|
response.Success(context, consts.CurdStatusOkMsg, recogWords)
|
||||||
|
} else {
|
||||||
|
response.Fail(context, consts.DocRefineFailCode, consts.DocRefineFailMsg, "")
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,47 @@
|
|||||||
|
package ai_doc
|
||||||
|
|
||||||
|
import (
|
||||||
|
"goskeleton/app/global/consts"
|
||||||
|
"goskeleton/app/http/controller/web"
|
||||||
|
"goskeleton/app/http/validator/core/data_transfer"
|
||||||
|
"goskeleton/app/utils/response"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DocRefine struct {
|
||||||
|
Type string `form:"type" json:"type" binding:"required"` // 必填、
|
||||||
|
Doc string `form:"doc" json:"doc" binding:"required"` // 必填、对于文本,表示它的长度>=1
|
||||||
|
Background string `form:"background" json:"background" `
|
||||||
|
}
|
||||||
|
|
||||||
|
var Types = []string{"abstract", "decorate", "sequel_writing", "rewrite_wrong", "translate"}
|
||||||
|
|
||||||
|
func (d DocRefine) CheckParams(context *gin.Context) {
|
||||||
|
if err := context.ShouldBind(&d); err != nil {
|
||||||
|
// 将表单参数验证器出现的错误直接交给错误翻译器统一处理即可
|
||||||
|
response.ValidatorError(context, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 判断是否在类型中
|
||||||
|
t := d.Type
|
||||||
|
isExit := false
|
||||||
|
for _, e := range Types {
|
||||||
|
if strings.TrimSpace(t) == e {
|
||||||
|
isExit = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !isExit {
|
||||||
|
response.ErrorSystem(context, "DocRefine表单参数验证器json化失败", "")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// 该函数主要是将本结构体的字段（成员）按照 consts.ValidatorPrefix+ json标签对应的 键 => 值 形式绑定在上下文，便于下一步（控制器）可以直接通过 context.Get(键) 获取相关值
|
||||||
|
extraAddBindDataContext := data_transfer.DataAddContext(d, consts.ValidatorPrefix, context)
|
||||||
|
if extraAddBindDataContext == nil {
|
||||||
|
response.ErrorSystem(context, "DocRefine表单参数验证器json化失败", "")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
(&web.DocRefine{}).DocRefine(extraAddBindDataContext)
|
||||||
|
}
|
||||||
Loading…
Reference in new issue