feat: Enhance AI completion capabilities with new model support and improved context handling

Author: SonyLeo

packages/plugins/robot/src/components/header-extension/robot-setting/RobotSetting.vue

@@ -42,7 +42,7 @@
                 快速模型
                 <tiny-tooltip
                   effect="light"
-                  content="用于代码补全、话题命名等快速响应场景。推荐选择带「代码」标签的模型以获得更好的代码补全效果。"
+                  content="用于代码补全、话题命名等场景。建议选择轻量模型以实现更快的响应速度，例如flash类型或8b/14b模型。"
                   placement="top"
                 >
                   <svg-icon class="help-link" name="plugin-icon-plugin-help"></svg-icon>
@@ -51,25 +51,11 @@
               <tiny-select
                 clearable
                 v-model="state.modelSelection.quickModel"
+                :options="compactModelOptions"
                 filterable
                 placeholder="请选择"
                 @change="handleCompactModelChange"
-                popper-class="model-select-popper"
-              >
-                <template v-for="item in compactModelOptions" :key="item.value">
-                  <tiny-option :label="item.label" :value="item.value">
-                    <span class="left">{{ item.label }}</span>
-                    <div>
-                      <tiny-tag v-if="item.capabilities?.codeCompletion" type="success" effect="light" size="small"
-                        >代码</tiny-tag
-                      >
-                      <tiny-tag v-if="item.capabilities?.compact" type="info" effect="light" size="small"
-                        >轻量</tiny-tag
-                      >
-                    </div>
-                  </tiny-option>
-                </template>
-              </tiny-select>
+              ></tiny-select>
             </tiny-form-item>
 
             <div v-if="selectedDefaultModelInfo" class="model-info">
