perf: 优化 Ollama AI

This commit is contained in:
kuaifan
2025-02-22 01:29:37 +08:00
parent b3e83e13bc
commit a8f3b02ee7
5 changed files with 57 additions and 2 deletions

View File

@@ -159,6 +159,35 @@ class Extranet
return Base::retSuccess("success", $result);
}
/**
* 获取 ollama 模型
* @param $baseUrl
* @return array
*/
public static function ollamaModels($baseUrl)
{
$res = Ihttp::ihttp_request($baseUrl . '/api/tags', [], [], 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']
]);
}
/**
* 获取IP地址经纬度
* @param string $ip