Golang 개발 팁: Baidu AI 인터페이스를 사용하여 텍스트 검토 구현

go get github.com/levigross/grequests

import (
    "github.com/levigross/grequests"
)

func TextCensor(text string) (bool, error) {
    url := "https://aip.baidubce.com/rest/2.0/solution/v1/text_censor/user_defined?access_token=<YOUR_ACCESS_TOKEN>"
    payload := map[string]interface{}{
        "tasks": []map[string]string{
            {
                "content": text,
            },
        },
    }
    headers := map[string]string{
        "Content-Type": "application/json;charset=UTF-8",
    }
    resp, err := grequests.Post(url, &grequests.RequestOptions{
        Headers: headers,
        JSON:    payload,
    })

    if err != nil {
        return false, err
    }

    if !resp.Ok {
        return false, fmt.Errorf("Request failed with status code: %d", resp.StatusCode)
    }

    type Response struct {
        Result []struct {
            Conclusion string `json:"conclusion"`
        } `json:"result"`
    }
    var data Response
    err = resp.JSON(&data)

    if err != nil {
        return false, err
    }

    if len(data.Result) == 0 {
        return false, errors.New("Empty response")
    }

    if data.Result[0].Conclusion != "合规" {
        return false, nil
    }

    return true, nil
}

func main() {
    text := "这是一段违规内容"
    result, err := TextCensor(text)
    if err != nil {
        log.Fatal(err)
    }
    if result {
        fmt.Println("文本合规")
    } else {
        fmt.Println("文本违规")
    }
}