414 lines
13 KiB
PHP
414 lines
13 KiB
PHP
<?php
|
||
|
||
namespace app\api\controller;
|
||
|
||
use think\Db;
|
||
use think\Cache;
|
||
use think\Env;
|
||
|
||
class Agent extends Base
|
||
{
|
||
|
||
public function __construct(\think\Request $request = null)
|
||
{
|
||
parent::__construct($request);
|
||
}
|
||
|
||
/**
|
||
* 获取所有叶子节点 major 及其完整路径，用于 AI 匹配
|
||
*/
|
||
private function getMajorTree()
|
||
{
|
||
$cacheKey = 'agent_major_tree';
|
||
$cached = Cache::get($cacheKey);
|
||
if ($cached) {
|
||
return $cached;
|
||
}
|
||
|
||
$allMajors = $this->major_obj
|
||
->where('major_state', 0)
|
||
->where('major_type', 0)
|
||
->select();
|
||
|
||
$majorMap = [];
|
||
foreach ($allMajors as $m) {
|
||
$majorMap[$m['major_id']] = $m;
|
||
}
|
||
|
||
$result = [];
|
||
foreach ($allMajors as $m) {
|
||
$hasChild = false;
|
||
foreach ($allMajors as $check) {
|
||
if ($check['pid'] == $m['major_id']) {
|
||
$hasChild = true;
|
||
break;
|
||
}
|
||
}
|
||
if (!$hasChild) {
|
||
$path = $this->buildMajorPath($m['major_id'], $majorMap);
|
||
$result[] = [
|
||
'major_id' => $m['major_id'],
|
||
'major_title' => $m['major_title'],
|
||
'full_path' => $path,
|
||
];
|
||
}
|
||
}
|
||
|
||
Cache::set($cacheKey, $result, 3600);
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 递归构建 major 的完整路径
|
||
*/
|
||
private function buildMajorPath($majorId, &$majorMap)
|
||
{
|
||
if (!isset($majorMap[$majorId])) {
|
||
return '';
|
||
}
|
||
$m = $majorMap[$majorId];
|
||
if ($m['pid'] == 0 || $m['pid'] == 1 || !isset($majorMap[$m['pid']])) {
|
||
return $m['major_title'];
|
||
}
|
||
return $this->buildMajorPath($m['pid'], $majorMap) . ' > ' . $m['major_title'];
|
||
}
|
||
|
||
/**
|
||
* 构建 major 列表提示文本（供 AI 使用）
|
||
*/
|
||
private function buildMajorListPrompt($majorTree)
|
||
{
|
||
$lines = [];
|
||
foreach ($majorTree as $item) {
|
||
$lines[] = "ID:{$item['major_id']} - {$item['full_path']}";
|
||
}
|
||
return implode("\n", $lines);
|
||
}
|
||
|
||
/**
|
||
* 调用 AI 将用户 field 描述匹配到标准 major_id
|
||
*/
|
||
private function matchFieldToMajor($field, $majorListPrompt)
|
||
{
|
||
$systemPrompt = "你是一位医学领域分类专家。用户会提供一段研究领域的描述文本，你需要从给定的标准领域列表中找出最匹配的1-3个领域。\n"
|
||
. "请严格按照JSON数组格式返回匹配结果，只返回major_id数组，如 [12,34,56]。\n"
|
||
. "如果没有合适的匹配，返回空数组 []。\n"
|
||
. "不要返回任何其他内容，只返回JSON数组。\n\n"
|
||
. "标准领域列表：\n" . $majorListPrompt;
|
||
|
||
$userPrompt = "请为以下研究领域描述匹配最合适的标准领域ID：\n" . $field;
|
||
|
||
$messages = [
|
||
['role' => 'system', 'content' => $systemPrompt],
|
||
['role' => 'user', 'content' => $userPrompt],
|
||
];
|
||
|
||
$apiKey = Env::get("gpt.api_key1", Env::get("gpt.api_key", ""));
|
||
$url = 'http://chat.taimed.cn/v1/chat/completions';
|
||
|
||
$data = [
|
||
'model' => 'gpt-4.1',
|
||
'messages' => $messages,
|
||
'temperature' => 0.1,
|
||
'max_tokens' => 200,
|
||
];
|
||
|
||
$ch = curl_init();
|
||
curl_setopt($ch, CURLOPT_URL, $url);
|
||
curl_setopt($ch, CURLOPT_HTTPHEADER, [
|
||
'Content-Type: application/json',
|
||
'Authorization: Bearer ' . $apiKey,
|
||
]);
|
||
curl_setopt($ch, CURLOPT_POST, true);
|
||
curl_setopt($ch, CURLOPT_POSTFIELDS, json_encode($data));
|
||
curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
|
||
curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
|
||
curl_setopt($ch, CURLOPT_TIMEOUT, 60);
|
||
|
||
$result = curl_exec($ch);
|
||
if (curl_errno($ch)) {
|
||
curl_close($ch);
|
||
return [];
|
||
}
|
||
curl_close($ch);
|
||
|
||
$res = json_decode($result, true);
|
||
if (!isset($res['choices'][0]['message']['content'])) {
|
||
return [];
|
||
}
|
||
|
||
$content = trim($res['choices'][0]['message']['content']);
|
||
// 提取 JSON 数组
|
||
if (preg_match('/\[[\d,\s]*\]/', $content, $matches)) {
|
||
$ids = json_decode($matches[0], true);
|
||
if (is_array($ids)) {
|
||
return array_map('intval', $ids);
|
||
}
|
||
}
|
||
|
||
return [];
|
||
}
|
||
|
||
/**
|
||
* 将匹配结果写入 t_major_to_user
|
||
*/
|
||
private function saveMajorToUser($userId, $majorIds)
|
||
{
|
||
$existing = $this->major_to_user_obj
|
||
->where('user_id', $userId)
|
||
->where('state', 0)
|
||
->column('major_id');
|
||
|
||
$toInsert = array_diff($majorIds, $existing);
|
||
foreach ($toInsert as $majorId) {
|
||
$this->major_to_user_obj->insert([
|
||
'user_id' => $userId,
|
||
'major_id' => $majorId,
|
||
'ctime' => time(),
|
||
]);
|
||
}
|
||
|
||
return count($toInsert);
|
||
}
|
||
|
||
/**
|
||
* 处理单个用户的 field 转 major
|
||
*
|
||
* @param int user_id 用户ID
|
||
*/
|
||
public function processOneUser()
|
||
{
|
||
$data = $this->request->param();
|
||
if (!isset($data['user_id']) || $data['user_id'] == '') {
|
||
return jsonError('user_id不能为空');
|
||
}
|
||
|
||
$userId = intval($data['user_id']);
|
||
$reviewerInfo = $this->user_reviewer_info_obj
|
||
->where('reviewer_id', $userId)
|
||
->where('state', 0)
|
||
->find();
|
||
|
||
if (!$reviewerInfo) {
|
||
return jsonError('未找到该用户的reviewer信息');
|
||
}
|
||
|
||
$field = trim($reviewerInfo['field']);
|
||
if ($field == '') {
|
||
return jsonError('该用户的field字段为空');
|
||
}
|
||
|
||
$majorTree = $this->getMajorTree();
|
||
if (empty($majorTree)) {
|
||
return jsonError('未获取到标准领域数据');
|
||
}
|
||
|
||
$majorListPrompt = $this->buildMajorListPrompt($majorTree);
|
||
|
||
|
||
|
||
$matchedIds = $this->matchFieldToMajor($field, $majorListPrompt);
|
||
|
||
if (empty($matchedIds)) {
|
||
return jsonSuccess([
|
||
'user_id' => $userId,
|
||
'field' => $field,
|
||
'matched_ids' => [],
|
||
'inserted' => 0,
|
||
'msg' => 'AI未匹配到合适的领域',
|
||
]);
|
||
}
|
||
|
||
// 验证 major_id 确实存在
|
||
$validMajors = $this->major_obj
|
||
->where('major_id', 'in', $matchedIds)
|
||
->where('major_state', 0)
|
||
->select();
|
||
// ->column('major_id');
|
||
// $matchedIds = array_intersect($matchedIds, $validMajors);
|
||
|
||
// $inserted = $this->saveMajorToUser($userId, $matchedIds);
|
||
|
||
foreach ($validMajors as $k => $major){
|
||
$validMajors[$k]['shu'] = getMajorShu($major['major_id']);
|
||
}
|
||
|
||
|
||
|
||
return jsonSuccess([
|
||
'user_id' => $userId,
|
||
'field' => $field,
|
||
'majors' => $validMajors,
|
||
// 'inserted' => $inserted,
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 批量处理：获取有 field 但没有 major_to_user 记录的用户，逐个用 AI 匹配
|
||
*
|
||
* @param int limit 每次处理的数量，默认10
|
||
* @param int skip_has_major 是否跳过已有major_to_user记录的用户，默认1
|
||
*/
|
||
public function batchProcess()
|
||
{
|
||
$data = $this->request->param();
|
||
$limit = isset($data['limit']) ? intval($data['limit']) : 10;
|
||
$skipHasMajor = isset($data['skip_has_major']) ? intval($data['skip_has_major']) : 1;
|
||
|
||
if ($limit > 50) {
|
||
$limit = 50;
|
||
}
|
||
|
||
$query = $this->user_reviewer_info_obj
|
||
->alias('ri')
|
||
->field('ri.reviewer_id, ri.field')
|
||
->where('ri.state', 0)
|
||
->where('ri.field', '<>', '');
|
||
|
||
if ($skipHasMajor) {
|
||
$subQuery = Db::name('major_to_user')->where('state', 0)->field('user_id')->buildSql();
|
||
$query = $query->where('ri.reviewer_id', 'not in', $subQuery);
|
||
}
|
||
|
||
$users = $query->limit($limit)->select();
|
||
|
||
if (empty($users)) {
|
||
return jsonSuccess([
|
||
'processed' => 0,
|
||
'msg' => '没有需要处理的用户',
|
||
]);
|
||
}
|
||
|
||
$majorTree = $this->getMajorTree();
|
||
if (empty($majorTree)) {
|
||
return jsonError('未获取到标准领域数据');
|
||
}
|
||
$majorListPrompt = $this->buildMajorListPrompt($majorTree);
|
||
|
||
$validMajorIds = $this->major_obj->where('major_state', 0)->column('major_id');
|
||
|
||
$results = [];
|
||
$successCount = 0;
|
||
$failCount = 0;
|
||
|
||
foreach ($users as $user) {
|
||
$field = trim($user['field']);
|
||
if ($field == '') {
|
||
continue;
|
||
}
|
||
|
||
$matchedIds = $this->matchFieldToMajor($field, $majorListPrompt);
|
||
$matchedIds = array_intersect($matchedIds, $validMajorIds);
|
||
|
||
if (!empty($matchedIds)) {
|
||
$inserted = $this->saveMajorToUser($user['reviewer_id'], $matchedIds);
|
||
$results[] = [
|
||
'user_id' => $user['reviewer_id'],
|
||
'field' => mb_substr($field, 0, 100),
|
||
'matched_ids' => array_values($matchedIds),
|
||
'inserted' => $inserted,
|
||
];
|
||
$successCount++;
|
||
} else {
|
||
$results[] = [
|
||
'user_id' => $user['reviewer_id'],
|
||
'field' => mb_substr($field, 0, 100),
|
||
'matched_ids' => [],
|
||
'inserted' => 0,
|
||
];
|
||
$failCount++;
|
||
}
|
||
}
|
||
|
||
return jsonSuccess([
|
||
'processed' => count($results),
|
||
'success_count' => $successCount,
|
||
'fail_count' => $failCount,
|
||
'details' => $results,
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 查看当前 major 树结构（调试用）
|
||
*/
|
||
public function getMajorList()
|
||
{
|
||
$majorTree = $this->getMajorTree();
|
||
return jsonSuccess([
|
||
'total' => count($majorTree),
|
||
'list' => $majorTree,
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 从 Excel 文件导入 major 数据到数据库（如需要）
|
||
*/
|
||
public function importMajorFromExcel()
|
||
{
|
||
$file = ROOT_PATH . 'public' . DS . 'system' . DS . 't_major.xlsx';
|
||
if (!file_exists($file)) {
|
||
return jsonError('Excel文件不存在: public/system/t_major.xlsx');
|
||
}
|
||
|
||
$spreadsheet = \PhpOffice\PhpSpreadsheet\IOFactory::load($file);
|
||
$sheet = $spreadsheet->getActiveSheet();
|
||
$highestRow = $sheet->getHighestRow();
|
||
$highestColumn = $sheet->getHighestColumn();
|
||
|
||
$headers = [];
|
||
$colCount = \PhpOffice\PhpSpreadsheet\Cell\Coordinate::columnIndexFromString($highestColumn);
|
||
for ($col = 1; $col <= $colCount; $col++) {
|
||
$headers[$col] = $sheet->getCellByColumnAndRow($col, 1)->getValue();
|
||
}
|
||
|
||
$rows = [];
|
||
for ($row = 2; $row <= $highestRow; $row++) {
|
||
$rowData = [];
|
||
for ($col = 1; $col <= $colCount; $col++) {
|
||
$rowData[$headers[$col]] = $sheet->getCellByColumnAndRow($col, $row)->getValue();
|
||
}
|
||
$rows[] = $rowData;
|
||
}
|
||
|
||
return jsonSuccess([
|
||
'headers' => array_values($headers),
|
||
'total' => count($rows),
|
||
'preview' => array_slice($rows, 0, 20),
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 统计当前 field 转 major 的覆盖情况
|
||
*/
|
||
public function statistics()
|
||
{
|
||
$totalReviewers = $this->user_reviewer_info_obj
|
||
->where('state', 0)
|
||
->count();
|
||
|
||
$hasField = $this->user_reviewer_info_obj
|
||
->where('state', 0)
|
||
->where('field', '<>', '')
|
||
->count();
|
||
|
||
$hasMajorToUser = Db::name('major_to_user')
|
||
->where('state', 0)
|
||
->group('user_id')
|
||
->count();
|
||
|
||
$hasFieldNoMajor = $this->user_reviewer_info_obj
|
||
->alias('ri')
|
||
->where('ri.state', 0)
|
||
->where('ri.field', '<>', '')
|
||
->where('ri.reviewer_id', 'not in', Db::name('major_to_user')->where('state', 0)->field('user_id')->buildSql())
|
||
->count();
|
||
|
||
return jsonSuccess([
|
||
'total_reviewers' => $totalReviewers,
|
||
'has_field' => $hasField,
|
||
'has_major_to_user' => $hasMajorToUser,
|
||
'has_field_no_major' => $hasFieldNoMajor,
|
||
]);
|
||
}
|
||
}
|