foliko 1.1.34 → 1.1.35

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "foliko",
-  "version": "1.1.34",
+  "version": "1.1.35",
   "description": "简约的插件化 Agent 框架",
   "main": "src/index.js",
   "type": "commonjs",

package/src/core/agent-chat.js

@@ -22,6 +22,7 @@ const {
 const { prepareMessagesForAPI, cleanResponse } = require('../utils');
 const { ChatQueueManager } = require('../utils/chat-queue');
 const { countTokens } = require('@anthropic-ai/tokenizer');
+const { isThinkingModel } = require('./provider');
 const fs = require('fs/promises');
 // 新模块
 const { ChatSession } = require('./chat-session');
@@ -46,7 +47,18 @@ class AgentChatHandler extends EventEmitter {
     this.baseURL = config.baseURL;
     this.providerOptions = config.providerOptions || {};
     this.providerOptions.maxOutputTokens = config.providerOptions?.maxOutputTokens || 8192;
-    this.providerOptions.temperature = config.providerOptions?.temperature || 0.3;
+    // DeepSeek thinking mode 模型不支持 temperature 参数
+    if (isThinkingModel(this.model)) {
+      this.providerOptions.temperature = undefined;
+      // DeepSeek 需要通过 extra_body 启用 thinking mode
+      // AI SDK 会将 providerOptions.deepseek 映射到请求体的 extra_body
+      if (!this.providerOptions.deepseek) {
+        this.providerOptions.deepseek = {};
+      }
+      this.providerOptions.deepseek.thinking = { type: 'enabled' };
+    } else {
+      this.providerOptions.temperature = config.providerOptions?.temperature || 0.3;
+    }
 
     this._systemPrompt = config.systemPrompt || 'You are a helpful assistant.';
     this._maxSteps = config.maxSteps || 20;
@@ -452,6 +464,10 @@ class AgentChatHandler extends EventEmitter {
         logger.debug(`[AgentChat] SDK onError: ${error?.message}`);
       };
 
+      // DeepSeek thinking mode: 收集 reasoning_content
+      const isThinking = isThinkingModel(this.model);
+      let reasoningContent = '';
+
       const result = await framework.runInSession(sessionId, {}, async () => {
         return agent.stream({ messages, ...this.providerOptions, onError: handleSDKError });
       });
@@ -466,6 +482,7 @@ class AgentChatHandler extends EventEmitter {
           fullText += text;
           yield { type: 'text', text };
         } else if (part.type === 'reasoning') {
+          reasoningContent += part.text || '';
           yield { type: 'thinking', text: part.text };
         } else if (part.type === 'tool-call') {
           yield { type: 'tool-call', toolName: part.toolName, input: part.input };
@@ -478,7 +495,24 @@ class AgentChatHandler extends EventEmitter {
         }
       }
 
+      // DeepSeek thinking mode: 标记 reasoning_content 已处理
+      // 在 finishMessages 中标记，避免重复添加
       const finishMessages = (await result.response).messages;
+      if (isThinking && reasoningContent) {
+        // DeepSeek 要求：当有 tool_calls 时，必须将 reasoning_content 添加到消息中
+        // AI SDK 的 finishMessage 不会包含 reasoning_content，需要手动添加
+        for (const msg of finishMessages) {
+          if (msg.role === 'assistant') {
+            // 检查是否有 tool_calls
+            const hasToolCalls = msg.tool_calls?.length > 0 ||
+              (Array.isArray(msg.content) && msg.content.some(c => c.type === 'tool-call' || c.type === 'tool-use'));
+            if (hasToolCalls) {
+              // DeepSeek API 要求：带 tool_calls 时必须传递 reasoning_content
+              msg.reasoning_content = reasoningContent;
+            }
+          }
+        }
+      }
       messages.push(...finishMessages);
       const usage = await result.totalUsage;
       if (usage) {
@@ -542,6 +576,13 @@ class AgentChatHandler extends EventEmitter {
       }
       const systemPrompt = framework._mainAgent._buildSystemPrompt();
       const tools = this._getAITools(aiTool);
+
+      // DeepSeek thinking mode: 处理参数
+      const apiOptions = { ...this.providerOptions };
+      if (isThinkingModel(this.model)) {
+        delete apiOptions.temperature;
+      }
+
       const agent = new ToolLoopAgent({
         model: this._aiClient,
         instructions: systemPrompt,
@@ -551,13 +592,29 @@ class AgentChatHandler extends EventEmitter {
       });
 
       const result = await framework.runInSession(sessionId, {}, async () => {
-        return agent.generate({ messages, ...this.providerOptions });
+        return agent.generate({ messages, ...apiOptions });
       });
 
       if (result.usage) {
         this._updateMessageStoreUsage(messageStore, result.usage);
       }
 
+      // DeepSeek thinking mode: 处理 reasoning_content
+      const isThinking = isThinkingModel(this.model);
+      if (isThinking && result.reasoningText) {
+        // 有 reasoning_text 需要添加到消息中（当有 tool_calls 时）
+        const finishMessages = result.response.messages;
+        for (const msg of finishMessages) {
+          if (msg.role === 'assistant') {
+            const hasToolCalls = msg.tool_calls?.length > 0 ||
+              (Array.isArray(msg.content) && msg.content.some(c => c.type === 'tool-call' || c.type === 'tool-use'));
+            if (hasToolCalls) {
+              msg.reasoning_content = result.reasoningText;
+            }
+          }
+        }
+      }
+
       messages.push(...result.response.messages);
       const userMsg = messages[messages.length - result.response.messages.length - 1];
       this.emit('message', { content: result.text, sessionId: sessionId, userMessage: userMsg });

package/src/core/context-compressor.js

@@ -15,6 +15,8 @@ const { Subagent } = require('./subagent');
 const MODEL_CONTEXT_LIMITS = {
   'deepseek-chat': 100000,
   'deepseek-coder': 100000,
+  'deepseek-v4-pro': 800000,
+  'deepseek-v4-flash': 800000,
   'deepseek-reasoner': 100000,
   'MiniMax-M2.7': 100000,
   'gpt-4': 100000,

package/src/core/provider.js

@@ -37,6 +37,28 @@ const DEFAULT_PROVIDERS = {
   },
 };
 
+/**
+ * 需要禁用 temperature 的模型（thinking mode 模型）
+ */
+const THINKING_MODELS = [
+  'deepseek-v4-pro',
+  'deepseek-v4-flash',
+  'deepseek-reasoner',
+];
+
+/**
+ * 检查是否是 thinking mode 模型
+ * @param {string} model - 模型名称
+ * @returns {boolean}
+ */
+function isThinkingModel(model) {
+  if (!model) return false;
+  const lowerModel = model.toLowerCase();
+  return THINKING_MODELS.some(
+    (tm) => lowerModel.includes(tm) || lowerModel === tm
+  );
+}
+
 /**
  * 创建 AI 客户端
  * @param {Object} config - 配置
@@ -48,9 +70,22 @@ const DEFAULT_PROVIDERS = {
 function createAI(config) {
   const { provider, model, apiKey, baseURL } = config;
   const providerName = (provider || 'deepseek').toLowerCase();
+  const isThinking = isThinkingModel(model);
+
   // 检查是否是预定义提供商
   if (DEFAULT_PROVIDERS[providerName]) {
     const providerConfig = DEFAULT_PROVIDERS[providerName];
+
+    // 为 DeepSeek thinking mode 模型添加特殊配置
+    const streamOptions = {
+      includeUsage: true,
+    };
+
+    // DeepSeek 模型添加 reasoning_content 输出
+    if (providerName === 'deepseek') {
+      streamOptions.includeReasoningContent = true;
+    }
+
     return createOpenAICompatible({
       name: providerConfig.name,
       baseURL: baseURL || providerConfig.baseURL,
@@ -65,9 +100,7 @@ function createAI(config) {
       models: {
         default: {
           id: model || 'deepseek-chat',
-          streamOptions: {
-            includeUsage: true,
-          },
+          streamOptions,
         },
       },
     });
@@ -111,4 +144,5 @@ module.exports = {
   createModel,
   getAvailableProviders,
   DEFAULT_PROVIDERS,
+  isThinkingModel,
 };