mirror of
https://github.com/kuaifan/dootask.git
synced 2025-12-11 18:42:54 +00:00
460 lines
16 KiB
PHP
460 lines
16 KiB
PHP
<?php
|
||
|
||
namespace App\Module;
|
||
|
||
use App\Models\Setting;
|
||
use Cache;
|
||
use Carbon\Carbon;
|
||
|
||
/**
|
||
* AI 助手模块
|
||
*/
|
||
class AI
|
||
{
|
||
protected $post = [];
|
||
protected $headers = [];
|
||
protected $urlPath = '';
|
||
protected $timeout = 30;
|
||
|
||
/**
|
||
* 构造函数
|
||
* @param array $post
|
||
* @param array $headers
|
||
*/
|
||
public function __construct($post = [], $headers = [])
|
||
{
|
||
$this->post = $post ?? [];
|
||
$this->headers = $headers ?? [];
|
||
}
|
||
|
||
/**
|
||
* 设置请求参数
|
||
* @param array $post
|
||
*/
|
||
public function setPost($post)
|
||
{
|
||
$this->post = array_merge($this->post, $post);
|
||
}
|
||
|
||
/**
|
||
* 设置请求头
|
||
* @param array $headers
|
||
*/
|
||
public function setHeaders($headers)
|
||
{
|
||
$this->headers = array_merge($this->headers, $headers);
|
||
}
|
||
|
||
/**
|
||
* 设置请求路径
|
||
* @param string $urlPath
|
||
*/
|
||
public function setUrlPath($urlPath)
|
||
{
|
||
$this->urlPath = $urlPath;
|
||
}
|
||
|
||
/**
|
||
* 设置请求超时时间
|
||
* @param int $timeout
|
||
*/
|
||
public function setTimeout($timeout)
|
||
{
|
||
$this->timeout = $timeout;
|
||
}
|
||
|
||
/**
|
||
* 请求 AI 接口
|
||
* @param bool $resRaw 是否返回原始数据
|
||
* @return array
|
||
*/
|
||
public function request($resRaw = false)
|
||
{
|
||
$aiSetting = Base::setting('aiSetting');
|
||
if (!Setting::AIOpen()) {
|
||
return Base::retError("AI 助手未开启");
|
||
}
|
||
|
||
$headers = [
|
||
'Content-Type' => 'application/json',
|
||
'Authorization' => 'Bearer ' . $aiSetting['ai_api_key'],
|
||
];
|
||
if ($aiSetting['ai_proxy']) {
|
||
$headers['CURLOPT_PROXY'] = $aiSetting['ai_proxy'];
|
||
$headers['CURLOPT_PROXYTYPE'] = str_contains($aiSetting['ai_proxy'], 'socks') ? CURLPROXY_SOCKS5 : CURLPROXY_HTTP;
|
||
}
|
||
$headers = array_merge($headers, $this->headers);
|
||
|
||
$url = $aiSetting['ai_api_url'] ?: 'https://api.openai.com/v1';
|
||
$url = $url . ($this->urlPath ?: '/chat/completions');
|
||
|
||
$result = Ihttp::ihttp_request($url, $this->post, $headers, $this->timeout);
|
||
if (Base::isError($result)) {
|
||
return Base::retError("AI 接口请求失败", $result);
|
||
}
|
||
$result = $result['data'];
|
||
|
||
if (!$resRaw) {
|
||
$resData = Base::json2array($result);
|
||
if (empty($resData['choices'])) {
|
||
return Base::retError("AI 接口返回数据格式错误", $resData);
|
||
}
|
||
$result = $resData['choices'][0]['message']['content'];
|
||
$result = trim($result);
|
||
if (empty($result)) {
|
||
return Base::retError("AI 接口返回数据为空");
|
||
}
|
||
}
|
||
|
||
return Base::retSuccess("success", $result);
|
||
}
|
||
|
||
/** ******************************************************************************************** */
|
||
/** ******************************************************************************************** */
|
||
/** ******************************************************************************************** */
|
||
|
||
/**
|
||
* 通过 openAI 语音转文字
|
||
* @param string $filePath 语音文件路径
|
||
* @param array $extParams 扩展参数
|
||
* @param bool $noCache 是否禁用缓存
|
||
* @return array
|
||
*/
|
||
public static function transcriptions($filePath, $extParams = [], $noCache = false)
|
||
{
|
||
if (!file_exists($filePath)) {
|
||
return Base::retError("语音文件不存在");
|
||
}
|
||
$systemSetting = Base::setting('system');
|
||
if ($systemSetting['voice2text'] !== 'open') {
|
||
return Base::retError("语音转文字功能未开启");
|
||
}
|
||
|
||
$cacheKey = "openAItranscriptions::" . md5($filePath . '_' . Base::array2json($extParams));
|
||
if ($noCache) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
|
||
$result = Cache::remember($cacheKey, Carbon::now()->addDays(), function () use ($extParams, $filePath) {
|
||
$post = array_merge($extParams, [
|
||
'file' => new \CURLFile($filePath),
|
||
'model' => 'whisper-1',
|
||
]);
|
||
$header = [
|
||
'Content-Type' => 'multipart/form-data',
|
||
];
|
||
|
||
$ai = new self($post, $header);
|
||
$ai->setUrlPath('/audio/transcriptions');
|
||
$ai->setTimeout(15);
|
||
|
||
$res = $ai->request(true);
|
||
if (Base::isError($res)) {
|
||
return Base::retError("语音转文字失败", $res);
|
||
}
|
||
$resData = Base::json2array($res['data']);
|
||
if (empty($resData['text'])) {
|
||
return Base::retError("语音转文字失败", $resData);
|
||
}
|
||
|
||
return Base::retSuccess("success", [
|
||
'file' => $filePath,
|
||
'text' => $resData['text'],
|
||
]);
|
||
});
|
||
if (Base::isError($result)) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 通过 openAI 翻译
|
||
* @param string $text 需要翻译的文本内容
|
||
* @param string $targetLanguage 目标语言（如：English, 简体中文, 日本語等）
|
||
* @param bool $noCache 是否禁用缓存
|
||
* @return array
|
||
*/
|
||
public static function translations($text, $targetLanguage, $noCache = false)
|
||
{
|
||
$systemSetting = Base::setting('system');
|
||
if ($systemSetting['translation'] !== 'open') {
|
||
return Base::retError("翻译功能未开启");
|
||
}
|
||
|
||
$cacheKey = "openAItranslations::" . md5($text . '_' . $targetLanguage);
|
||
if ($noCache) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
|
||
$result = Cache::remember($cacheKey, Carbon::now()->addDays(7), function () use ($text, $targetLanguage) {
|
||
$post = json_encode([
|
||
"model" => "gpt-5-mini",
|
||
"reasoning_effort" => "minimal",
|
||
"messages" => [
|
||
[
|
||
"role" => "system",
|
||
"content" => <<<EOF
|
||
你是一名资深的专业翻译专家，专门从事项目任务管理系统的多语言本地化工作。
|
||
|
||
翻译任务：将提供的文本内容翻译为 {$targetLanguage}
|
||
|
||
专业要求：
|
||
1. 术语一致性：确保项目管理、任务管理、团队协作等专业术语的准确翻译
|
||
2. 上下文理解：根据项目管理场景选择最合适的表达方式
|
||
3. 格式保持：严格保持原文的格式、结构、标点符号和排版
|
||
4. 语言规范：使用目标语言的标准表达，符合该语言的语法和习惯
|
||
5. 专业性：体现项目管理领域的专业水准和准确性
|
||
6. 简洁性：避免冗余表达，保持语言简洁明了
|
||
|
||
注意事项：
|
||
- 保留所有HTML标签、特殊符号、数字、日期格式
|
||
- 对于专有名词（如软件名称、品牌名）保持原文
|
||
- 确保翻译后的文本自然流畅，符合目标语言的表达习惯
|
||
- 如遇到歧义表达，优先选择项目管理场景下的含义
|
||
|
||
请直接返回翻译结果，不要包含任何解释或标记。
|
||
EOF
|
||
],
|
||
[
|
||
"role" => "user",
|
||
"content" => "请将以下内容翻译为 {$targetLanguage}：\n\n{$text}"
|
||
]
|
||
],
|
||
]);
|
||
|
||
$ai = new self($post);
|
||
$ai->setTimeout(60);
|
||
|
||
$res = $ai->request();
|
||
if (Base::isError($res)) {
|
||
return Base::retError("翻译请求失败", $res);
|
||
}
|
||
$result = $res['data'];
|
||
if (empty($result)) {
|
||
return Base::retError("翻译结果为空");
|
||
}
|
||
|
||
return Base::retSuccess("success", [
|
||
'translated_text' => $result,
|
||
'target_language' => $targetLanguage,
|
||
'translated_at' => date('Y-m-d H:i:s')
|
||
]);
|
||
});
|
||
|
||
if (Base::isError($result)) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 通过 openAI 生成标题
|
||
* @param string $text 需要生成标题的文本内容
|
||
* @param bool $noCache 是否禁用缓存
|
||
* @return array
|
||
*/
|
||
public static function generateTitle($text, $noCache = false)
|
||
{
|
||
$cacheKey = "openAIGenerateTitle::" . md5($text);
|
||
if ($noCache) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
|
||
$result = Cache::remember($cacheKey, Carbon::now()->addHours(24), function () use ($text) {
|
||
$post = json_encode([
|
||
"model" => "gpt-5-mini",
|
||
"reasoning_effort" => "minimal",
|
||
"messages" => [
|
||
[
|
||
"role" => "system",
|
||
"content" => <<<EOF
|
||
你是一个专业的标题生成器，专门为项目任务管理系统的对话内容生成精准、简洁的标题。
|
||
|
||
要求：
|
||
1. 标题要准确概括文本的核心内容和主要意图
|
||
2. 标题长度控制在5-20个字符之间
|
||
3. 语言简洁明了，避免冗余词汇
|
||
4. 适合在项目管理场景中使用
|
||
5. 不要包含引号或特殊符号
|
||
6. 如果是技术讨论，突出技术要点
|
||
7. 如果是项目管理内容，突出关键动作或目标
|
||
8. 如果是需求讨论，突出需求的核心点
|
||
|
||
请直接返回标题，不要包含任何解释或其他内容。
|
||
EOF
|
||
],
|
||
[
|
||
"role" => "user",
|
||
"content" => "请为以下内容生成一个合适的标题：\n\n" . $text
|
||
]
|
||
],
|
||
]);
|
||
|
||
$ai = new self($post);
|
||
$ai->setTimeout(10);
|
||
|
||
$res = $ai->request();
|
||
if (Base::isError($res)) {
|
||
return Base::retError("标题生成失败", $res);
|
||
}
|
||
$result = $res['data'];
|
||
if (empty($result)) {
|
||
return Base::retError("生成的标题为空");
|
||
}
|
||
|
||
return Base::retSuccess("success", [
|
||
'title' => $result,
|
||
'length' => mb_strlen($result),
|
||
'generated_at' => date('Y-m-d H:i:s')
|
||
]);
|
||
});
|
||
|
||
if (Base::isError($result)) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 通过 openAI 生成职场笑话、心灵鸡汤
|
||
* @param bool $noCache 是否禁用缓存
|
||
* @return array 返回20个笑话和20个心灵鸡汤
|
||
*/
|
||
public static function generateJokeAndSoup($noCache = false)
|
||
{
|
||
$cacheKey = "openAIJokeAndSoup::" . md5(date('Y-m-d'));
|
||
if ($noCache) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
|
||
$result = Cache::remember($cacheKey, Carbon::now()->addHours(6), function () {
|
||
$post = json_encode([
|
||
"model" => "gpt-5-mini",
|
||
"reasoning_effort" => "minimal",
|
||
"messages" => [
|
||
[
|
||
"role" => "system",
|
||
"content" => <<<EOF
|
||
你是一个专业的内容生成器。
|
||
|
||
要求：
|
||
1. 笑话要幽默风趣，适合职场环境，内容积极正面
|
||
2. 心灵鸡汤要励志温暖，适合职场人士阅读
|
||
3. 每个笑话和鸡汤都要简洁明了，尽量不超过100字
|
||
4. 必须严格按照以下JSON格式返回，不要markdown格式，不要包含任何其他内容：
|
||
|
||
{
|
||
"jokes": [
|
||
"笑话内容1",
|
||
"笑话内容2",
|
||
...
|
||
],
|
||
"soups": [
|
||
"心灵鸡汤内容1",
|
||
"心灵鸡汤内容2",
|
||
...
|
||
]
|
||
}
|
||
EOF
|
||
],
|
||
[
|
||
"role" => "user",
|
||
"content" => "请生成20个职场笑话和20个心灵鸡汤"
|
||
]
|
||
],
|
||
]);
|
||
|
||
$ai = new self($post);
|
||
$ai->setTimeout(120);
|
||
|
||
$res = $ai->request();
|
||
if (Base::isError($res)) {
|
||
return Base::retError("生成失败", $res);
|
||
}
|
||
|
||
// 清理可能的markdown代码块标记
|
||
$content = $res['data'];
|
||
$content = preg_replace('/^\s*```json\s*/', '', $content);
|
||
$content = preg_replace('/\s*```\s*$/', '', $content);
|
||
if (empty($content)) {
|
||
return Base::retError("翻译结果为空");
|
||
}
|
||
|
||
// 解析JSON
|
||
$parsedData = Base::json2array($content);
|
||
if (!$parsedData || !isset($parsedData['jokes']) || !isset($parsedData['soups'])) {
|
||
return Base::retError("生成内容格式错误", $content);
|
||
}
|
||
|
||
// 验证数据完整性
|
||
if (!is_array($parsedData['jokes']) || !is_array($parsedData['soups'])) {
|
||
return Base::retError("生成内容格式错误", $parsedData);
|
||
}
|
||
|
||
// 过滤空内容并确保有内容
|
||
$jokes = array_filter(array_map('trim', $parsedData['jokes']));
|
||
$soups = array_filter(array_map('trim', $parsedData['soups']));
|
||
|
||
if (empty($jokes) || empty($soups)) {
|
||
return Base::retError("生成内容为空", $parsedData);
|
||
}
|
||
|
||
return Base::retSuccess("success", [
|
||
'jokes' => array_values($jokes),
|
||
'soups' => array_values($soups),
|
||
'total_jokes' => count($jokes),
|
||
'total_soups' => count($soups),
|
||
'generated_at' => date('Y-m-d H:i:s')
|
||
]);
|
||
});
|
||
|
||
if (Base::isError($result)) {
|
||
Cache::forget($cacheKey);
|
||
}
|
||
return $result;
|
||
}
|
||
|
||
/**
|
||
* 获取 ollama 模型
|
||
* @param $baseUrl
|
||
* @param $key
|
||
* @param $agency
|
||
* @return array
|
||
*/
|
||
public static function ollamaModels($baseUrl, $key = null, $agency = null)
|
||
{
|
||
$extra = [
|
||
'Content-Type' => 'application/json',
|
||
];
|
||
if ($key) {
|
||
$extra['Authorization'] = 'Bearer ' . $key;
|
||
}
|
||
if ($agency) {
|
||
$extra['CURLOPT_PROXY'] = $agency;
|
||
$extra['CURLOPT_PROXYTYPE'] = str_contains($agency, 'socks') ? CURLPROXY_SOCKS5 : CURLPROXY_HTTP;
|
||
}
|
||
$res = Ihttp::ihttp_request(rtrim($baseUrl, '/') . '/api/tags', [], $extra, 15);
|
||
if (Base::isError($res)) {
|
||
return Base::retError("获取失败", $res);
|
||
}
|
||
$resData = Base::json2array($res['data']);
|
||
if (empty($resData['models'])) {
|
||
return Base::retError("获取失败", $resData);
|
||
}
|
||
$models = [];
|
||
foreach ($resData['models'] as $model) {
|
||
if ($model['name'] !== $model['model']) {
|
||
$models[] = "{$model['model']} | {$model['name']}";
|
||
} else {
|
||
$models[] = $model['model'];
|
||
}
|
||
}
|
||
return Base::retSuccess("success", [
|
||
'models' => $models,
|
||
'original' => $resData['models']
|
||
]);
|
||
}
|
||
}
|