From e9bae8580228a690ca4d2310d28bc9a118ddc572 Mon Sep 17 00:00:00 2001
From: nanhaoluo <3075912108@qq.com>
Date: Mon, 26 Jan 2026 11:01:51 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E5=B0=8F=E7=B1=B3=20?=
 =?UTF-8?q?Mimo=20AI=20=E6=A8=A1=E5=9E=8B=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 在设置页 AI 服务商列表中添加小米 Mimo 选项
- 实现 argon_call_xiaomi_api() 函数，使用 OpenAI 兼容格式
- 实现 argon_get_xiaomi_models() 模型列表获取函数
- 在 AI 摘要生成中添加 xiaomi 分支支持
- 在垃圾评论检测中添加小米 Mimo 端点配置
- 在关键词提取中添加小米 Mimo 端点配置
- 在批量垃圾评论检测中添加小米 Mimo 端点配置
- 默认模型：MiMo-V2-Flash
- API 端点：https://api.mimo.xiaomi.com/v1/chat/completions
---
 functions.php | 102 ++++++++++++++++++++++++++++++++++++++++++++++++++
 settings.php  |   1 +
 2 files changed, 103 insertions(+)

diff --git a/functions.php b/functions.php
index cf17372..0e19d77 100644
--- a/functions.php
+++ b/functions.php
@@ -6487,6 +6487,8 @@ function argon_generate_ai_summary($post) {
 			return argon_call_anthropic_api($api_key, $prompt, $content);
 		case 'deepseek':
 			return argon_call_deepseek_api($api_key, $prompt, $content);
+		case 'xiaomi':
+			return argon_call_xiaomi_api($api_key, $prompt, $content);
 		case 'qianwen':
 			return argon_call_qianwen_api($api_key, $prompt, $content);
 		case 'wenxin':
@@ -6948,6 +6950,53 @@ function argon_call_deepseek_api($api_key, $prompt, $content) {
 	return false;
 }
 
+/**
+ * 调用小米 Mimo API
+ */
+function argon_call_xiaomi_api($api_key, $prompt, $content) {
+	$endpoint = get_option('argon_ai_summary_api_endpoint', '');
+	if (empty($endpoint)) {
+		$endpoint = 'https://api.mimo.xiaomi.com/v1/chat/completions';
+	}
+	
+	$model = get_option('argon_ai_summary_model', '');
+	if (empty($model)) {
+		$model = 'MiMo-V2-Flash';
+	}
+
+	$data = [
+		'model' => $model,
+		'messages' => [
+			['role' => 'system', 'content' => $prompt],
+			['role' => 'user', 'content' => $content]
+		],
+		'temperature' => 0.7,
+		'max_tokens' => 500
+	];
+
+	$response = wp_remote_post($endpoint, [
+		'headers' => [
+			'Content-Type' => 'application/json',
+			'Authorization' => 'Bearer ' . $api_key
+		],
+		'body' => json_encode($data),
+		'timeout' => 30
+	]);
+
+	if (is_wp_error($response)) {
+		error_log('Argon AI Summary Error: ' . $response->get_error_message());
+		return false;
+	}
+
+	$body = json_decode(wp_remote_retrieve_body($response), true);
+	
+	if (isset($body['choices'][0]['message']['content'])) {
+		return trim($body['choices'][0]['message']['content']);
+	}
+
+	return false;
+}
+
 /**
  * 调用豆包 (火山引擎) API
  */
@@ -7073,6 +7122,9 @@ function argon_get_ai_models() {
 		case 'deepseek':
 			$models = argon_get_deepseek_models($api_key, $api_endpoint);
 			break;
+		case 'xiaomi':
+			$models = argon_get_xiaomi_models($api_key, $api_endpoint);
+			break;
 		case 'qianwen':
 			$models = argon_get_qianwen_models($api_key, $api_endpoint);
 			break;
@@ -7224,6 +7276,50 @@ function argon_get_deepseek_models($api_key, $custom_endpoint = '') {
 	];
 }
 
+/**
+ * 获取小米 Mimo 模型列表
+ */
+function argon_get_xiaomi_models($api_key, $custom_endpoint = '') {
+	$endpoint = !empty($custom_endpoint) ? $custom_endpoint : 'https://api.mimo.xiaomi.com/v1/models';
+	$endpoint = preg_replace('#/v1/chat/completions$#', '/v1/models', $endpoint);
+	
+	$response = wp_remote_get($endpoint, [
+		'headers' => [
+			'Authorization' => 'Bearer ' . $api_key
+		],
+		'timeout' => 15
+	]);
+	
+	if (is_wp_error($response)) {
+		// API 调用失败,返回预设列表
+		return [
+			['id' => 'MiMo-V2-Flash', 'name' => 'MiMo-V2-Flash (推荐)']
+		];
+	}
+	
+	$body = json_decode(wp_remote_retrieve_body($response), true);
+	
+	if (!isset($body['data'])) {
+		return [
+			['id' => 'MiMo-V2-Flash', 'name' => 'MiMo-V2-Flash (推荐)']
+		];
+	}
+	
+	$models = [];
+	foreach ($body['data'] as $model) {
+		if (isset($model['id'])) {
+			$models[] = [
+				'id' => $model['id'],
+				'name' => $model['id']
+			];
+		}
+	}
+	
+	return !empty($models) ? $models : [
+		['id' => 'MiMo-V2-Flash', 'name' => 'MiMo-V2-Flash (推荐)']
+	];
+}
+
 /**
  * 获取通义千问模型列表
  */
@@ -7757,6 +7853,7 @@ function argon_call_ai_api_for_spam_detection($provider, $api_key, $model, $prom
 		'openai' => 'gpt-4o-mini',
 		'anthropic' => 'claude-3-5-haiku-20241022',
 		'deepseek' => 'deepseek-chat',
+		'xiaomi' => 'MiMo-V2-Flash',
 		'qianwen' => 'qwen-turbo',
 		'wenxin' => 'ernie-4.0-turbo-8k',
 		'doubao' => 'doubao-pro-32k',
@@ -7788,6 +7885,7 @@ function argon_call_ai_api_for_spam_detection($provider, $api_key, $model, $prom
 			'openai' => 'https://api.openai.com/v1/chat/completions',
 			'anthropic' => 'https://api.anthropic.com/v1/messages',
 			'deepseek' => 'https://api.deepseek.com/v1/chat/completions',
+			'xiaomi' => 'https://api.mimo.xiaomi.com/v1/chat/completions',
 			'qianwen' => 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
 			'wenxin' => 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions',
 			'doubao' => 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
@@ -8009,6 +8107,7 @@ function argon_call_ai_for_keyword_extraction($provider, $api_key, $model, $prom
 		'openai' => 'gpt-4o-mini',
 		'anthropic' => 'claude-3-5-haiku-20241022',
 		'deepseek' => 'deepseek-chat',
+		'xiaomi' => 'MiMo-V2-Flash',
 		'siliconflow' => 'Qwen/Qwen2.5-7B-Instruct'
 	];
 	
@@ -8032,6 +8131,7 @@ function argon_call_ai_for_keyword_extraction($provider, $api_key, $model, $prom
 			'openai' => 'https://api.openai.com/v1/chat/completions',
 			'anthropic' => 'https://api.anthropic.com/v1/messages',
 			'deepseek' => 'https://api.deepseek.com/v1/chat/completions',
+			'xiaomi' => 'https://api.mimo.xiaomi.com/v1/chat/completions',
 			'siliconflow' => 'https://api.siliconflow.cn/v1/chat/completions'
 		];
 		$endpoint = isset($endpoints[$provider]) ? $endpoints[$provider] : $endpoints['openai'];
@@ -8776,6 +8876,7 @@ function argon_call_ai_api_for_batch_spam_detection($provider, $api_key, $model,
 		'openai' => 'gpt-4o-mini',
 		'anthropic' => 'claude-3-5-haiku-20241022',
 		'deepseek' => 'deepseek-chat',
+		'xiaomi' => 'MiMo-V2-Flash',
 		'qianwen' => 'qwen-turbo',
 		'wenxin' => 'ernie-4.0-turbo-8k',
 		'doubao' => 'doubao-pro-32k',
@@ -8807,6 +8908,7 @@ function argon_call_ai_api_for_batch_spam_detection($provider, $api_key, $model,
 			'openai' => 'https://api.openai.com/v1/chat/completions',
 			'anthropic' => 'https://api.anthropic.com/v1/messages',
 			'deepseek' => 'https://api.deepseek.com/v1/chat/completions',
+			'xiaomi' => 'https://api.mimo.xiaomi.com/v1/chat/completions',
 			'qianwen' => 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions',
 			'wenxin' => 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions',
 			'doubao' => 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
diff --git a/settings.php b/settings.php
index 897be70..86a790d 100644
--- a/settings.php
+++ b/settings.php
@@ -2007,6 +2007,7 @@ function themeoptions_page(){
 								<option value="openai" <?php if ($argon_ai_summary_provider=='openai'){echo 'selected';} ?>>OpenAI (ChatGPT)</option>
 								<option value="anthropic" <?php if ($argon_ai_summary_provider=='anthropic'){echo 'selected';} ?>>Anthropic (Claude)</option>
 								<option value="deepseek" <?php if ($argon_ai_summary_provider=='deepseek'){echo 'selected';} ?>>DeepSeek</option>
+								<option value="xiaomi" <?php if ($argon_ai_summary_provider=='xiaomi'){echo 'selected';} ?>><?php _e('小米 Mimo', 'argon');?></option>
 								<option value="qianwen" <?php if ($argon_ai_summary_provider=='qianwen'){echo 'selected';} ?>><?php _e('通义千问', 'argon');?></option>
 								<option value="wenxin" <?php if ($argon_ai_summary_provider=='wenxin'){echo 'selected';} ?>><?php _e('文心一言', 'argon');?></option>
 								<option value="doubao" <?php if ($argon_ai_summary_provider=='doubao'){echo 'selected';} ?>><?php _e('豆包 (火山引擎)', 'argon');?></option>