feat: ALL_DIALOG_RECORD_ENABLE

README.md

@@ -128,6 +128,7 @@ Render 可以直接部署 docker 镜像,不需要 fork 仓库：[Render](https:/
 |        变量参数        |                        变量描述                         | 是否必填 | 
 |:------------------:|:---------------------------------------------------:|:----:|
 | AUDIT_CHANNEL_TYPE |   审核渠道类型[openai:openai、ali:阿里、baidu:百度、qiniu:七牛]    |  Y   |  
+| ALL_DIALOG_RECORD_ENABLE |             全量上下文审核开关[0:关闭、1:打开]（默认:0）              |  N   |  
 |      BASE_URL      | 审核通过后的转发接口请求地址域名或IP:端口（例如: https://api.openai.com ） |  Y   |
 |      API_KEY       |           鉴权密钥，与转发接口的API-Key保持一致，多个以`,`分隔           |  Y   |
 |       ENABLE       |               审核启用开关[0:关闭、1:打开]（默认:1）               |  N   |

common/config/config.go

@@ -85,6 +85,7 @@ var QiNiuAuditContentLength = env.Int("QINIU_AUDIT_CONTENT_LENGTH", 4000)
 var OpenaiModerationBaseUrl = os.Getenv("OPENAI_MODERATION_BASE_URL")
 var OpenaiModerationApiKey = os.Getenv("OPENAI_MODERATION_API_KEY")
 var OpenaiModerationLabel = os.Getenv("OPENAI_MODERATION_LABEL")
+var AllDialogRecordEnable = env.Int("ALL_DIALOG_RECORD_ENABLE", 0)
 var OpenaiModerationAuditContentLength = env.Int("OPENAI_MODERATION_AUDIT_CONTENT_LENGTH", 4000)
 
 var DebugEnabled = strings.ToLower(os.Getenv("DEBUG")) == "true"

common/constants.go

@@ -3,4 +3,4 @@ package common
 import "time"
 
 var StartTime = time.Now().Unix() // unit: second
-var Version = "v1.0.1"            // this hard coding will be replaced automatically when building, no need to manually change
+var Version = "v1.0.2"            // this hard coding will be replaced automatically when building, no need to manually change

model/openai.go

@@ -1,5 +1,7 @@
 package model
 
+import "gpt-content-audit/common/config"
+
 type OpenAIChatCompletionRequest struct {
 	Model    string              `json:"model"`
 	Stream   bool                `json:"stream"`
@@ -95,12 +97,27 @@ type GetUserContent interface {
 
 func (r OpenAIChatCompletionRequest) GetUserContent() []string {
 	var userContent []string
-	for _, msg := range r.Messages {
-		switch contentObj := msg.Content.(type) {
-		case string:
-			userContent = append(userContent, contentObj)
+	if config.AllDialogRecordEnable == 1 {
+		for _, msg := range r.Messages {
+			if msg.Role == "user" {
+				switch contentObj := msg.Content.(type) {
+				case string:
+					userContent = append(userContent, contentObj)
+				}
+			}
+		}
+	} else {
+		for i := len(r.Messages) - 1; i >= 0; i-- {
+			if r.Messages[i].Role == "user" {
+				switch contentObj := r.Messages[i].Content.(type) {
+				case string:
+					userContent = append(userContent, contentObj)
+				}
+				break
+			}
 		}
 	}
+
 	return userContent
 }