refactor(ai-completion): Restructure stop sequences and context handling

Author: SonyLeo

packages/plugins/script/src/ai-completion/constants.js

@@ -117,54 +117,28 @@ export const MODEL_COMMON_CONFIG = {
   }
 }
 
-/**
- * 通用停止符配置（JS/TS）
- */
-export const STOP_SEQUENCES = [
-  // 通用停止符
-  '\n\n',
-  '```',
-
-  // JS/TS 语言特性
-  '\nfunction ',
-  '\nclass ',
-  '\nconst ',
-  '\nlet ',
-  '\nvar ',
-  '\nexport ',
-  '\nimport ',
-  '\ninterface ',
-  '\ntype ',
-  '\nenum ',
-
-  // 注释边界
-  '\n//',
-  '\n/*',
-
-  // 代码块边界
-  '\n}',
-  '\n};'
-]
+// 停止符配置（API 限制：最多 16 个）
+export const STOP_SEQUENCES = {
+  CORE: ['\n\n', '```'],
+  NEW_SCOPE: ['\nfunction ', '\nclass ', '\nexport ', '\nimport '],
+  BLOCK_END: ['\n}', '\n};']
+}
 
-/**
- * FIM (Fill-In-the-Middle) 配置
- */
+// 上下文特定停止符
+export const CONTEXT_STOP_SEQUENCES = {
+  EXPRESSION: [';', ',', '\n)'],
+  COMMENT: ['\n\n', '*/'],
+  OBJECT: ['\n}', '\n};'],
+  FUNCTION: ['\n}', '\nfunction ', '\nreturn ']
+}
+
+// FIM 标记配置
 export const FIM_CONFIG = {
   MARKERS: {
     PREFIX: '<|fim_prefix|>',
     SUFFIX: '<|fim_suffix|>',
     MIDDLE: '<|fim_middle|>',
     CURSOR: '[CURSOR]'
-  },
-
-  // FIM 专用停止符（会与 STOP_SEQUENCES 合并）
-  FIM_MARKERS_STOPS: ['<|fim_prefix|>', '<|fim_suffix|>', '<|fim_middle|>'],
-
-  // 上下文特定的额外停止符
-  CONTEXT_STOPS: {
-    EXPRESSION: [';', '\n)', ','],
-    STATEMENT: [], // 使用通用停止符即可
-    OBJECT: [] // 使用通用停止符即可
   }
 }
 

packages/plugins/script/src/ai-completion/utils/modelUtils.js

@@ -1,4 +1,4 @@
-import { MODEL_CONFIG, MODEL_COMMON_CONFIG, STOP_SEQUENCES, FIM_CONFIG } from '../constants.js'
+import { MODEL_CONFIG, MODEL_COMMON_CONFIG, STOP_SEQUENCES, CONTEXT_STOP_SEQUENCES } from '../constants.js'
 
 /**
  * 检测模型类型
@@ -46,33 +46,41 @@ export function calculateTokens(cursorContext) {
   return limits.DEFAULT
 }
 
-/**
- * 获取动态停止符
- * @param {Object} cursorContext - 光标上下文
- * @param {string} modelType - 模型类型
- * @returns {string[]} 停止符数组
- */
-export function getStopSequences(cursorContext, modelType) {
-  // 基础停止符：通用停止符
-  const stops = [...STOP_SEQUENCES]
+// 获取动态停止符（最多 16 个）
+export function getStopSequences(cursorContext, _modelType) {
+  const stops = []
 
-  // Qwen 模型添加 FIM 标记
-  if (modelType === 'qwen') {
-    stops.push(...FIM_CONFIG.FIM_MARKERS_STOPS)
-  }
+  // 核心停止符
+  stops.push(...STOP_SEQUENCES.CORE)
 
-  if (!cursorContext) {
-    return stops
+  if (cursorContext) {
+    if (cursorContext.inBlockComment || cursorContext.inLineComment) {
+      stops.push(...CONTEXT_STOP_SEQUENCES.COMMENT)
+    } else if (cursorContext.needsExpression) {
+      stops.push(...CONTEXT_STOP_SEQUENCES.EXPRESSION)
+      stops.push(...STOP_SEQUENCES.NEW_SCOPE)
+    } else if (cursorContext.inObject) {
+      stops.push(...CONTEXT_STOP_SEQUENCES.OBJECT)
+      stops.push(...STOP_SEQUENCES.NEW_SCOPE)
+    } else if (cursorContext.inFunction) {
+      stops.push(...CONTEXT_STOP_SEQUENCES.FUNCTION)
+      stops.push(...STOP_SEQUENCES.BLOCK_END)
+    } else {
+      stops.push(...STOP_SEQUENCES.NEW_SCOPE)
+      stops.push(...STOP_SEQUENCES.BLOCK_END)
+    }
+  } else {
+    stops.push(...STOP_SEQUENCES.NEW_SCOPE)
+    stops.push(...STOP_SEQUENCES.BLOCK_END)
   }
 
-  // 根据上下文添加特定停止符
-  if (cursorContext.needsExpression) {
-    stops.push(...FIM_CONFIG.CONTEXT_STOPS.EXPRESSION)
-  } else if (cursorContext.needsStatement) {
-    stops.push(...FIM_CONFIG.CONTEXT_STOPS.STATEMENT)
-  } else if (cursorContext.inObject) {
-    stops.push(...FIM_CONFIG.CONTEXT_STOPS.OBJECT)
+  const uniqueStops = [...new Set(stops)]
+
+  if (uniqueStops.length > 16) {
+    // eslint-disable-next-line no-console
+    console.warn(`⚠️ 停止符超过限制: ${uniqueStops.length}，截断到 16 个`)
+    return uniqueStops.slice(0, 16)
   }
 
-  return stops
+  return uniqueStops
 }