You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
3.0 KiB
PHP
93 lines
3.0 KiB
PHP
<?php
|
|
namespace Isekai\AIReview;
|
|
|
|
use AlibabaCloud\Client\AlibabaCloud;
|
|
use AlibabaCloud\Green\Green;
|
|
|
|
class AliyunAIReview {
|
|
private const MAX_LENGTH = 10000;
|
|
|
|
public function __construct(){
|
|
global $wgAIReviewEndpoint, $wgAIReviewAccessKeyId, $wgAIReviewAccessKeySecret;
|
|
AlibabaCloud::accessKeyClient($wgAIReviewAccessKeyId, $wgAIReviewAccessKeySecret)
|
|
->connectTimeout(3)
|
|
->regionId($wgAIReviewEndpoint)
|
|
->asDefaultClient();
|
|
}
|
|
|
|
public function reviewText($text, $rawResponse = false) {
|
|
$reqData = $this->buildRequestData($text);
|
|
$response = $this->doRequest($reqData, $rawResponse);
|
|
return $response;
|
|
}
|
|
|
|
public function buildRequestData($text){
|
|
global $wgAIReviewBizType;
|
|
|
|
$reqData = [
|
|
'scenes' => ['antispam'],
|
|
'tasks' => $this->buildTasks($text),
|
|
];
|
|
|
|
if($wgAIReviewBizType) $reqData['bizType'] = $wgAIReviewBizType;
|
|
return $reqData;
|
|
}
|
|
|
|
public function buildTasks($text){
|
|
$splitter = new SectionSplitter($text, self::MAX_LENGTH);
|
|
$chunkList = $splitter->getChunkList();
|
|
$taskList = [];
|
|
foreach($chunkList as $chunk){
|
|
$task = [
|
|
'dataId' => uniqid(),
|
|
'content' => $chunk,
|
|
];
|
|
$taskList[] = $task;
|
|
};
|
|
unset($chunkList);
|
|
return $taskList;
|
|
}
|
|
|
|
/**
|
|
* @throws \AlibabaCloud\Client\Exception\ServerException
|
|
* @throws \AlibabaCloud\Client\Exception\ClientException
|
|
*/
|
|
public function doRequest($requestData, $rawResponse = false) {
|
|
$textScan = Green::v20180509()->textScan();
|
|
$response = $textScan->body(json_encode($requestData))->request();
|
|
|
|
if ($rawResponse) {
|
|
return $response;
|
|
} else {
|
|
if ($response->getReasonPhrase() === 'OK') {
|
|
return $this->parseResponse($response->toArray());
|
|
} else {
|
|
return ['pass' => false, 'reason' => wfMessage('isekai-aireview-aliyun-server-error', $response->getStatusCode())->escaped()];
|
|
}
|
|
}
|
|
}
|
|
|
|
public function parseResponse($response){
|
|
if($response['code'] !== 200)
|
|
return ['pass' => false, 'reason' => wfMessage('isekai-aireview-aliyun-server-error', $response['code'])->escaped()];
|
|
|
|
$pass = true;
|
|
$reasons = [];
|
|
foreach($response['data'] as $task){
|
|
if(is_array($task['results'])){
|
|
foreach($task['results'] as $result){
|
|
if($result['suggestion'] !== 'pass'){
|
|
$pass = false;
|
|
foreach($result['details'] as $detail){
|
|
$reason = $detail['label'];
|
|
if(!in_array($reason, $reasons)){
|
|
$reasons[] = $reason;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return ['pass' => $pass, 'reason' => $reasons];
|
|
}
|
|
} |