Refactor AI completion module: enhance model handling, improve context analysis, and streamline prompt generation

Author: SonyLeo

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

@@ -246,7 +246,7 @@ const syncModelSelection = () => {
   )
 }
 
-const notifyMissingApiKey = (service: any) => {
+const notifyMissingApiKey = (service: ModelService) => {
   useNotify({
     type: 'warning',
     title: '未配置API Key',
@@ -364,7 +364,7 @@ const addService = () => {
   state.showServiceDialog = true
 }
 
-const editService = (service: ModelService) => {
+const editService = (service: any) => {
   state.editingService = JSON.parse(JSON.stringify(service))
   state.showServiceDialog = true
 }

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

@@ -145,6 +145,22 @@ const getFallbackCompletionModel = (services: ModelService[], preferredServiceId
   }
 }
 
+const resolveCompletionModelSelection = (services: ModelService[], serviceId = '', modelName = '') => {
+  if (serviceId && modelName) {
+    const selectedService = services.find((service) => service.id === serviceId)
+    const selectedModel = selectedService?.models.find((model) => model.name === modelName)
+
+    if (isCompletionCapableModel(selectedService, selectedModel)) {
+      return {
+        serviceId,
+        modelName
+      }
+    }
+  }
+
+  return getFallbackCompletionModel(services, serviceId)
+}
+
 const initBuiltInServices = (): ModelService[] => {
   return getAIModelOptions().map((service: any) => ({
     id: service.provider,
@@ -229,19 +245,21 @@ const migrateOldSettings = (oldSettings: any): RobotSettings | null => {
     services.push(customService)
   }
 
-  // 确定默认模型和快速模型
+  // 确定默认模型、快速模型和代码补全模型
   const selectedModel = activeName === 'existingModels' ? existModel : customizeModel
   const defaultServiceId =
     activeName === 'existingModels' ? services.find((s) => s.baseUrl === selectedModel?.baseUrl)?.id : ''
+  const legacyCompleteModelName = selectedModel?.completeModel || ''
 
   const quickModel = {
     serviceId: defaultServiceId || '',
-    modelName: selectedModel?.completeModel || ''
+    modelName: legacyCompleteModelName
   }
-  const completionModel =
-    quickModel.modelName && quickModel.serviceId
-      ? getFallbackCompletionModel(services, quickModel.serviceId)
-      : getFallbackCompletionModel(services, defaultServiceId || services[0]?.id || '')
+  const completionModel = resolveCompletionModelSelection(
+    services,
+    defaultServiceId || services[0]?.id || '',
+    legacyCompleteModelName
+  )
 
   return {
     version: SETTING_VERSION,
@@ -546,13 +564,6 @@ 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>)
 
@@ -567,7 +578,6 @@ const getSelectedQuickModelInfo = (): SelectedModelInfo => {
     // 模型兼容字段
     model: robotSettingState.quickModel.modelName,
     completeModel: robotSettingState.quickModel.modelName || '',
-    completionProtocol,
     // 服务兼容字段
     baseUrl: currentService?.baseUrl || '',
     apiKey: currentService?.apiKey || ''

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

@@ -94,7 +94,6 @@ export const DEFAULT_LLM_MODELS = [
         name: 'qwen-coder-turbo-latest',
         capabilities: {
           toolCalling: true,
-          compact: true,
           jsonOutput: bailianJsonOutputExtraBody,
           completionProtocol: 'qwen'
         }
@@ -122,7 +121,6 @@ export const DEFAULT_LLM_MODELS = [
         name: 'deepseek-chat',
         capabilities: {
           toolCalling: true,
-          compact: true,
           completionProtocol: 'deepseek',
           reasoning: {
             extraBody: {

packages/plugins/script/src/Main.vue

@@ -122,8 +122,9 @@ export default {
       // 保留原有的 ESLint
       state.linterWorker = initLinter(editor, monacoRef.value.getMonaco(), state) as any
 
-      const { aiCompletionEnabled, aiCompletionTrigger = 'onIdle' } =
-        getMergeMeta('engine.plugins.pagecontroller')?.options || {}
+      const pageControllerOptions = getMergeMeta('engine.plugins.pagecontroller')?.options || {}
+      const aiCompletionEnabled = pageControllerOptions.aiCompletionEnabled ?? pageControllerOptions.enableAICompletion
+      const aiCompletionTrigger = pageControllerOptions.aiCompletionTrigger || 'onIdle'
 
       if (aiCompletionEnabled) {
         try {

packages/plugins/script/src/ai-completion/adapters/deepseekAdapter.js

@@ -1,5 +1,39 @@
 import { HTTP_CONFIG, ERROR_MESSAGES, DEEPSEEK_CONFIG } from '../constants.js'
 
+export function buildDeepSeekCompletionsUrl(baseUrl) {
+  const normalizedBaseUrl = String(baseUrl || '').trim()
+
+  if (!normalizedBaseUrl) {
+    return ''
+  }
+
+  const normalizePath = (path = '') => {
+    const trimmedPath = path.replace(/\/+$/, '')
+
+    if (trimmedPath.endsWith('/beta/completions')) {
+      return trimmedPath
+    }
+
+    if (trimmedPath.endsWith('/beta')) {
+      return `${trimmedPath}/completions`
+    }
+
+    if (/\/v1$/i.test(trimmedPath)) {
+      return trimmedPath.replace(/\/v1$/i, '/beta/completions')
+    }
+
+    return `${trimmedPath}/beta/completions`
+  }
+
+  try {
+    const parsedUrl = new URL(normalizedBaseUrl)
+    parsedUrl.pathname = normalizePath(parsedUrl.pathname)
+    return parsedUrl.toString()
+  } catch {
+    return normalizePath(normalizedBaseUrl)
+  }
+}
+
 /**
  * 构建 DeepSeek FIM 格式的请求参数
  * @param {string} fileContent - 文件内容（包含 [CURSOR] 标记）
@@ -27,8 +61,7 @@ export function buildDeepSeekFIMParams(fileContent, fimBuilder, metadata = {}) {
  * @returns {Promise<string>} 补全文本
  */
 export async function callDeepSeekAPI(prompt, suffix, config, apiKey, baseUrl) {
-  // 构建 DeepSeek FIM API URL：将 /v1 替换为 /beta/completions
-  const completionsUrl = baseUrl.replace(DEEPSEEK_CONFIG.PATH_REPLACE, DEEPSEEK_CONFIG.COMPLETION_PATH) + '/completions'
+  const completionsUrl = buildDeepSeekCompletionsUrl(baseUrl)
 
   const requestBody = {
     model: config.model,

packages/plugins/script/src/ai-completion/adapters/index.js

@@ -3,7 +3,7 @@ import { createSmartPrompt } from '../builders/promptBuilder.js'
 import { FIMPromptBuilder } from '../builders/fimPromptBuilder.js'
 import { detectModelType, calculateTokens, getStopSequences } from '../utils/modelUtils.js'
 import { cleanCompletion, buildLowcodeMetadata } from '../utils/completionUtils.js'
-import { buildQwenMessages, callQwenAPI } from './qwenAdapter.js'
+import { buildQwenFIMPrompt, callQwenAPI } from './qwenAdapter.js'
 import { buildDeepSeekFIMParams, callDeepSeekAPI } from './deepseekAdapter.js'
 import { QWEN_CONFIG, DEEPSEEK_CONFIG, DEFAULTS, ERROR_MESSAGES, MODEL_CONFIG } from '../constants.js'
 
@@ -56,7 +56,6 @@ export function createCompletionHandler() {
         textAfterCursor,
         language,
         filename,
-        technologies: DEFAULTS.TECHNOLOGIES,
         lowcodeMetadata
       })
 
@@ -87,15 +86,15 @@ export function createCompletionHandler() {
 
       if (modelType === MODEL_CONFIG.QWEN.TYPE) {
         // ===== Qwen 流程 =====
-        const { messages, cursorContext: ctx } = buildQwenMessages(fileContent, qwenFimBuilder, fimMetadata)
+        const { prompt, cursorContext: ctx } = buildQwenFIMPrompt(fileContent, qwenFimBuilder, fimMetadata)
         cursorContext = ctx
 
         completionText = await callQwenAPI(
-          messages,
+          prompt,
           {
             model: completeModel,
             maxTokens: calculateTokens(ctx),
-            stopSequences: getStopSequences(ctx, modelType)
+            stopSequences: getStopSequences(ctx)
           },
           apiKey,
           baseUrl
@@ -115,7 +114,7 @@ export function createCompletionHandler() {
           {
             model: completeModel,
             maxTokens: calculateTokens(ctx),
-            stopSequences: getStopSequences(ctx, modelType)
+            stopSequences: getStopSequences(ctx)
           },
           apiKey,
           baseUrl
@@ -124,7 +123,7 @@ export function createCompletionHandler() {
 
       // 7. 处理补全结果
       if (completionText) {
-        completionText = cleanCompletion(completionText, modelType, cursorContext, textAfterCursor)
+        completionText = cleanCompletion(completionText, cursorContext, textAfterCursor)
 
         if (!completionText) {
           return {

packages/plugins/script/src/ai-completion/adapters/qwenAdapter.js

@@ -1,41 +1,36 @@
 import { QWEN_CONFIG, HTTP_CONFIG, ERROR_MESSAGES } from '../constants.js'
 
 /**
- * 构建 Qwen FIM 格式的 messages
+ * 构建 Qwen FIM prompt
  * @param {string} fileContent - 文件内容（包含 [CURSOR] 标记）
  * @param {Object} fimBuilder - FIM 构建器实例
- * @param {Object} metadata - 元数据（language, lowcodeMetadata 等）
- * @returns {{ messages: Array, cursorContext: Object }} Messages 和上下文
+ * @param {Object} metadata - 元数据
+ * @returns {{ prompt: string, cursorContext: Object }} Prompt 和上下文
  */
-export function buildQwenMessages(fileContent, fimBuilder, metadata = {}) {
+export function buildQwenFIMPrompt(fileContent, fimBuilder, metadata = {}) {
   const { fimPrompt, cursorContext } = fimBuilder.buildOptimizedFIMPrompt(fileContent, metadata)
 
   return {
-    messages: [
-      {
-        role: 'user',
-        content: fimPrompt
-      }
-    ],
+    prompt: fimPrompt,
     cursorContext
   }
 }
 
 /**
  * 调用 Qwen Completions API
- * @param {Array} messages - Messages 数组
+ * @param {string} prompt - FIM prompt
  * @param {Object} config - 配置对象
  * @param {string} apiKey - API 密钥
  * @param {string} baseUrl - 基础 URL
  * @returns {Promise<string>} 补全文本
  */
-export async function callQwenAPI(messages, config, apiKey, baseUrl) {
+export async function callQwenAPI(prompt, config, apiKey, baseUrl) {
   // 构建完整的 Completions API URL
   const completionsUrl = `${baseUrl}${QWEN_CONFIG.COMPLETION_PATH}`
 
   const requestBody = {
     model: config.model,
-    prompt: messages[0].content, // FIM prompt
+    prompt,
     max_tokens: config.maxTokens,
     temperature: QWEN_CONFIG.DEFAULT_TEMPERATURE,
     top_p: QWEN_CONFIG.TOP_P,

packages/plugins/script/src/ai-completion/builders/fimPromptBuilder.js

@@ -32,7 +32,7 @@ export class FIMPromptBuilder {
       return {
         prefix: fileContent,
         suffix: '',
-        cursorContext: { type: 'unknown', hasPrefix: true, hasSuffix: false }
+        cursorContext: { type: 'unknown' }
       }
     }
 
@@ -118,18 +118,14 @@ export class FIMPromptBuilder {
   /**
    * 分析光标上下文
    * @param {string} prefix - 前缀代码
-   * @param {string} suffix - 后缀代码
    * @returns {Object} 上下文信息
    */
-  analyzeCursorContext(prefix, suffix) {
+  analyzeCursorContext(prefix) {
     const context = {
       type: 'unknown',
-      hasPrefix: prefix.trim().length > 0,
-      hasSuffix: suffix.trim().length > 0,
       inFunction: false,
       inClass: false,
       inObject: false,
-      inArray: false,
       inBlockComment: false,
       inLineComment: false,
       needsExpression: false,