diff --git a/internal/api/text_scanner.go b/internal/api/text_scanner.go index 586bcc9..ec01a21 100644 --- a/internal/api/text_scanner.go +++ b/internal/api/text_scanner.go @@ -78,6 +78,10 @@ func (s *TextScanner) ScanText(content string, dataID string, serviceType TextSe } // 转换响应格式 + if response.Body.Data == nil { + return nil, fmt.Errorf("请求成功,但响应体为空: %v", response) + } + scanResponse := &model.TextScanResponse{ Code: tea.Int32Value(response.Body.Code), Message: tea.StringValue(response.Body.Message), @@ -91,6 +95,7 @@ func (s *TextScanner) ScanText(content string, dataID string, serviceType TextSe Message: tea.StringValue(response.Body.Message), DetectedLanguage: tea.StringValue(response.Body.Data.DetectedLanguage), TranslatedContent: tea.StringValue(response.Body.Data.TranslatedContent), + RiskLevel: tea.StringValue(response.Body.Data.RiskLevel), Results: make([]*green20220302.TextModerationPlusResponseBodyDataResult, 0, len(response.Body.Data.Result)), } @@ -152,6 +157,7 @@ func (s *TextScanner) ScanTextBatch(texts []string, dataIDs []string, serviceTyp Message: tea.StringValue(response.Body.Message), DetectedLanguage: tea.StringValue(response.Body.Data.DetectedLanguage), TranslatedContent: tea.StringValue(response.Body.Data.TranslatedContent), + RiskLevel: tea.StringValue(response.Body.Data.RiskLevel), Results: make([]*green20220302.TextModerationPlusResponseBodyDataResult, 0, len(response.Body.Data.Result)), } @@ -177,6 +183,7 @@ func (s *TextScanner) PrintResult(response *model.TextScanResponse) { fmt.Printf("处理状态: %d - %s\n", data.Code, data.Message) fmt.Printf("检测语言: %s\n", data.DetectedLanguage) fmt.Printf("翻译内容: %s\n", data.TranslatedContent) + fmt.Printf("风险等级: %s\n", data.RiskLevel) for _, result := range data.Results { if result.Confidence != nil { diff --git a/internal/model/text.go b/internal/model/text.go index 8842b8b..ef2a7d0 100644 --- a/internal/model/text.go +++ b/internal/model/text.go @@ -30,5 +30,6 @@ type TextScanData struct { DataID string `json:"dataId"` DetectedLanguage string `json:"detectedLanguage"` TranslatedContent string `json:"translatedContent"` + RiskLevel string `json:"riskLevel"` Results []*green20220302.TextModerationPlusResponseBodyDataResult `json:"results"` }