lzt/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: violation fee check

Browse Source
This commit is contained in:
Seefs
2026-02-23 22:02:59 +08:00
parent 194b53f061
commit b15ad2924e
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
service/violation_fee.go
+5 -4
View File
@@ -18,8 +18,9 @@ import (
) )
const ( const (
ViolationFeeCodePrefix = "violation_fee." ViolationFeeCodePrefix = "violation_fee."
CSAMViolationMarker = "Failed check: SAFETY_CHECK_TYPE_CSAM" CSAMViolationMarker = "Failed check: SAFETY_CHECK_TYPE"
ContentViolatesUsageMarker = "Content violates usage guidelines"
) )
func IsViolationFeeCode(code types.ErrorCode) bool { func IsViolationFeeCode(code types.ErrorCode) bool {
@@ -30,11 +31,11 @@ func HasCSAMViolationMarker(err *types.NewAPIError) bool {
if err == nil { if err == nil {
return false return false
} }
if strings.Contains(err.Error(), CSAMViolationMarker) { if strings.Contains(err.Error(), CSAMViolationMarker) || strings.Contains(err.Error(), ContentViolatesUsageMarker) {
return true return true
} }
msg := err.ToOpenAIError().Message msg := err.ToOpenAIError().Message
return strings.Contains(msg, CSAMViolationMarker) return strings.Contains(msg, CSAMViolationMarker) || strings.Contains(err.Error(), ContentViolatesUsageMarker)
} }
func WrapAsViolationFeeGrokCSAM(err *types.NewAPIError) *types.NewAPIError { func WrapAsViolationFeeGrokCSAM(err *types.NewAPIError) *types.NewAPIError {