@ww_nero/mini-cli 1.0.80 → 1.0.82

package/src/utils/think.js

@@ -1,214 +1,214 @@
-const THINK_TAGS = [
-  { name: 'think', open: '<think>', close: '</think>' },
-  { name: 'tihinking', open: '<tihinking>', close: '</tihinking>' },
-  { name: 'thought', open: '<thought>', close: '</thought>' }
-];
-
-const THINK_TAG_MAP = new Map(
-  THINK_TAGS.map(tag => [tag.name.toLowerCase(), { ...tag, closeLower: tag.close.toLowerCase() }])
-);
-
-const findTagEnd = (text, startIndex) => {
-  let inSingleQuote = false;
-  let inDoubleQuote = false;
-
-  for (let i = startIndex; i < text.length; i++) {
-    const char = text[i];
-
-    if (char === '\'' && !inDoubleQuote) {
-      inSingleQuote = !inSingleQuote;
-      continue;
-    }
-
-    if (char === '"' && !inSingleQuote) {
-      inDoubleQuote = !inDoubleQuote;
-      continue;
-    }
-
-    if (char === '>' && !inSingleQuote && !inDoubleQuote) {
-      return i;
-    }
-  }
-
-  return -1;
-};
-
-const parseTagInfo = (tagText) => {
-  if (!tagText.startsWith('<') || !tagText.endsWith('>')) return null;
-
-  let inner = tagText.slice(1, -1).trim();
-  if (!inner) return null;
-
-  let type = 'start';
-  if (inner.startsWith('/')) {
-    type = 'end';
-    inner = inner.slice(1).trim();
-  } else if (inner.endsWith('/')) {
-    type = 'self';
-    inner = inner.slice(0, -1).trim();
-  }
-
-  const nameMatch = inner.match(/^([a-zA-Z0-9_-]+)/);
-  if (!nameMatch) return null;
-
-  return {
-    type,
-    name: nameMatch[1].toLowerCase()
-  };
-};
-
-const createThinkParserState = () => ({
-  buffer: '',
-  currentThink: null
-});
-
-const processThinkChunk = (state, chunk = '') => {
-  if (!state) {
-    throw new Error('processThinkChunk requires a state object.');
-  }
-
-  if (chunk == null) {
-    chunk = '';
-  }
-
-  if (typeof chunk !== 'string') {
-    chunk = String(chunk);
-  }
-
-  if (!state.buffer) {
-    state.buffer = '';
-  }
-
-  state.buffer += chunk;
-
-  let contentChunk = '';
-  let reasonChunk = '';
-
-  const consumePrefix = (length) => {
-    const prefix = state.buffer.slice(0, length);
-    state.buffer = state.buffer.slice(length);
-    return prefix;
-  };
-
-  while (state.buffer.length > 0) {
-    if (state.currentThink) {
-      const thinkInfo = THINK_TAG_MAP.get(state.currentThink);
-      if (!thinkInfo) {
-        state.currentThink = null;
-        continue;
-      }
-
-      const bufferLower = state.buffer.toLowerCase();
-      const closeIndex = bufferLower.indexOf(thinkInfo.closeLower);
-
-      if (closeIndex === -1) {
-        reasonChunk += consumePrefix(state.buffer.length);
-        return { contentChunk, reasonChunk };
-      }
-
-      reasonChunk += consumePrefix(closeIndex);
-      consumePrefix(thinkInfo.close.length);
-      state.currentThink = null;
-      continue;
-    }
-
-    const nextLt = state.buffer.indexOf('<');
-
-    if (nextLt === -1) {
-      contentChunk += consumePrefix(state.buffer.length);
-      break;
-    }
-
-    if (nextLt > 0) {
-      contentChunk += consumePrefix(nextLt);
-      continue;
-    }
-
-    const tagEnd = findTagEnd(state.buffer, 1);
-    if (tagEnd === -1) {
-      break;
-    }
-
-    const rawTag = consumePrefix(tagEnd + 1);
-    const tagInfo = parseTagInfo(rawTag);
-
-    if (!tagInfo) {
-      contentChunk += rawTag;
-      continue;
-    }
-
-    if (tagInfo.type === 'start') {
-      if (THINK_TAG_MAP.has(tagInfo.name)) {
-        state.currentThink = tagInfo.name;
-        continue;
-      }
-
-      contentChunk += rawTag;
-      continue;
-    }
-
-    if (tagInfo.type === 'end') {
-      contentChunk += rawTag;
-      continue;
-    }
-
-    contentChunk += rawTag;
-  }
-
-  return { contentChunk, reasonChunk };
-};
-
-const flushThinkState = (state) => {
-  if (!state || !state.buffer) {
-    return { contentChunk: '', reasonChunk: '' };
-  }
-
-  const remaining = state.buffer;
-  state.buffer = '';
-
-  if (state.currentThink) {
-    state.currentThink = null;
-    return { contentChunk: '', reasonChunk: remaining };
-  }
-
-  return { contentChunk: remaining, reasonChunk: '' };
-};
-
-const splitThinkContent = (text = '') => {
-  if (!text) {
-    return { content: '', reasoning: '' };
-  }
-
-  const state = createThinkParserState();
-  let content = '';
-  let reasoning = '';
-
-  const result = processThinkChunk(state, text);
-  content += result.contentChunk;
-  reasoning += result.reasonChunk;
-
-  const tail = flushThinkState(state);
-  content += tail.contentChunk;
-  reasoning += tail.reasonChunk;
-
-  return { content, reasoning };
-};
-
-const summarizeReasoning = (text = '', maxLength = 100) => {
-  if (!text) return '';
-  const normalized = text.replace(/\s+/g, ' ').trim();
-  if (!normalized) return '';
-  if (normalized.length > maxLength) {
-    return normalized.substring(0, maxLength) + '...';
-  }
-  return normalized;
-};
-
-module.exports = {
-  THINK_TAGS,
-  createThinkParserState,
-  processThinkChunk,
-  flushThinkState,
-  splitThinkContent,
-  summarizeReasoning
-};
+const THINK_TAGS = [
+  { name: 'think', open: '<think>', close: '</think>' },
+  { name: 'tihinking', open: '<tihinking>', close: '</tihinking>' },
+  { name: 'thought', open: '<thought>', close: '</thought>' }
+];
+
+const THINK_TAG_MAP = new Map(
+  THINK_TAGS.map(tag => [tag.name.toLowerCase(), { ...tag, closeLower: tag.close.toLowerCase() }])
+);
+
+const findTagEnd = (text, startIndex) => {
+  let inSingleQuote = false;
+  let inDoubleQuote = false;
+
+  for (let i = startIndex; i < text.length; i++) {
+    const char = text[i];
+
+    if (char === '\'' && !inDoubleQuote) {
+      inSingleQuote = !inSingleQuote;
+      continue;
+    }
+
+    if (char === '"' && !inSingleQuote) {
+      inDoubleQuote = !inDoubleQuote;
+      continue;
+    }
+
+    if (char === '>' && !inSingleQuote && !inDoubleQuote) {
+      return i;
+    }
+  }
+
+  return -1;
+};
+
+const parseTagInfo = (tagText) => {
+  if (!tagText.startsWith('<') || !tagText.endsWith('>')) return null;
+
+  let inner = tagText.slice(1, -1).trim();
+  if (!inner) return null;
+
+  let type = 'start';
+  if (inner.startsWith('/')) {
+    type = 'end';
+    inner = inner.slice(1).trim();
+  } else if (inner.endsWith('/')) {
+    type = 'self';
+    inner = inner.slice(0, -1).trim();
+  }
+
+  const nameMatch = inner.match(/^([a-zA-Z0-9_-]+)/);
+  if (!nameMatch) return null;
+
+  return {
+    type,
+    name: nameMatch[1].toLowerCase()
+  };
+};
+
+const createThinkParserState = () => ({
+  buffer: '',
+  currentThink: null
+});
+
+const processThinkChunk = (state, chunk = '') => {
+  if (!state) {
+    throw new Error('processThinkChunk requires a state object.');
+  }
+
+  if (chunk == null) {
+    chunk = '';
+  }
+
+  if (typeof chunk !== 'string') {
+    chunk = String(chunk);
+  }
+
+  if (!state.buffer) {
+    state.buffer = '';
+  }
+
+  state.buffer += chunk;
+
+  let contentChunk = '';
+  let reasonChunk = '';
+
+  const consumePrefix = (length) => {
+    const prefix = state.buffer.slice(0, length);
+    state.buffer = state.buffer.slice(length);
+    return prefix;
+  };
+
+  while (state.buffer.length > 0) {
+    if (state.currentThink) {
+      const thinkInfo = THINK_TAG_MAP.get(state.currentThink);
+      if (!thinkInfo) {
+        state.currentThink = null;
+        continue;
+      }
+
+      const bufferLower = state.buffer.toLowerCase();
+      const closeIndex = bufferLower.indexOf(thinkInfo.closeLower);
+
+      if (closeIndex === -1) {
+        reasonChunk += consumePrefix(state.buffer.length);
+        return { contentChunk, reasonChunk };
+      }
+
+      reasonChunk += consumePrefix(closeIndex);
+      consumePrefix(thinkInfo.close.length);
+      state.currentThink = null;
+      continue;
+    }
+
+    const nextLt = state.buffer.indexOf('<');
+
+    if (nextLt === -1) {
+      contentChunk += consumePrefix(state.buffer.length);
+      break;
+    }
+
+    if (nextLt > 0) {
+      contentChunk += consumePrefix(nextLt);
+      continue;
+    }
+
+    const tagEnd = findTagEnd(state.buffer, 1);
+    if (tagEnd === -1) {
+      break;
+    }
+
+    const rawTag = consumePrefix(tagEnd + 1);
+    const tagInfo = parseTagInfo(rawTag);
+
+    if (!tagInfo) {
+      contentChunk += rawTag;
+      continue;
+    }
+
+    if (tagInfo.type === 'start') {
+      if (THINK_TAG_MAP.has(tagInfo.name)) {
+        state.currentThink = tagInfo.name;
+        continue;
+      }
+
+      contentChunk += rawTag;
+      continue;
+    }
+
+    if (tagInfo.type === 'end') {
+      contentChunk += rawTag;
+      continue;
+    }
+
+    contentChunk += rawTag;
+  }
+
+  return { contentChunk, reasonChunk };
+};
+
+const flushThinkState = (state) => {
+  if (!state || !state.buffer) {
+    return { contentChunk: '', reasonChunk: '' };
+  }
+
+  const remaining = state.buffer;
+  state.buffer = '';
+
+  if (state.currentThink) {
+    state.currentThink = null;
+    return { contentChunk: '', reasonChunk: remaining };
+  }
+
+  return { contentChunk: remaining, reasonChunk: '' };
+};
+
+const splitThinkContent = (text = '') => {
+  if (!text) {
+    return { content: '', reasoning: '' };
+  }
+
+  const state = createThinkParserState();
+  let content = '';
+  let reasoning = '';
+
+  const result = processThinkChunk(state, text);
+  content += result.contentChunk;
+  reasoning += result.reasonChunk;
+
+  const tail = flushThinkState(state);
+  content += tail.contentChunk;
+  reasoning += tail.reasonChunk;
+
+  return { content, reasoning };
+};
+
+const summarizeReasoning = (text = '', maxLength = 100) => {
+  if (!text) return '';
+  const normalized = text.replace(/\s+/g, ' ').trim();
+  if (!normalized) return '';
+  if (normalized.length > maxLength) {
+    return normalized.substring(0, maxLength) + '...';
+  }
+  return normalized;
+};
+
+module.exports = {
+  THINK_TAGS,
+  createThinkParserState,
+  processThinkChunk,
+  flushThinkState,
+  splitThinkContent,
+  summarizeReasoning
+};