@@ -186,8 +172,8 @@ const emit = defineEmits(['close'])
 const {
   robotSettingState,
   saveRobotSettingState,
+  getAllAvailableModels,
   getCompactModels,
-  getNonCodeCompletionModels,
   addCustomService,
   updateService,
   deleteService,
@@ -200,6 +186,15 @@ const getModelValue = (serviceId: string, modelName: string) => {
   return serviceId && modelName ? `${serviceId}::${modelName}` : ''
 }
 
+const parseModelValue = (value = '') => {
+  const [serviceId = '', modelName = ''] = value.split('::')
+
+  return {
+    serviceId,
+    modelName
+  }
+}
+
 const state = reactive({
   activeTab: 'model-selection',
   modelSelection: {
@@ -210,9 +205,28 @@ const state = reactive({
   editingService: undefined as ModelService | undefined
 })
 
-// 获取所有可用模型选项（排除代码补全专用模型）
+const syncModelSelection = () => {
+  state.modelSelection.defaultModel = getModelValue(
+    robotSettingState.defaultModel.serviceId,
+    robotSettingState.defaultModel.modelName
+  )
+  state.modelSelection.quickModel = getModelValue(
+    robotSettingState.quickModel.serviceId,
+    robotSettingState.quickModel.modelName
+  )
+}
+
+const notifyMissingApiKey = (service: ModelService) => {
+  useNotify({
+    type: 'warning',
+    title: '未配置API Key',
+    message: `请先为 ${service.label} 配置API Key`
+  })
+}
+
+// 获取所有可用模型选项
 const allModelOptions = computed(() => {
-  return getNonCodeCompletionModels().map((model) => ({
+  return getAllAvailableModels().map((model) => ({
     label: model.displayLabel,
     value: model.value,
     capabilities: model.capabilities
@@ -221,14 +235,10 @@ const allModelOptions = computed(() => {
 
 // 获取快速模型选项
 const compactModelOptions = computed(() => {
-  const models = getCompactModels().map((model) => ({
+  return getCompactModels().map((model) => ({
     label: model.displayLabel,
-    value: model.value,
-    capabilities: model.capabilities,
-    serviceName: model.serviceName
+    value: model.value
   }))
-
-  return models.sort((a, b) => a.serviceName.localeCompare(b.serviceName, 'zh-CN'))
 })
 
 // 获取当前选择的默认模型信息
@@ -249,16 +259,15 @@ const handleBack = () => {
 }
 
 const handleModelChange = () => {
-  const [defaultServiceId, defaultModelName] = state.modelSelection.defaultModel.split('::')
+  const { serviceId: defaultServiceId, modelName: defaultModelName } = parseModelValue(
+    state.modelSelection.defaultModel
+  )
 
   // 检查API Key
   const defaultService = getServiceById(defaultServiceId)
   if (defaultService && !defaultService.apiKey && !defaultService.allowEmptyApiKey) {
-    useNotify({
-      type: 'warning',
-      title: '未配置API Key',
-      message: `请先为 ${defaultService.label} 配置API Key`
-    })
+    notifyMissingApiKey(defaultService)
+    syncModelSelection()
     state.activeTab = 'services'
     return
   }
@@ -273,7 +282,16 @@ const handleModelChange = () => {
 }
 
 const handleCompactModelChange = () => {
-  const [quickServiceId = '', quickModelName = ''] = (state.modelSelection.quickModel || '').split('::')
+  const { serviceId: quickServiceId, modelName: quickModelName } = parseModelValue(state.modelSelection.quickModel)
+  const quickService = getServiceById(quickServiceId)
+
+  if (quickService && !quickService.apiKey && !quickService.allowEmptyApiKey) {
+    notifyMissingApiKey(quickService)
+    syncModelSelection()
+    state.activeTab = 'services'
+    return
+  }
+
   const updatedState = {
     quickModel: {
       serviceId: quickServiceId,
@@ -289,22 +307,50 @@ const addService = () => {
 }
 
 const editService = (service: any) => {
-  state.editingService = JSON.parse(JSON.stringify(service)) as ModelService
+  state.editingService = JSON.parse(JSON.stringify(service))
   state.showServiceDialog = true
 }
 
 const handleDeleteService = (serviceId: string) => {
   deleteService(serviceId)
+
+  const shouldResetDefaultModel = robotSettingState.defaultModel.serviceId === serviceId
+  const shouldResetQuickModel = robotSettingState.quickModel.serviceId === serviceId
+
+  if (shouldResetDefaultModel || shouldResetQuickModel) {
+    saveRobotSettingState({
+      ...(shouldResetDefaultModel
+        ? {
+            defaultModel: {
+              serviceId: '',
+              modelName: ''
+            }
+          }
+        : {}),
+      ...(shouldResetQuickModel
+        ? {
+            quickModel: {
+              serviceId: '',
+              modelName: ''
+            }
+          }
+        : {})
+    })
+  }
+
+  syncModelSelection()
 }
 
-const handleServiceConfirm = (serviceData: Partial<ModelService>) => {
+const handleServiceConfirm = async (serviceData: Partial<ModelService>) => {
   if (serviceData.id) {
     // 更新现有服务
-    updateService(serviceData.id, serviceData)
+    await updateService(serviceData.id, serviceData)
   } else {
     // 添加新服务
-    addCustomService(serviceData as any)
+    await addCustomService(serviceData as any)
   }
+
+  syncModelSelection()
 }
 </script>
 

packages/plugins/robot/src/composables/core/useConfig.ts

@@ -15,7 +15,13 @@ import { reactive, readonly } from 'vue'
 import { DEFAULT_LLM_MODELS } from '../../constants'
 import { getRobotServiceOptions } from '../../utils'
 import { ChatMode } from '../../types/mode.types'
-import type { ModelConfig, ModelService, RobotSettings, SelectedModelInfo } from '../../types/setting.types'
+import type {
+  CompletionProtocol,
+  ModelConfig,
+  ModelService,
+  RobotSettings,
+  SelectedModelInfo
+} from '../../types/setting.types'
 import apiService from '../../services/api'
 
 const SETTING_STORAGE_KEY = 'tiny-engine-robot-settings'
@@ -44,6 +50,48 @@ const getAIModelOptions = () => {
   return mergeAIModelOptions(DEFAULT_LLM_MODELS, customAIModels) // eslint-disable-line
 }
 
+const inferCompletionProtocol = ({
+  provider = '',
+  baseUrl = '',
+  modelName = '',
+  capabilities = {}
+}: {
+  provider?: string
+  baseUrl?: string
+  modelName?: string
+  capabilities?: ModelConfig['capabilities']
+}): CompletionProtocol | null => {
+  if (capabilities?.completionProtocol) {
+    return capabilities.completionProtocol
+  }
+
+  const normalizedProvider = provider.toLowerCase()
+  if (normalizedProvider === 'bailian') {
+    return 'qwen'
+  }
+  if (normalizedProvider === 'deepseek') {
+    return 'deepseek'
+  }
+
+  const normalizedBaseUrl = baseUrl.toLowerCase()
+  if (normalizedBaseUrl.includes('dashscope.aliyuncs.com')) {
+    return 'qwen'
+  }
+  if (normalizedBaseUrl.includes('deepseek.com')) {
+    return 'deepseek'
+  }
+
+  const normalizedModelName = modelName.toLowerCase()
+  if (normalizedModelName.includes('qwen')) {
+    return 'qwen'
+  }
+  if (normalizedModelName.includes('deepseek')) {
+    return 'deepseek'
+  }
+
+  return null
+}
+
 // 初始化内置服务
 const initBuiltInServices = (): ModelService[] => {
   return getAIModelOptions().map((service: any) => ({
@@ -120,7 +168,9 @@ const migrateOldSettings = (oldSettings: any): RobotSettings | null => {
       customService.models.push({
         name: customizeModel.completeModel,
         label: customizeModel.completeModel,
-        capabilities: { compact: true }
+        capabilities: {
+          compact: true
+        }
       })
     }
     services.push(customService)
@@ -318,19 +368,9 @@ const getAllAvailableModels = () => {
   )
 }
 
-// 获取快速模型列表（包含 compact 或 codeCompletion 的模型）
+// 获取快速模型列表
 const getCompactModels = () => {
-  return getAllAvailableModels().filter((model) => model.capabilities?.compact || model.capabilities?.codeCompletion)
-}
-
-// 获取代码补全优化模型列表
-const getCodeCompletionModels = () => {
-  return getAllAvailableModels().filter((model) => model.capabilities?.codeCompletion)
-}
-
-// 获取非代码补全模型列表（用于默认助手模型）
-const getNonCodeCompletionModels = () => {
-  return getAllAvailableModels().filter((model) => !model.capabilities?.codeCompletion)
+  return getAllAvailableModels().filter((model) => model.capabilities?.compact)
 }
 
 const updateThinkingState = (value: boolean) => {
@@ -430,6 +470,13 @@ const getSelectedQuickModelInfo = (): SelectedModelInfo => {
     (m) => m.name === robotSettingState.quickModel.modelName
   )
   const { name = '', label = '', capabilities = {} } = currentModel || {}
+  const completionProtocol =
+    inferCompletionProtocol({
+      provider: currentService?.provider,
+      baseUrl: currentService?.baseUrl,
+      modelName: name,
+      capabilities
+    }) || null
 
   const { models, ...service } = currentService ?? ({} as Partial<ModelService>)
 
@@ -444,6 +491,7 @@ const getSelectedQuickModelInfo = (): SelectedModelInfo => {
     // 模型兼容字段
     model: robotSettingState.quickModel.modelName,
     completeModel: robotSettingState.quickModel.modelName || '',
+    completionProtocol,
     // 服务兼容字段
     baseUrl: currentService?.baseUrl || '',
     apiKey: currentService?.apiKey || ''
@@ -466,8 +514,6 @@ export default () => {
     getModelCapabilities,
     getAllAvailableModels,
     getCompactModels,
-    getCodeCompletionModels, // 代码补全模型列表
-    getNonCodeCompletionModels, // 非代码补全模型列表
     getSelectedModelInfo, // 对话模型信息
     getSelectedQuickModelInfo, // 快速模型信息
 

packages/plugins/robot/src/constants/model-config.ts

@@ -57,7 +57,6 @@ export const DEFAULT_LLM_MODELS = [
         name: 'qwen3-coder-plus',
         capabilities: {
           toolCalling: true,
-          codeCompletion: true,
           reasoning: reasoningExtraBody,
           jsonOutput: bailianJsonOutputExtraBody
         }
@@ -82,24 +81,23 @@ export const DEFAULT_LLM_MODELS = [
         }
       },
       {
-        label: 'Qwen2.5 Coder编程模型-最快响应',
-        name: 'qwen-coder-turbo-latest',
+        label: 'Qwen Coder编程模型（Flash）',
+        name: 'qwen3-coder-flash',
         capabilities: {
           toolCalling: true,
           compact: true,
-          codeCompletion: true,
           jsonOutput: bailianJsonOutputExtraBody
         }
       },
       {
-        label: 'Qwen2.5 Coder编程模型（32B）',
-        name: 'qwen2.5-coder-32b-instruct',
-        capabilities: {
-          toolCalling: true,
-          compact: true,
-          codeCompletion: true,
-          jsonOutput: bailianJsonOutputExtraBody
-        }
+        label: 'Qwen3（14b）',
+        name: 'qwen3-14b',
+        capabilities: { compact: true, toolCalling: true, jsonOutput: bailianJsonOutputExtraBody }
+      },
+      {
+        label: 'Qwen3（8b）',
+        name: 'qwen3-8b',
+        capabilities: { compact: true, toolCalling: true, jsonOutput: bailianJsonOutputExtraBody }
       }
     ]
   },
@@ -122,16 +120,6 @@ export const DEFAULT_LLM_MODELS = [
           },
           jsonOutput: jsonOutputExtraBody
         }
-      },
-      {
-        label: 'Deepseek Coder编程模型',
-        name: 'deepseek-coder',
-        capabilities: {
-          toolCalling: true,
-          compact: true,
-          codeCompletion: true,
-          jsonOutput: jsonOutputExtraBody
-        }
       }
     ]
   }