chore: 优化模型分组逻辑 (#60)

Author: jiangwel

images/image-list-model.jpg

@@ -15,7 +15,7 @@ export const CodeTag = ({ size, showTooltip, showLabel, ...restProps }: Props) =
     <CustomTag
       size={size}
       color="#6366f1"
-      icon="Code"
+      icon="代码生成"
       tooltip={showTooltip ? t('models.type.code_generation') : undefined}
       {...restProps}>
       {showLabel ? t('models.type.code_generation') : ''}

ui/ModelModal/src/components/Tags/ModelCapabilities/CodeTag.tsx

@@ -14,7 +14,7 @@ export const ReasoningTag = ({ size, showTooltip, showLabel, ...restProps }: Pro
     <CustomTag
       size={size}
       color="#6372bd"
-      icon="推理"
+      icon="深度思考"
       tooltip={showTooltip ? t('models.type.reasoning') : undefined}
       {...restProps}>
       {showLabel ? t('models.type.reasoning') : ''}

ui/ModelModal/src/components/Tags/ModelCapabilities/ReasoningTag.tsx

@@ -562,17 +562,6 @@ export function isWebSearchModel(model_id: string, provider: string): boolean {
   return false
 }
 
-export const getModelGroup = (model_id: string): string => {
-  // 1. 提取model_id第一个-之前的部分
-  const firstPart = model_id.split('-')[0];
-  
-  // 2. 从头获取连续的纯字母部分
-  const withoutNumbers = firstPart.match(/^[a-zA-Z]+/)?.[0] || '';
-  
-  // 3. 返回结果
-  return withoutNumbers;
-}
-
 export function getModelLogo(modelId: string) {
   const isLight = true
 
@@ -706,4 +695,52 @@ export function getModelLogo(modelId: string) {
   }
 
   return undefined
+}
+
+/**
+ * 从模型 ID 中提取默认组名。
+ * 规则如下：
+ * 1. 第一类分隔规则：以第一个出现的分隔符分割，取第 0 个部分作为组名。
+ * 2. 第二类分隔规则：取前两个部分拼接（如 'a-b-c' 得到 'a-b'）。
+ * 3. 其他情况返回 id。
+ *
+ * 例如：
+ * - 'gpt-3.5-turbo-16k-0613' => 'gpt-3.5'
+ * - 'qwen3:32b' => 'qwen3'
+ * - 'Qwen/Qwen3-32b' => 'qwen'
+ * - 'deepseek-r1' => 'deepseek-r1'
+ * - 'o3' => 'o3'
+ *
+ * @param {string} id 模型 ID 字符串
+ * @param {string} [provider] 提供商 ID 字符串
+ * @returns {string} 提取的组名
+ */
+export const getModelGroup = (id: string, provider?: string): string => {
+  const str = id.toLowerCase()
+
+  // 定义分隔符
+  let firstDelimiters = ['/', ' ', ':']
+  let secondDelimiters = ['-', '_']
+
+  if (provider && ['aihubmix', 'silicon', 'ocoolai', 'o3', 'dmxapi'].includes(provider.toLowerCase())) {
+    firstDelimiters = ['/', ' ', '-', '_', ':']
+    secondDelimiters = []
+  }
+
+  // 第一类分隔规则
+  for (const delimiter of firstDelimiters) {
+    if (str.includes(delimiter)) {
+      return str.split(delimiter)[0]
+    }
+  }
+
+  // 第二类分隔规则
+  for (const delimiter of secondDelimiters) {
+    if (str.includes(delimiter)) {
+      const parts = str.split(delimiter)
+      return parts.length > 1 ? parts[0] + '-' + parts[1] : parts[0]
+    }
+  }
+
+  return str
 }