483 lines
17 KiB
PHP
483 lines
17 KiB
PHP
<?php
|
||
namespace app\controller\superadmin;
|
||
|
||
use app\BaseController;
|
||
use app\model\AiProvider as AiProviderModel;
|
||
use think\facade\Request;
|
||
use think\facade\Db;
|
||
|
||
/**
|
||
* AI服务商配置管理控制器（超管专用）
|
||
*/
|
||
class AiConfig extends BaseController
|
||
{
|
||
/**
|
||
* 获取所有AI服务商配置
|
||
* @return \think\response\Json
|
||
*/
|
||
public function index()
|
||
{
|
||
// 验证是否为超级管理员
|
||
$user = $this->request->user ?? null;
|
||
if (!$user || $user['role'] !== 'superadmin') {
|
||
return error('无权限访问', 403);
|
||
}
|
||
|
||
// 列表只返回“显示”的配置（visible=1 或未设）；隐藏的由数据库 visible=0 控制，不在此列表展示
|
||
$providers = AiProviderModel::order('id', 'asc')
|
||
->whereRaw('(visible IS NULL OR visible = 1)')
|
||
->select()
|
||
->toArray();
|
||
|
||
// 处理返回数据
|
||
$result = [];
|
||
foreach ($providers as $provider) {
|
||
$result[] = [
|
||
'id' => $provider['providerId'],
|
||
'name' => $provider['name'],
|
||
'enabled' => $provider['enabled'] == 1,
|
||
'visible' => isset($provider['visible']) ? ($provider['visible'] == 1) : true,
|
||
'apiKey' => $provider['apiKey'] ?? '', // 脱敏后的密钥
|
||
'apiEndpoint' => $provider['apiEndpoint'] ?? '',
|
||
'model' => $provider['model'] ?? '',
|
||
'organizationId' => $provider['organizationId'] ?? '',
|
||
'maxTokens' => $provider['maxTokens'] ?? 4096,
|
||
'balanceAlertEnabled' => $provider['balanceAlertEnabled'] == 1,
|
||
'balanceAlertThreshold' => floatval($provider['balanceAlertThreshold'] ?? 10),
|
||
'notes' => $provider['notes'] ?? '',
|
||
'docUrl' => $provider['docUrl'] ?? '',
|
||
'isFree' => $provider['isFree'] == 1,
|
||
'supportsBalance' => $provider['supportsBalance'] == 1,
|
||
'_hasKey' => !empty($provider['apiKey']),
|
||
'lastBalance' => $provider['lastBalance'] ? floatval($provider['lastBalance']) : null,
|
||
'lastBalanceCurrency' => $provider['lastBalanceCurrency'] ?? null,
|
||
'lastBalanceCheckedAt' => $provider['lastBalanceCheckedAt'] ? date('Y-m-d H:i:s', $provider['lastBalanceCheckedAt']) : null,
|
||
'extraConfig' => is_array($provider['extraConfig'] ?? null) ? $provider['extraConfig'] : (isset($provider['extraConfig']) && is_string($provider['extraConfig']) ? (json_decode($provider['extraConfig'], true) ?: []) : [])
|
||
];
|
||
}
|
||
|
||
return success($result);
|
||
}
|
||
|
||
/**
|
||
* 更新AI服务商配置
|
||
* @return \think\response\Json
|
||
*/
|
||
public function update()
|
||
{
|
||
// 验证是否为超级管理员
|
||
$user = $this->request->user ?? null;
|
||
if (!$user || $user['role'] !== 'superadmin') {
|
||
return error('无权限访问', 403);
|
||
}
|
||
|
||
$providerId = Request::param('providerId', '');
|
||
$data = Request::only([
|
||
'name', 'enabled', 'visible', 'apiKey', 'apiEndpoint', 'model', 'organizationId',
|
||
'maxTokens', 'balanceAlertEnabled', 'balanceAlertThreshold', 'notes',
|
||
'extraConfig'
|
||
]);
|
||
|
||
if (empty($providerId)) {
|
||
return error('服务商ID不能为空', 400);
|
||
}
|
||
|
||
// 查找服务商配置
|
||
$provider = AiProviderModel::where('providerId', $providerId)->find();
|
||
|
||
if (!$provider) {
|
||
return error('服务商配置不存在', 404);
|
||
}
|
||
|
||
// 处理 enabled 字段（前端传的是布尔值）
|
||
if (isset($data['enabled'])) {
|
||
$data['enabled'] = $data['enabled'] ? 1 : 0;
|
||
}
|
||
|
||
// 处理 balanceAlertEnabled 字段
|
||
if (isset($data['balanceAlertEnabled'])) {
|
||
$data['balanceAlertEnabled'] = $data['balanceAlertEnabled'] ? 1 : 0;
|
||
}
|
||
|
||
// 处理 visible 字段（显示/隐藏，数据库直接控制）
|
||
if (isset($data['visible'])) {
|
||
$data['visible'] = $data['visible'] ? 1 : 0;
|
||
}
|
||
|
||
// extraConfig 可为数组或 JSON 字符串，模型 type=json 会处理
|
||
if (isset($data['extraConfig']) && is_string($data['extraConfig'])) {
|
||
$decoded = json_decode($data['extraConfig'], true);
|
||
$data['extraConfig'] = is_array($decoded) ? $decoded : [];
|
||
}
|
||
|
||
// 如果API Key为空或包含脱敏标记（****），不更新（保持原值）
|
||
if (isset($data['apiKey'])) {
|
||
if (empty($data['apiKey']) || strpos($data['apiKey'], '****') !== false) {
|
||
unset($data['apiKey']);
|
||
}
|
||
}
|
||
|
||
// 更新配置
|
||
$provider->save($data);
|
||
|
||
// 返回更新后的数据（脱敏）
|
||
$result = [
|
||
'id' => $provider->providerId,
|
||
'name' => $provider->name,
|
||
'enabled' => $provider->enabled == 1,
|
||
'visible' => isset($provider->visible) ? ($provider->visible == 1) : true,
|
||
'apiKey' => $provider->apiKey ?? '',
|
||
'apiEndpoint' => $provider->apiEndpoint ?? '',
|
||
'model' => $provider->model ?? '',
|
||
'organizationId' => $provider->organizationId ?? '',
|
||
'maxTokens' => $provider->maxTokens ?? 4096,
|
||
'balanceAlertEnabled' => $provider->balanceAlertEnabled == 1,
|
||
'balanceAlertThreshold' => floatval($provider->balanceAlertThreshold ?? 10),
|
||
'notes' => $provider->notes ?? '',
|
||
'isFree' => $provider->isFree == 1,
|
||
'supportsBalance' => $provider->supportsBalance == 1,
|
||
'_hasKey' => !empty($provider->apiKey),
|
||
'extraConfig' => $provider->extraConfig ?? []
|
||
];
|
||
|
||
return success($result, '保存成功');
|
||
}
|
||
|
||
/**
|
||
* 批量更新AI服务商配置
|
||
* @return \think\response\Json
|
||
*/
|
||
public function batchUpdate()
|
||
{
|
||
// 验证是否为超级管理员
|
||
$user = $this->request->user ?? null;
|
||
if (!$user || $user['role'] !== 'superadmin') {
|
||
return error('无权限访问', 403);
|
||
}
|
||
|
||
$providers = Request::param('providers', []);
|
||
|
||
if (empty($providers) || !is_array($providers)) {
|
||
return error('配置数据不能为空', 400);
|
||
}
|
||
|
||
$successCount = 0;
|
||
$errors = [];
|
||
|
||
Db::startTrans();
|
||
try {
|
||
foreach ($providers as $providerData) {
|
||
$providerId = $providerData['id'] ?? $providerData['providerId'] ?? '';
|
||
|
||
if (empty($providerId)) {
|
||
$errors[] = '服务商ID不能为空';
|
||
continue;
|
||
}
|
||
|
||
$provider = AiProviderModel::where('providerId', $providerId)->find();
|
||
|
||
if (!$provider) {
|
||
$errors[] = "服务商 {$providerId} 不存在";
|
||
continue;
|
||
}
|
||
|
||
// 准备更新数据
|
||
$updateData = [];
|
||
if (isset($providerData['enabled'])) {
|
||
$updateData['enabled'] = $providerData['enabled'] ? 1 : 0;
|
||
}
|
||
if (isset($providerData['apiKey']) && !empty($providerData['apiKey'])) {
|
||
$updateData['apiKey'] = $providerData['apiKey'];
|
||
}
|
||
if (isset($providerData['apiEndpoint'])) {
|
||
$updateData['apiEndpoint'] = $providerData['apiEndpoint'];
|
||
}
|
||
if (isset($providerData['model'])) {
|
||
$updateData['model'] = $providerData['model'];
|
||
}
|
||
if (isset($providerData['organizationId'])) {
|
||
$updateData['organizationId'] = $providerData['organizationId'];
|
||
}
|
||
if (isset($providerData['maxTokens'])) {
|
||
$updateData['maxTokens'] = intval($providerData['maxTokens']);
|
||
}
|
||
if (isset($providerData['balanceAlertEnabled'])) {
|
||
$updateData['balanceAlertEnabled'] = $providerData['balanceAlertEnabled'] ? 1 : 0;
|
||
}
|
||
if (isset($providerData['balanceAlertThreshold'])) {
|
||
$updateData['balanceAlertThreshold'] = floatval($providerData['balanceAlertThreshold']);
|
||
}
|
||
if (isset($providerData['notes'])) {
|
||
$updateData['notes'] = $providerData['notes'];
|
||
}
|
||
if (isset($providerData['visible'])) {
|
||
$updateData['visible'] = $providerData['visible'] ? 1 : 0;
|
||
}
|
||
if (isset($providerData['extraConfig'])) {
|
||
$updateData['extraConfig'] = is_array($providerData['extraConfig'])
|
||
? $providerData['extraConfig']
|
||
: (is_string($providerData['extraConfig']) ? json_decode($providerData['extraConfig'], true) : []);
|
||
if (!is_array($updateData['extraConfig'])) {
|
||
$updateData['extraConfig'] = [];
|
||
}
|
||
}
|
||
|
||
$provider->save($updateData);
|
||
$successCount++;
|
||
}
|
||
|
||
Db::commit();
|
||
} catch (\Exception $e) {
|
||
Db::rollback();
|
||
return error('批量保存失败：' . $e->getMessage(), 500);
|
||
}
|
||
|
||
if (!empty($errors)) {
|
||
return error('部分配置保存失败：' . implode('；', $errors), 400);
|
||
}
|
||
|
||
return success(null, "成功保存 {$successCount} 个配置");
|
||
}
|
||
|
||
/**
|
||
* 查询余额（单个服务商）
|
||
* @return \think\response\Json
|
||
*/
|
||
public function queryBalance()
|
||
{
|
||
// 验证是否为超级管理员
|
||
$user = $this->request->user ?? null;
|
||
if (!$user || $user['role'] !== 'superadmin') {
|
||
return error('无权限访问', 403);
|
||
}
|
||
|
||
$providerId = Request::param('providerId', '');
|
||
|
||
if (empty($providerId)) {
|
||
return error('服务商ID不能为空', 400);
|
||
}
|
||
|
||
$provider = AiProviderModel::where('providerId', $providerId)->find();
|
||
|
||
if (!$provider) {
|
||
return error('服务商配置不存在', 404);
|
||
}
|
||
|
||
if (empty($provider->apiKey)) {
|
||
return error('请先配置 API Key', 400);
|
||
}
|
||
|
||
if (!$provider->supportsBalance) {
|
||
return error('该服务商暂不支持余额查询', 400);
|
||
}
|
||
|
||
// 调用余额查询服务
|
||
$balanceResult = $this->queryProviderBalance($provider);
|
||
|
||
// 更新最后查询的余额
|
||
if ($balanceResult['status'] === 'success' && isset($balanceResult['balance'])) {
|
||
$provider->lastBalance = $balanceResult['balance'];
|
||
$provider->lastBalanceCurrency = $balanceResult['currency'] ?? 'CNY';
|
||
$provider->lastBalanceCheckedAt = time();
|
||
$provider->save();
|
||
}
|
||
|
||
return success($balanceResult);
|
||
}
|
||
|
||
/**
|
||
* 批量查询余额
|
||
* @return \think\response\Json
|
||
*/
|
||
public function queryAllBalances()
|
||
{
|
||
// 验证是否为超级管理员
|
||
$user = $this->request->user ?? null;
|
||
if (!$user || $user['role'] !== 'superadmin') {
|
||
return error('无权限访问', 403);
|
||
}
|
||
|
||
$providerIds = Request::param('providerIds', []);
|
||
|
||
// 如果没有指定，查询所有已启用且已配置密钥的服务商
|
||
if (empty($providerIds)) {
|
||
$providers = AiProviderModel::where('enabled', 1)
|
||
->where('apiKey', '<>', '')
|
||
->where('apiKey', '<>', null)
|
||
->select();
|
||
} else {
|
||
$providers = AiProviderModel::where('providerId', 'in', $providerIds)
|
||
->where('apiKey', '<>', '')
|
||
->where('apiKey', '<>', null)
|
||
->select();
|
||
}
|
||
|
||
$results = [];
|
||
foreach ($providers as $provider) {
|
||
if (!$provider->supportsBalance) {
|
||
continue;
|
||
}
|
||
|
||
$balanceResult = $this->queryProviderBalance($provider);
|
||
|
||
// 更新最后查询的余额
|
||
if ($balanceResult['status'] === 'success' && isset($balanceResult['balance'])) {
|
||
$provider->lastBalance = $balanceResult['balance'];
|
||
$provider->lastBalanceCurrency = $balanceResult['currency'] ?? 'CNY';
|
||
$provider->lastBalanceCheckedAt = time();
|
||
$provider->save();
|
||
}
|
||
|
||
$results[] = $balanceResult;
|
||
}
|
||
|
||
return success($results);
|
||
}
|
||
|
||
/**
|
||
* 查询服务商余额（内部方法）
|
||
* @param AiProviderModel $provider
|
||
* @return array
|
||
*/
|
||
private function queryProviderBalance($provider)
|
||
{
|
||
// 这里需要实现各服务商的余额查询逻辑
|
||
// 由于各服务商的API不同，这里提供一个基础框架
|
||
|
||
$providerId = $provider->providerId;
|
||
$apiKey = $provider->getRawApiKey(); // 获取原始密钥用于API调用
|
||
|
||
// TODO: 实现各服务商的余额查询API调用
|
||
// 目前返回模拟数据，实际需要调用各服务商的API
|
||
|
||
try {
|
||
switch ($providerId) {
|
||
case 'openai':
|
||
// OpenAI余额查询逻辑
|
||
return $this->queryOpenAIBalance($apiKey);
|
||
|
||
case 'deepseek':
|
||
// DeepSeek余额查询逻辑
|
||
return $this->queryDeepSeekBalance($apiKey);
|
||
|
||
case 'moonshot':
|
||
// Moonshot余额查询逻辑
|
||
return $this->queryMoonshotBalance($apiKey);
|
||
|
||
default:
|
||
return [
|
||
'providerId' => $providerId,
|
||
'providerName' => $provider->name,
|
||
'status' => 'unsupported',
|
||
'message' => '该服务商暂不支持余额查询',
|
||
'balance' => null,
|
||
'currency' => null,
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
}
|
||
} catch (\Exception $e) {
|
||
return [
|
||
'providerId' => $providerId,
|
||
'providerName' => $provider->name,
|
||
'status' => 'error',
|
||
'message' => '查询失败：' . $e->getMessage(),
|
||
'balance' => null,
|
||
'currency' => null,
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 查询OpenAI余额
|
||
* @param string $apiKey
|
||
* @return array
|
||
*/
|
||
private function queryOpenAIBalance($apiKey)
|
||
{
|
||
// TODO: 实现OpenAI余额查询
|
||
// OpenAI没有直接的余额查询API，需要通过使用情况估算
|
||
return [
|
||
'providerId' => 'openai',
|
||
'providerName' => 'OpenAI (GPT)',
|
||
'status' => 'success',
|
||
'message' => '余额查询成功：$100.00',
|
||
'balance' => 100.00,
|
||
'currency' => 'USD',
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 查询DeepSeek余额
|
||
* @param string $apiKey
|
||
* @return array
|
||
*/
|
||
private function queryDeepSeekBalance($apiKey)
|
||
{
|
||
// TODO: 实现DeepSeek余额查询
|
||
try {
|
||
// 示例：调用DeepSeek API查询余额
|
||
// $response = file_get_contents('https://api.deepseek.com/v1/balance', [
|
||
// 'http' => [
|
||
// 'method' => 'GET',
|
||
// 'header' => "Authorization: Bearer {$apiKey}\r\n"
|
||
// ]
|
||
// ]);
|
||
|
||
return [
|
||
'providerId' => 'deepseek',
|
||
'providerName' => 'DeepSeek',
|
||
'status' => 'success',
|
||
'message' => '余额查询成功：¥500.00',
|
||
'balance' => 500.00,
|
||
'currency' => 'CNY',
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
} catch (\Exception $e) {
|
||
return [
|
||
'providerId' => 'deepseek',
|
||
'providerName' => 'DeepSeek',
|
||
'status' => 'error',
|
||
'message' => '查询失败：' . $e->getMessage(),
|
||
'balance' => null,
|
||
'currency' => null,
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 查询Moonshot余额
|
||
* @param string $apiKey
|
||
* @return array
|
||
*/
|
||
private function queryMoonshotBalance($apiKey)
|
||
{
|
||
// TODO: 实现Moonshot余额查询
|
||
try {
|
||
// 示例：调用Moonshot API查询余额
|
||
return [
|
||
'providerId' => 'moonshot',
|
||
'providerName' => 'Moonshot (Kimi)',
|
||
'status' => 'success',
|
||
'message' => '余额查询成功：¥200.00',
|
||
'balance' => 200.00,
|
||
'currency' => 'CNY',
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
} catch (\Exception $e) {
|
||
return [
|
||
'providerId' => 'moonshot',
|
||
'providerName' => 'Moonshot (Kimi)',
|
||
'status' => 'error',
|
||
'message' => '查询失败：' . $e->getMessage(),
|
||
'balance' => null,
|
||
'currency' => null,
|
||
'checkedAt' => date('Y-m-d H:i:s')
|
||
];
|
||
}
|
||
}
|
||
}
|
||
|