@ww_nero/mini-cli 1.0.81 → 1.0.83

package/src/request.js

@@ -1,328 +1,345 @@
-const fetch = require('node-fetch');
-const { createParser } = require('eventsource-parser');
-const { COMPACT_SUMMARY_PROMPT } = require('./config');
-
-const joinUrl = (base, pathname) => {
-  const normalizedBase = String(base || '').replace(/\/$/, '');
-  const normalizedPath = pathname.startsWith('/') ? pathname : `/${pathname}`;
-  return `${normalizedBase}${normalizedPath}`;
-};
-
-const createAbortError = () => {
-  if (typeof DOMException === 'function') {
-    return new DOMException('Aborted', 'AbortError');
-  }
-  const error = new Error('Aborted');
-  error.name = 'AbortError';
-  return error;
-};
-
-const createFetchOptions = (endpoint, body, abortController) => ({
-  method: 'POST',
-  headers: {
-    Authorization: `Bearer ${endpoint.key}`,
-    'Content-Type': 'application/json'
-  },
-  body: JSON.stringify(body),
-  signal: abortController.signal
-});
-
-const makeRequestWithRetry = async (endpoint, requestBody, abortController, options = {}) => {
-  const { maxRetries = 5, retryDelay = 3000, onRetry = null } = options;
-  const url = joinUrl(endpoint.baseUrl, '/chat/completions');
-
-  const attemptRequest = async () => {
-    try {
-      const response = await fetch(url, createFetchOptions(endpoint, requestBody, abortController));
-      if (!response.ok) {
-        throw new Error(`HTTP ${response.status}`);
-      }
-      return { response, error: null };
-    } catch (error) {
-      if (abortController.signal.aborted) {
-        throw createAbortError();
-      }
-      return { response: null, error };
-    }
-  };
-
-  let attempt = 0;
-  let result = await attemptRequest();
-
-  while (!result.response && attempt < maxRetries) {
-    attempt += 1;
-    if (typeof onRetry === 'function') {
-      onRetry(attempt, result.error);
-    }
-    await new Promise(resolve => {
-      const handleAbort = () => {
-        abortController.signal.removeEventListener('abort', handleAbort);
-        clearTimeout(timeout);
-        resolve();
-      };
-      const timeout = setTimeout(() => {
-        abortController.signal.removeEventListener('abort', handleAbort);
-        resolve();
-      }, retryDelay);
-      abortController.signal.addEventListener('abort', handleAbort);
-    });
-    if (abortController.signal.aborted) {
-      throw createAbortError();
-    }
-    result = await attemptRequest();
-  }
-
-  if (!result.response) {
-    throw result.error;
-  }
-
-  return result.response;
-};
-
-const processStreamResponse = (response, options = {}) => {
-  const {
-    onContent = null,
-    onReasoningContent = null,
-    onComplete = null,
-    abortController = null,
-    includeToolCalls = false
-  } = options;
-
-  return new Promise((resolve, reject) => {
-    let abortHandler = null;
-    const cleanupAbortListener = () => {
-      if (abortController && abortHandler) {
-        abortController.signal.removeEventListener('abort', abortHandler);
-        abortHandler = null;
-      }
-    };
-
-    const rejectWithAbort = () => {
-      cleanupAbortListener();
-      if (response.body && typeof response.body.destroy === 'function') {
-        response.body.destroy();
-      }
-      reject(createAbortError());
-    };
-
-    const checkAbort = () => {
-      if (abortController && abortController.signal.aborted) {
-        rejectWithAbort();
-        return true;
-      }
-      return false;
-    };
-
-    try {
-      let fullContent = '';
-      let toolCalls = [];
-      let currentMessage = null;
-      let fullReasoningContent = '';
-      let usage = null;
-      const decoder = new TextDecoder('utf-8', { fatal: false });
-
-      const handleEvent = (event) => {
-        if (checkAbort()) return;
-        if (!event || event.type !== 'event') {
-          return;
-        }
-        if (!event.data || event.data === '[DONE]') {
-          return;
-        }
-        try {
-          const payload = JSON.parse(event.data);
-          if (payload.usage && typeof payload.usage === 'object') {
-            usage = payload.usage;
-          }
-          const choice = payload.choices && payload.choices[0];
-          if (!choice || !choice.delta) return;
-          const delta = choice.delta;
-          const contentChunk = delta.content || '';
-          if (contentChunk) {
-            fullContent += contentChunk;
-            if (onContent) {
-              onContent(contentChunk, fullContent);
-            }
-          }
-
-          const reasoningChunk = delta.reasoning_content || '';
-          if (reasoningChunk) {
-            fullReasoningContent += reasoningChunk;
-            if (onReasoningContent) {
-              onReasoningContent(reasoningChunk, fullReasoningContent);
-            }
-          }
-
-          if (includeToolCalls && Array.isArray(delta.tool_calls)) {
-            delta.tool_calls.forEach((call, index) => {
-              if (!toolCalls[index]) {
-                toolCalls[index] = {
-                  id: call.id || '',
-                  type: call.type || 'function',
-                  function: {
-                    name: (call.function && call.function.name) || '',
-                    arguments: ''
-                  }
-                };
-              }
-
-              if (call.function && typeof call.function.arguments === 'string') {
-                toolCalls[index].function.arguments += call.function.arguments;
-              }
-            });
-          }
-
-          if (includeToolCalls) {
-            currentMessage = {
-              role: 'assistant',
-              content: fullContent || null,
-              ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),
-              ...(fullReasoningContent ? { reasoning_content: fullReasoningContent } : {})
-            };
-          }
-        } catch (error) {
-          console.warn('SSE parse error:', error.message);
-        }
-      };
-
-      const parser = createParser(handleEvent);
-
-      const safeFeed = (text) => {
-        try {
-          parser.feed(text);
-        } catch (error) {
-          console.warn('SSE error:', error.message);
-        }
-      };
-
-      response.body.on('data', chunk => {
-        if (checkAbort()) return;
-        const text = decoder.decode(chunk, { stream: true });
-        safeFeed(text);
-      });
-
-      response.body.on('end', () => {
-        if (checkAbort()) return;
-        cleanupAbortListener();
-        const rest = decoder.decode();
-        if (rest) {
-          safeFeed(rest);
-        }
-        onComplete && onComplete();
-        if (includeToolCalls) {
-          const filteredToolCalls = toolCalls.filter(Boolean);
-          const assistantMessage = currentMessage || {
-            role: 'assistant',
-            content: fullContent || null,
-            ...(filteredToolCalls.length > 0 ? { tool_calls: filteredToolCalls } : {})
-          };
-
-          if (fullReasoningContent && !assistantMessage.reasoning_content) {
-            assistantMessage.reasoning_content = fullReasoningContent;
-          }
-
-          resolve({
-            content: fullContent,
-            toolCalls: filteredToolCalls,
-            message: assistantMessage,
-            reasoningContent: fullReasoningContent,
-            usage
-          });
-        } else {
-          resolve({
-            content: fullContent,
-            reasoningContent: fullReasoningContent,
-            usage
-          });
-        }
-      });
-
-      response.body.on('error', (error) => {
-        cleanupAbortListener();
-        reject(error);
-      });
-
-      if (abortController) {
-        abortHandler = () => {
-          rejectWithAbort();
-        };
-        abortController.signal.addEventListener('abort', abortHandler);
-      }
-    } catch (error) {
-      reject(error);
-    }
-  });
-};
-
-const buildConversationForSummary = (messages) => {
-  const parts = [];
-  for (const msg of messages) {
-    if (msg.role === 'system') continue;
-
-    const role = msg.role === 'user' ? '用户' : msg.role === 'assistant' ? '助手' : '工具';
-    let content = msg.content || '';
-
-    if (msg.tool_call_id) {
-      parts.push(`[工具返回 ${msg.tool_call_id}]: ${content}`);
-    } else if (msg.tool_calls && msg.tool_calls.length > 0) {
-      const toolNames = msg.tool_calls.map(tc => tc.function?.name || '未知工具').join(', ');
-      parts.push(`[${role}]: ${content || ''}\n[调用工具: ${toolNames}]`);
-    } else {
-      parts.push(`[${role}]: ${content}`);
-    }
-  }
-  return parts.join('\n\n');
-};
-
-const performCompactSummary = async (endpoint, messages, abortController) => {
-  const conversationText = buildConversationForSummary(messages);
-
-  const summaryMessages = [
-    {
-      role: 'system',
-      content: COMPACT_SUMMARY_PROMPT
-    },
-    {
-      role: 'user',
-      content: `以下是需要总结的对话内容：\n\n${conversationText}`
-    }
-  ];
-
-  const requestBody = {
-    stream: false,
-    messages: summaryMessages,
-    model: endpoint.model
-  };
-
-  const url = joinUrl(endpoint.baseUrl, '/chat/completions');
-
-  try {
-    const response = await fetch(url, createFetchOptions(endpoint, requestBody, abortController));
-
-    if (!response.ok) {
-      throw new Error(`HTTP error! status: ${response.status}`);
-    }
-
-    const data = await response.json();
-    const choice = data.choices && data.choices[0];
-
-    if (choice && choice.message && choice.message.content) {
-      return choice.message.content;
-    }
-
-    throw new Error('无法获取总结内容');
-  } catch (error) {
-    if (error.name === 'AbortError') {
-      throw error;
-    }
-    console.error('Compact 总结请求失败:', error);
-    throw error;
-  }
-};
-
-module.exports = {
-  makeRequestWithRetry,
-  processStreamResponse,
-  buildConversationForSummary,
-  performCompactSummary
-};
+const fetch = require('node-fetch');
+const { createParser } = require('eventsource-parser');
+const { COMPACT_SUMMARY_PROMPT } = require('./config');
+
+const joinUrl = (base, pathname) => {
+  const normalizedBase = String(base || '').replace(/\/$/, '');
+  const normalizedPath = pathname.startsWith('/') ? pathname : `/${pathname}`;
+  return `${normalizedBase}${normalizedPath}`;
+};
+
+const createAbortError = () => {
+  if (typeof DOMException === 'function') {
+    return new DOMException('Aborted', 'AbortError');
+  }
+  const error = new Error('Aborted');
+  error.name = 'AbortError';
+  return error;
+};
+
+const createFetchOptions = (endpoint, body, abortController) => ({
+  method: 'POST',
+  headers: {
+    Authorization: `Bearer ${endpoint.key}`,
+    'Content-Type': 'application/json'
+  },
+  body: JSON.stringify(body),
+  signal: abortController.signal
+});
+
+const makeRequestWithRetry = async (endpoint, requestBody, abortController, options = {}) => {
+  const { maxRetries = 5, retryDelay = 3000, onRetry = null } = options;
+  const url = joinUrl(endpoint.baseUrl, '/chat/completions');
+
+  const attemptRequest = async () => {
+    try {
+      const response = await fetch(url, createFetchOptions(endpoint, requestBody, abortController));
+      if (!response.ok) {
+        let detail = '';
+        try {
+          const body = await response.text();
+          detail = body ? `: ${body}` : '';
+        } catch (_) {}
+        throw new Error(`HTTP ${response.status}${detail}`);
+      }
+      return { response, error: null };
+    } catch (error) {
+      if (abortController.signal.aborted) {
+        throw createAbortError();
+      }
+      return { response: null, error };
+    }
+  };
+
+  let attempt = 0;
+  let result = await attemptRequest();
+
+  while (!result.response && attempt < maxRetries) {
+    attempt += 1;
+    if (typeof onRetry === 'function') {
+      onRetry(attempt, result.error);
+    }
+    await new Promise(resolve => {
+      const handleAbort = () => {
+        abortController.signal.removeEventListener('abort', handleAbort);
+        clearTimeout(timeout);
+        resolve();
+      };
+      const timeout = setTimeout(() => {
+        abortController.signal.removeEventListener('abort', handleAbort);
+        resolve();
+      }, retryDelay);
+      abortController.signal.addEventListener('abort', handleAbort);
+    });
+    if (abortController.signal.aborted) {
+      throw createAbortError();
+    }
+    result = await attemptRequest();
+  }
+
+  if (!result.response) {
+    throw result.error;
+  }
+
+  return result.response;
+};
+
+const processStreamResponse = (response, options = {}) => {
+  const {
+    onContent = null,
+    onReasoningContent = null,
+    onComplete = null,
+    abortController = null,
+    includeToolCalls = false
+  } = options;
+
+  return new Promise((resolve, reject) => {
+    let abortHandler = null;
+    const cleanupAbortListener = () => {
+      if (abortController && abortHandler) {
+        abortController.signal.removeEventListener('abort', abortHandler);
+        abortHandler = null;
+      }
+    };
+
+    const rejectWithAbort = () => {
+      cleanupAbortListener();
+      if (response.body && typeof response.body.destroy === 'function') {
+        response.body.destroy();
+      }
+      reject(createAbortError());
+    };
+
+    const checkAbort = () => {
+      if (abortController && abortController.signal.aborted) {
+        rejectWithAbort();
+        return true;
+      }
+      return false;
+    };
+
+    try {
+      let fullContent = '';
+      let toolCalls = [];
+      let currentMessage = null;
+      let fullReasoningContent = '';
+      let usage = null;
+      const decoder = new TextDecoder('utf-8', { fatal: false });
+
+      const handleEvent = (event) => {
+        if (checkAbort()) return;
+        if (!event || event.type !== 'event') {
+          return;
+        }
+        if (!event.data || event.data === '[DONE]') {
+          return;
+        }
+        try {
+          const payload = JSON.parse(event.data);
+          if (payload.usage && typeof payload.usage === 'object') {
+            usage = payload.usage;
+          }
+          const choice = payload.choices && payload.choices[0];
+          if (!choice || !choice.delta) return;
+          const delta = choice.delta;
+          const contentChunk = delta.content || '';
+          if (contentChunk) {
+            fullContent += contentChunk;
+            if (onContent) {
+              onContent(contentChunk, fullContent);
+            }
+          }
+
+          const reasoningChunk = delta.reasoning_content || '';
+          if (reasoningChunk) {
+            fullReasoningContent += reasoningChunk;
+            if (onReasoningContent) {
+              onReasoningContent(reasoningChunk, fullReasoningContent);
+            }
+          }
+
+          if (includeToolCalls && Array.isArray(delta.tool_calls)) {
+            delta.tool_calls.forEach((call) => {
+              const idx = typeof call.index === 'number' ? call.index : toolCalls.length;
+              if (!toolCalls[idx]) {
+                toolCalls[idx] = {
+                  id: call.id || '',
+                  type: call.type || 'function',
+                  function: {
+                    name: '',
+                    arguments: ''
+                  }
+                };
+              }
+
+              if (call.id) {
+                toolCalls[idx].id = call.id;
+              }
+              if (call.function && call.function.name) {
+                toolCalls[idx].function.name += call.function.name;
+              }
+              if (call.function && typeof call.function.arguments === 'string') {
+                toolCalls[idx].function.arguments += call.function.arguments;
+              }
+            });
+          }
+
+          if (includeToolCalls) {
+            currentMessage = {
+              role: 'assistant',
+              content: fullContent || null,
+              ...(toolCalls.length > 0 ? { tool_calls: toolCalls } : {}),
+              ...(fullReasoningContent ? { reasoning_content: fullReasoningContent } : {})
+            };
+          }
+        } catch (error) {
+          console.warn('SSE parse error:', error.message);
+        }
+      };
+
+      const parser = createParser(handleEvent);
+
+      const safeFeed = (text) => {
+        try {
+          parser.feed(text);
+        } catch (error) {
+          console.warn('SSE error:', error.message);
+        }
+      };
+
+      response.body.on('data', chunk => {
+        if (checkAbort()) return;
+        const text = decoder.decode(chunk, { stream: true });
+        safeFeed(text);
+      });
+
+      response.body.on('end', () => {
+        if (checkAbort()) return;
+        cleanupAbortListener();
+        const rest = decoder.decode();
+        if (rest) {
+          safeFeed(rest);
+        }
+        onComplete && onComplete();
+        if (includeToolCalls) {
+          const filteredToolCalls = toolCalls.filter(Boolean);
+          const assistantMessage = currentMessage || {
+            role: 'assistant',
+            content: fullContent || null,
+            ...(filteredToolCalls.length > 0 ? { tool_calls: filteredToolCalls } : {})
+          };
+
+          if (fullReasoningContent && !assistantMessage.reasoning_content) {
+            assistantMessage.reasoning_content = fullReasoningContent;
+          }
+
+          resolve({
+            content: fullContent,
+            toolCalls: filteredToolCalls,
+            message: assistantMessage,
+            reasoningContent: fullReasoningContent,
+            usage
+          });
+        } else {
+          resolve({
+            content: fullContent,
+            reasoningContent: fullReasoningContent,
+            usage
+          });
+        }
+      });
+
+      response.body.on('error', (error) => {
+        cleanupAbortListener();
+        reject(error);
+      });
+
+      if (abortController) {
+        abortHandler = () => {
+          rejectWithAbort();
+        };
+        abortController.signal.addEventListener('abort', abortHandler);
+      }
+    } catch (error) {
+      reject(error);
+    }
+  });
+};
+
+const buildConversationForSummary = (messages) => {
+  const parts = [];
+  for (const msg of messages) {
+    if (msg.role === 'system') continue;
+
+    const role = msg.role === 'user' ? '用户' : msg.role === 'assistant' ? '助手' : '工具';
+    let content = msg.content || '';
+
+    if (msg.tool_call_id) {
+      parts.push(`[工具返回 ${msg.tool_call_id}]: ${content}`);
+    } else if (msg.tool_calls && msg.tool_calls.length > 0) {
+      const toolNames = msg.tool_calls.map(tc => tc.function?.name || '未知工具').join(', ');
+      parts.push(`[${role}]: ${content || ''}\n[调用工具: ${toolNames}]`);
+    } else {
+      parts.push(`[${role}]: ${content}`);
+    }
+  }
+  return parts.join('\n\n');
+};
+
+const performCompactSummary = async (endpoint, messages, abortController) => {
+  const conversationText = buildConversationForSummary(messages);
+
+  const summaryMessages = [
+    {
+      role: 'system',
+      content: COMPACT_SUMMARY_PROMPT
+    },
+    {
+      role: 'user',
+      content: `以下是需要总结的对话内容：\n\n${conversationText}`
+    }
+  ];
+
+  const requestBody = {
+    stream: false,
+    messages: summaryMessages,
+    model: endpoint.model
+  };
+
+  const url = joinUrl(endpoint.baseUrl, '/chat/completions');
+
+  try {
+    const response = await fetch(url, createFetchOptions(endpoint, requestBody, abortController));
+
+    if (!response.ok) {
+      let detail = '';
+      try {
+        const body = await response.text();
+        detail = body ? `: ${body}` : '';
+      } catch (_) {}
+      throw new Error(`HTTP ${response.status}${detail}`);
+    }
+
+    const data = await response.json();
+    const choice = data.choices && data.choices[0];
+
+    if (choice && choice.message && choice.message.content) {
+      return choice.message.content;
+    }
+
+    throw new Error('无法获取总结内容');
+  } catch (error) {
+    if (error.name === 'AbortError') {
+      throw error;
+    }
+    console.error('Compact 总结请求失败:', error);
+    throw error;
+  }
+};
+
+module.exports = {
+  makeRequestWithRetry,
+  processStreamResponse,
+  buildConversationForSummary,
+  performCompactSummary
+};