完成文本审查服务开发

api/core/config.go

@@ -148,14 +148,24 @@ func LoadSystemConfig(db *gorm.DB) *types.SystemConfig {
 		logger.Error("load payment config error: ", err)
 	}
 
+	// 加载文本审查配置
+	var moderationConfig types.ModerationConfig
+	sysConfig.Id = 0
+	db.Where("name", types.ConfigKeyModeration).First(&sysConfig)
+	err = utils.JsonDecode(sysConfig.Value, &moderationConfig)
+	if err != nil {
+		logger.Error("load moderation config error: ", err)
+	}
+
 	return &types.SystemConfig{
-		Base:    baseConfig,
-		License: license,
-		SMS:     smsConfig,
-		OSS:     ossConfig,
-		SMTP:    smtpConfig,
-		Payment: paymentConfig,
-		Captcha: captchaConfig,
-		WxLogin: wxLoginConfig,
+		Base:       baseConfig,
+		License:    license,
+		SMS:        smsConfig,
+		OSS:        ossConfig,
+		SMTP:       smtpConfig,
+		Payment:    paymentConfig,
+		Captcha:    captchaConfig,
+		WxLogin:    wxLoginConfig,
+		Moderation: moderationConfig,
 	}
 }

api/core/types/config.go

@@ -100,29 +100,31 @@ type BaseConfig struct {
 }
 
 type SystemConfig struct {
-	Base    BaseConfig
-	Payment PaymentConfig
-	OSS     OSSConfig
-	SMS     SMSConfig
-	SMTP    SmtpConfig
-	Captcha CaptchaConfig
-	WxLogin WxLoginConfig
-	Jimeng  JimengConfig
-	License License
+	Base       BaseConfig
+	Payment    PaymentConfig
+	OSS        OSSConfig
+	SMS        SMSConfig
+	SMTP       SmtpConfig
+	Captcha    CaptchaConfig
+	WxLogin    WxLoginConfig
+	Jimeng     JimengConfig
+	License    License
+	Moderation ModerationConfig
 }
 
 // 配置键名常量
 const (
-	ConfigKeySystem    = "system"
-	ConfigKeyNotice    = "notice"
-	ConfigKeyAgreement = "agreement"
-	ConfigKeyPrivacy   = "privacy"
-	ConfigKeyMarkMap   = "mark_map"
-	ConfigKeyCaptcha   = "captcha"
-	ConfigKeyWxLogin   = "wx_login"
-	ConfigKeyLicense   = "license"
-	ConfigKeySms       = "sms"
-	ConfigKeySmtp      = "smtp"
-	ConfigKeyOss       = "oss"
-	ConfigKeyPayment   = "payment"
+	ConfigKeySystem     = "system"
+	ConfigKeyNotice     = "notice"
+	ConfigKeyAgreement  = "agreement"
+	ConfigKeyPrivacy    = "privacy"
+	ConfigKeyMarkMap    = "mark_map"
+	ConfigKeyCaptcha    = "captcha"
+	ConfigKeyWxLogin    = "wx_login"
+	ConfigKeyLicense    = "license"
+	ConfigKeySms        = "sms"
+	ConfigKeySmtp       = "smtp"
+	ConfigKeyOss        = "oss"
+	ConfigKeyPayment    = "payment"
+	ConfigKeyModeration = "moderation"
 )

api/core/types/moderation.go

@@ -0,0 +1,55 @@
+package types
+
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+// * Copyright 2023 The Geek-AI Authors. All rights reserved.
+// * Use of this source code is governed by a Apache-2.0 license
+// * that can be found in the LICENSE file.
+// * @Author yangjian102621@163.com
+// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
+
+// 文本审查
+type ModerationConfig struct {
+	Enable      bool                    `json:"enable"` // 是否启用文本审查
+	Active      string                  `json:"active"`
+	GuidePrompt string                  `json:"guide_prompt"` // 模型引导提示词
+	Gitee       ModerationGiteeConfig   `json:"gitee"`
+	Baidu       ModerationBaiduConfig   `json:"baidu"`
+	Tencent     ModerationTencentConfig `json:"tencent"`
+}
+
+const (
+	ModerationGitee   = "gitee"
+	ModerationBaidu   = "baidu"
+	ModerationTencent = "tencent"
+)
+
+// GiteeAI 文本审查配置
+type ModerationGiteeConfig struct {
+	ApiKey string `json:"api_key"`
+	Model  string `json:"model"` // 文本审核模型
+}
+
+// 百度文本审查配置
+type ModerationBaiduConfig struct {
+	AccessKey string `json:"access_key"`
+	SecretKey string `json:"secret_key"`
+}
+
+// 腾讯云文本审查配置
+type ModerationTencentConfig struct {
+	AccessKey string `json:"access_key"`
+	SecretKey string `json:"secret_key"`
+}
+
+type ModerationResult struct {
+	Flagged        bool               `json:"flagged"`
+	Categories     map[string]bool    `json:"categories"`
+	CategoryScores map[string]float64 `json:"category_scores"`
+}
+
+var ModerationCategories = map[string]string{
+	"politic":  "内容涉及人物、事件或敏感的政治观点",
+	"porn":     "明确的色情内容",
+	"insult":   "具有侮辱、攻击性语言、人身攻击或冒犯性表达",
+	"violence": "包含暴力、血腥、攻击行为或煽动暴力的言论",
+}