package services
import ( "bytes" "encoding/json" "fmt" "io" "net/http" "time" )
type AliyunAuditService struct { apiKey string appID string client *http.Client }
func NewAliyunAuditService(apiKey, appID string) *AliyunAuditService { return &AliyunAuditService{ apiKey: apiKey, appID: appID, client: &http.Client{Timeout: 30 * time.Second}, } }
func (s *AliyunAuditService) AuditComment(content string) (bool, error) { url := fmt.Sprintf("https://dashscope.aliyuncs.com/api/v1/apps/%s/completion", s.appID)
requestBody := map[string]interface{}{ "input": map[string]interface{}{ "prompt": content, "biz_params": map[string]interface{}{ "user_defined_params": map[string]interface{}{ "content": content, }, }, }, "parameters": map[string]interface{}{}, "debug": map[string]interface{}{}, }
jsonData, _ := json.Marshal(requestBody) req, _ := http.NewRequest("POST", url, bytes.NewBuffer(jsonData)) req.Header.Set("Authorization", "Bearer "+s.apiKey) req.Header.Set("Content-Type", "application/json")
resp, err := s.client.Do(req) if err != nil { return false, err } defer resp.Body.Close()
body, _ := io.ReadAll(resp.Body) var apiResp struct { Output struct { Text string `json:"text"` } `json:"output"` } if err := json.Unmarshal(body, &apiResp); err != nil { return false, err }
var auditResult struct { Status bool `json:"status"` } if err := json.Unmarshal([]byte(apiResp.Output.Text), &auditResult); err != nil { return false, err }
return auditResult.Status, nil }
|
评论
0 条评论