@myassis/gateway 1.0.1 → 1.0.2

package/dist/services/llm/LLMClient.js

@@ -1,5 +1,5 @@
 import { persistStore } from "@/stores";
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 import fs from 'fs/promises';
 const logger = getLogger('LLMClient');
 /**
@@ -249,6 +249,7 @@ export class LLMClient {
         }
         else {
             // OpenAI compatible API
+            let existAttachments = false;
             const body = {
                 model: model.modelId,
                 messages: this.messages.map((m) => {
@@ -319,14 +320,8 @@ export class LLMClient {
                     }
                     // 处理用户消息中的附件
                     if (m.role === 'user' && m.attachments && m.attachments.length > 0) {
+                        existAttachments = true;
                         const contentArray = [];
-                        // 如果有文本内容，先添加文本
-                        if (m.content && m.content.trim()) {
-                            contentArray.push({
-                                type: 'text',
-                                text: m.content,
-                            });
-                        }
                         // 添加附件
                         for (const attachment of m.attachments) {
                             if (attachment.type === 'image') {
@@ -373,6 +368,13 @@ export class LLMClient {
                                 }
                             }
                         }
+                        // 如果有文本内容，添加文本
+                        if (m.content && m.content.trim()) {
+                            contentArray.push({
+                                type: 'text',
+                                text: m.content,
+                            });
+                        }
                         return [{
                                 role: m.role,
                                 content: contentArray,
@@ -392,7 +394,7 @@ export class LLMClient {
             }
             const tokens = JSON.stringify(body).length;
             logger.debug('tokens', tokens);
-            if (tokens > 100000) {
+            if (tokens > 200000) {
                 logger.warn('message is too large and has writen to the file debug.json');
                 await fs.writeFile('debug.json', JSON.stringify(body.messages), 'utf-8');
                 throw Error('exceed max message tokens');

package/dist/services/memory/MemoryManager.js

@@ -1,7 +1,8 @@
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 const logger = getLogger('MemoryManager');
 import { toModel } from '../models';
 import { modelsService } from '../dataService';
+import { authStore } from '@/stores';
 import { LLMClient } from '../llm/LLMClient';
 import { appConfig } from '@/config';
 const DEFAULT_CONFIG = {
@@ -16,9 +17,11 @@ export class MemoryManager {
     session;
     signal;
     config = DEFAULT_CONFIG;
-    constructor(session, signal) {
+    childAgent;
+    constructor(session, signal, childAgent) {
         this.session = session;
         this.signal = signal;
+        this.childAgent = childAgent;
     }
     toSummaryMessage(summary, createdAt, modelName) {
         return {
@@ -32,7 +35,11 @@ export class MemoryManager {
     }
     async getHistoryMessagesAsync() {
         // 1. 获取所有消息
-        const messages = this.session.getMessages();
+        let messages = this.session.getMessages();
+        if (this.childAgent) {
+            messages.splice(-1);
+            logger.debug('history', messages);
+        }
         // 2. 消息少于摘要配置的阈值数，直接返回
         logger.debug('message length:', messages.length);
         if (messages.length < this.config.summaryThreshold) {
@@ -77,32 +84,120 @@ export class MemoryManager {
             }
         }
     }
+    /**
+     * 格式化工具调用内容，提取关键信息供摘要模型理解
+     */
+    formatToolCalls(toolCalls) {
+        if (!toolCalls || toolCalls.length === 0)
+            return '';
+        return toolCalls.map(tc => {
+            const parts = [];
+            // 外层 toolCall 节点
+            if (tc.content)
+                parts.push(`思路: ${tc.content}`);
+            if (tc.reasoningContent)
+                parts.push(`推理: ${tc.reasoningContent}`);
+            // 内层工具调用详情
+            if (tc.toolCalls && tc.toolCalls.length > 0) {
+                const callDetails = tc.toolCalls.map((call) => {
+                    const name = `${call.toolName}${call.actionName ? '.' + call.actionName : ''}`;
+                    let detail = name;
+                    if (call.input) {
+                        try {
+                            const input = typeof call.input === 'string' ? JSON.parse(call.input) : call.input;
+                            // 提取关键输入信息，避免过长
+                            const inputStr = JSON.stringify(input).slice(0, 500);
+                            detail += ` 输入: ${inputStr}`;
+                        }
+                        catch {
+                            detail += ` 输入: ${String(call.input).slice(0, 500)}`;
+                        }
+                    }
+                    if (call.output) {
+                        const outputStr = typeof call.output === 'string' ? call.output : JSON.stringify(call.output);
+                        detail += ` 结果: ${outputStr.slice(0, 500)}`;
+                    }
+                    if (call.status === 'error')
+                        detail += ' [失败]';
+                    return detail;
+                }).join('; ');
+                parts.push(`工具调用: ${callDetails}`);
+            }
+            return parts.join('\n');
+        }).join('\n');
+    }
+    /**
+     * 格式化单条消息，提取有意义的内容
+     */
+    formatMessage(m) {
+        const role = m.role === 'user' ? '用户' : m.role === 'assistant' ? '助手' : '系统';
+        let content = m.content || '';
+        // 附加工具调用信息
+        const toolCallInfo = this.formatToolCalls(m.toolCalls);
+        if (toolCallInfo) {
+            content += `\n[${toolCallInfo}]`;
+        }
+        // 截断过长内容
+        if (content.length > 2000) {
+            content = content.slice(0, 2000) + '...(内容过长已截断)';
+        }
+        return `${role}: ${content}`;
+    }
     /**
      * 生成摘要提示词
      */
     buildSummaryPrompt(messages, lastSummary) {
         const conversation = messages
             .filter((m) => m.role === 'user' || m.role === 'assistant')
-            .map((m) => `${m.role === 'user' ? '用户' : '助手'}: ${m.content}${m.toolCalls ? JSON.stringify(m.toolCalls) : ''}`)
-            .join('\n');
-        return `请分析以下对话内容，生成一个简洁的摘要，保留关键信息、错误信息和上下文。
+            .map((m) => this.formatMessage(m))
+            .join('\n\n');
+        return `你是一个对话摘要专家。请分析以下对话内容，生成一份结构化的摘要。
+
+重要规则：
+1. 如果对话涉及代码开发/项目工作，必须详细总结项目结构，包括：技术栈、目录结构、关键文件及其职责、模块间关系
+2. 项目结构描述不限字数，务必完整清晰
+3. 保留所有关键决策、修改内容和上下文
+4. 保留错误信息和解决方案，避免重复犯错
+5. 如果有上一轮摘要，合并其中的关键信息
 
-对话内容：
+${lastSummary ? '=== 上一轮摘要 ===\n' + lastSummary + '\n\n' : ''}=== 对话内容 ===
 ${conversation}
 
-${lastSummary ? '上一次内容摘要:\n' + lastSummary : ''}
+=== 输出格式 ===
+请按以下格式输出（各部分字数不限，以完整准确为优先）：
+
+【摘要】简要概括对话主题和关键进展
+
+【项目结构】（如涉及项目开发）详细描述：
+- 技术栈与框架
+- 目录结构与各目录职责
+- 关键文件及其作用
+- 模块间依赖关系
+- 已完成和待完成的功能
 
-请用以下格式返回摘要（只返回摘要，不要其他内容）：
-【摘要】：简要概括对话主题和关键信息（不超过100字）
-【要点】：列出2-5个关键要点
-【错误】：列出犯过的错误`;
+【关键决策】列出重要决策及原因
+
+【要点】列出3-8个关键要点
+
+【错误与修复】列出遇到的错误及解决方案，标注"⚠️"避免重复`;
     }
     /**
-     * 生成对话摘要
+     * 生成对话摘要（支持分层摘要）
      */
     async generateSummaryAsync(messages, lastSummary) {
+        const MAX_INPUT_CHARS = 30000; // 单次摘要最大输入字符数
+        // 格式化所有消息用于估算长度
+        const formattedMessages = messages
+            .filter((m) => m.role === 'user' || m.role === 'assistant')
+            .map((m) => this.formatMessage(m));
+        const totalChars = formattedMessages.join('').length;
+        if (totalChars > MAX_INPUT_CHARS && formattedMessages.length > 20) {
+            // 分层摘要：分批生成子摘要，再合并
+            return await this.hierarchicalSummary(formattedMessages, lastSummary);
+        }
         const summaryPrompt = this.buildSummaryPrompt(messages, lastSummary);
-        const llmClient = new LLMClient((await modelsService.list()).data.map((x) => toModel(x)), [{ role: 'user', content: summaryPrompt }], this.signal, []);
+        const token = authStore.get(this.session.userId).accessToken;
+        const llmClient = new LLMClient((await modelsService.list(token)).data.map((x) => toModel(x)), [{ role: 'user', content: summaryPrompt }], this.signal, []);
         const response = await llmClient.Chat();
         const content = (response.content || '').trim();
         if (!content) {
@@ -114,4 +209,72 @@ ${lastSummary ? '上一次内容摘要:\n' + lastSummary : ''}
         }
         return content;
     }
+    /**
+     * 分层摘要：将消息分批生成子摘要，再合并为最终摘要
+     */
+    async hierarchicalSummary(formattedMessages, lastSummary) {
+        const BATCH_SIZE = 15; // 每批消息数
+        const subSummaries = [];
+        for (let i = 0; i < formattedMessages.length; i += BATCH_SIZE) {
+            if (this.signal.aborted)
+                break;
+            const batch = formattedMessages.slice(i, i + BATCH_SIZE);
+            const batchPrompt = `请简洁总结以下对话片段的关键信息，特别关注：项目结构、关键决策、错误及修复。\n\n${batch.join('\n\n')}`;
+            const token = authStore.get(this.session.userId).accessToken;
+            const llmClient = new LLMClient((await modelsService.list(token)).data.map((x) => toModel(x)), [{ role: 'user', content: batchPrompt }], this.signal, []);
+            const response = await llmClient.Chat();
+            const subContent = (response.content || '').trim();
+            if (subContent) {
+                subSummaries.push(subContent);
+            }
+        }
+        // 合并所有子摘要
+        if (subSummaries.length === 0) {
+            return lastSummary || '（摘要生成失败）';
+        }
+        if (subSummaries.length === 1) {
+            return subSummaries[0];
+        }
+        // 合并子摘要为最终摘要
+        const mergePrompt = this.buildMergePrompt(subSummaries, lastSummary);
+        const token = authStore.get(this.session.userId).accessToken;
+        const mergeLlmClient = new LLMClient((await modelsService.list(token)).data.map((x) => toModel(x)), [{ role: 'user', content: mergePrompt }], this.signal, []);
+        const mergeResponse = await mergeLlmClient.Chat();
+        const mergedContent = (mergeResponse.content || '').trim();
+        return mergedContent || subSummaries.join('\n\n');
+    }
+    /**
+     * 构建合并摘要提示词
+     */
+    buildMergePrompt(subSummaries, lastSummary) {
+        const summaries = subSummaries
+            .map((s, i) => `--- 片段 ${i + 1} ---\n${s}`)
+            .join('\n\n');
+        return `你是一个对话摘要专家。以下是多个对话片段的摘要，请合并为一份完整的结构化摘要。
+
+重要规则：
+1. 合并重复信息，去重项目结构描述
+2. 如果涉及项目开发，务必完整保留项目结构信息（技术栈、目录结构、关键文件、模块关系）
+3. 按时间顺序整理决策和进展
+4. 保留所有错误信息及解决方案
+
+${lastSummary ? '=== 上一轮摘要 ===\n' + lastSummary + '\n\n' : ''}=== 分片段摘要 ===
+${summaries}
+
+=== 输出格式 ===
+【摘要】简要概括对话主题和关键进展
+
+【项目结构】（如涉及项目开发）详细描述：
+- 技术栈与框架
+- 目录结构与各目录职责
+- 关键文件及其作用
+- 模块间依赖关系
+- 已完成和待完成的功能
+
+【关键决策】列出重要决策及原因
+
+【要点】列出3-8个关键要点
+
+【错误与修复】列出遇到的错误及解决方案，标注"⚠️"避免重复`;
+    }
 }

package/dist/services/session/MigrationManager.js

@@ -1,6 +1,6 @@
 import path from 'path';
 import fs from 'fs';
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 const logger = getLogger('MigrationManager');
 export class MigrationManager {
     db;

package/dist/services/session/Session.js

@@ -1,15 +1,16 @@
 import { v4 as uuidv4 } from 'uuid';
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 import { SessionStore } from './SessionStore';
 import { AgentStore } from '@/services/agent/AgentStore';
+import { authStore } from '@/stores/authStore';
 import { toModel } from '@/services/models';
 import { executeTool, getToolDefinitions } from '@/services/tools';
 import { getSystemPromptAsync } from '../systemPrompt';
 import { LLMClient, parseAruments } from '../llm/LLMClient';
 import { modelsService, settingsService } from '@/services/dataService';
 import { MemoryManager } from '../memory/MemoryManager';
-import { sessionManager } from './SessionManager';
 import { appConfig } from '@/config';
+import { getSessionManager } from './SessionManager';
 const logger = getLogger('Session');
 const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));
 // AgentStore 单例 - 需要通过 SessionStore 获取数据库实例
@@ -46,7 +47,6 @@ export class Session {
     unreadCount = 0;
     currentMessageId = null;
     abortController = null;
-    toolCalls;
     toInsertMessages;
     constructor(data) {
         this.id = data.id;
@@ -103,16 +103,13 @@ export class Session {
         return this.messages.reduce((total, msg) => total + msg.content.length, 0);
     }
     // ========== Message Operations ==========
-    /**
-     * Add user message
-     */
-    addUserMessage(content, userMessageId, attachments) {
+    getUserMessage(content, userMessageId, attachments) {
         // 将附件统一为 Attachment 对象格式
         const normalizedAttachments = (attachments || []).map((att) => {
             if (typeof att === 'string') {
                 // URL 字符串格式，转换为对象
                 const url = att;
-                const name = url.split('/').pop() || 'attachment';
+                const name = url.split('/').pop().split('?')[0] || 'attachment';
                 const ext = name.split('.').pop()?.toLowerCase() || '';
                 const imageExts = ['jpg', 'jpeg', 'png', 'gif', 'webp', 'bmp', 'svg'];
                 const videoExts = ['mp4', 'avi', 'mov', 'mkv', 'webm'];
@@ -145,6 +142,13 @@ export class Session {
             createdAt: Date.now(),
             modelName: ''
         };
+        return message;
+    }
+    /**
+     * Add user message
+     */
+    addUserMessage(content, userMessageId, attachments) {
+        const message = this.getUserMessage(content, userMessageId, attachments);
         this.messages.push(message);
         this.toInsertMessages.push(message);
         return message;
@@ -375,15 +379,27 @@ export class Session {
      * Stream chat response (SSE) - 使用 ModelSelector 根据选择的模型调用
      * 支持工具调用流程
      */
-    async streamChat(content, attachments = [], res, userMessageId, assistantMessageId) {
+    async streamChat(content, attachments = [], res, userMessageId, assistantMessageId, childAgent = false) {
+        // 如果正在生成，等待当前生成完成（最多等待 2 秒）
+        if (this.isGenerating) {
+            const maxWait = 2000;
+            const startTime = Date.now();
+            while (this.isGenerating && Date.now() - startTime < maxWait) {
+                await sleep(100);
+            }
+            // 超时后强制中断
+            if (this.isGenerating) {
+                this.stopGenerating();
+            }
+        }
         // Add user message
         this.getAbortController();
         this.isGenerating = true;
-        this.toolCalls = [];
-        const settings = await settingsService.get();
+        const token = authStore.get(this.userId).accessToken;
+        const settings = await settingsService.get(token);
         const streamDelay = this.getStreamDelay(settings.streamSpeed);
         this.currentMessageId = assistantMessageId;
-        const memoryManager = new MemoryManager(this, this.abortController.signal);
+        const memoryManager = new MemoryManager(this, this.abortController.signal, childAgent);
         const historyMessages = await memoryManager.getHistoryMessagesAsync();
         // SSE 辅助方法：res 为 null 时跳过写入（本地执行模式）
         const sendSSE = (res, data) => {
@@ -433,8 +449,13 @@ export class Session {
             messages.unshift({ role: 'system', content: systemPrompt, attachments: [] });
         }
         //插入用户消息
-        this.addUserMessage(content, userMessageId, attachments);
-        messages.push(this.messages.at(-1));
+        if (!childAgent) {
+            this.addUserMessage(content, userMessageId, attachments);
+            messages.push(this.messages.at(-1));
+        }
+        else {
+            messages.push(this.getUserMessage(content, userMessageId, attachments));
+        }
         let toolCall = null;
         let modelNames = [];
         //用于后续的消息滑动处理
@@ -450,6 +471,7 @@ export class Session {
                 message.reasoning_content = message.reasoning_content.substring(0, 100) + '...(内容过长已截断)';
             }
         };
+        const toolCalls = [];
         // 递归处理函数（支持多轮工具调用）
         const processModelResponse = async () => {
             if (!this.abortController || !this.abortController.signal || this.abortController.signal.aborted) {
@@ -460,7 +482,7 @@ export class Session {
             // 清理旧工具调用：只保留最近 keep 轮的完整上下文
             {
                 const MAX_TOOL_PAIRS = keep;
-                const allToolCallItems = this.toolCalls.flatMap(x => x.toolCalls);
+                const allToolCallItems = toolCalls.flatMap(x => x.toolCalls);
                 // 1) 清理 this.toolCalls 中超出 keep 的 output（供前端展示用）
                 allToolCallItems.forEach((item, index) => {
                     if (index < allToolCallItems.length - MAX_TOOL_PAIRS) {
@@ -498,7 +520,7 @@ export class Session {
             }
             // 获取本地工具定义
             const tools = getToolDefinitions();
-            const models = (await modelsService.list()).data.map(x => toModel(x));
+            const models = (await modelsService.list(token)).data.map(x => toModel(x));
             // 使用 LLMClient 进行流式调用
             if (!this.abortController || !this.abortController.signal || this.abortController.signal.aborted) {
                 throw Error('aborted');
@@ -568,7 +590,7 @@ export class Session {
                 }
                 // 执行工具调用并收集结果
                 const toolResults = [];
-                for (const pendingToolCall of llmResult.toolCalls) {
+                const startToolCall = async (pendingToolCall) => {
                     // 发送工具开始执行事件
                     const toolStartEvent = {
                         type: 'tool_call_execute',
@@ -581,7 +603,7 @@ export class Session {
                     };
                     sendSSE(res, toolStartEvent);
                     try {
-                        const result = await executeTool(pendingToolCall.toolName, parseAruments(pendingToolCall.input), this.id, this.currentMessageId);
+                        const result = await executeTool(pendingToolCall.toolName, parseAruments(pendingToolCall.input), this.id, this.currentMessageId, this.userId);
                         toolResults.push({
                             id: pendingToolCall.id,
                             name: pendingToolCall.toolName,
@@ -638,8 +660,9 @@ export class Session {
                         };
                         sendSSE(res, errorEvent);
                     }
-                }
-                this.toolCalls.push(toolCall);
+                };
+                await Promise.all(llmResult.toolCalls.map(x => startToolCall(x)));
+                toolCalls.push(toolCall);
                 // 将工具结果添加到消息历史（格式化为 tool 角色的消息）
                 for (const result of toolResults) {
                     const toolResultMessage = {
@@ -655,7 +678,7 @@ export class Session {
                     messages.push(toolResultMessage);
                 }
                 // 继续递归处理（带上工具结果继续调用模型）
-                await processModelResponse();
+                return await processModelResponse();
             }
             else {
                 // ========== 没有工具调用，保存消息并结束 ==========
@@ -672,34 +695,43 @@ export class Session {
                         await sleep(streamDelay);
                     }
                 }
-                this.addAssistantMessage(llmResult.content || llmResult.reasoningContent, this.toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
+                if (!childAgent) {
+                    this.addAssistantMessage(llmResult.content || llmResult.reasoningContent, toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
+                }
                 sendSSE(res, { type: 'complete', modelName: [...new Set(modelNames)].join(",") });
                 sendSSE(res, { type: '[DONE]' });
+                return llmResult.content;
             }
         };
         try {
-            await processModelResponse();
+            const result = await processModelResponse();
+            return { success: true, data: result };
         }
         catch (error) {
             logger.error('stream chat ', error);
-            if (this.toolCalls) {
-                this.addAssistantMessage('', this.toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
-            }
-            if (error.message !== 'aborted') {
-                sendSSE(res, { type: 'error', error: error.message });
+            if (!childAgent) {
+                if (toolCalls) {
+                    this.addAssistantMessage('', toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
+                }
+                if (error.message !== 'aborted') {
+                    sendSSE(res, { type: 'error', error: error.message });
+                }
             }
+            return { success: false, error: error.message };
         }
         finally {
-            this.isGenerating = false;
-            this.abortController = null;
-            if (res) {
-                res.end();
-            }
-            // 非当前查看的 session，助手回复完成后增加未读计数
-            sessionManager.onMessageComplete(this.id);
-            // 只在非 abort 的情况下保存（abort 由 stopGenerating 负责保存）
-            if (this.currentMessageId !== null) {
-                this.saveMessage();
+            if (!childAgent) {
+                this.isGenerating = false;
+                this.abortController = null;
+                if (res) {
+                    res.end();
+                }
+                // 非当前查看的 session，助手回复完成后增加未读计数
+                getSessionManager(this.userId).onMessageComplete(this.id);
+                // 只在非 abort 的情况下保存（abort 由 stopGenerating 负责保存）
+                if (this.currentMessageId !== null) {
+                    this.saveMessage();
+                }
             }
         }
     }
@@ -712,14 +744,6 @@ export class Session {
             this.abortController.abort();
             this.abortController = null;
         }
-        // stopGenerating 负责保存中断的消息，streamChat 的 finally 不再重复保存
-        if (this.currentMessageId) {
-            if (this.toolCalls && this.toolCalls.length > 0) {
-                this.addAssistantMessage('', this.toolCalls, this.currentMessageId);
-            }
-            this.saveMessage();
-            this.currentMessageId = null;
-        }
     }
     /**
      * 获取 AbortController 用于中断请求

package/dist/services/session/SessionManager.js

@@ -1,5 +1,5 @@
 import { v4 as uuidv4 } from 'uuid';
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 const logger = getLogger('SessionManager');
 import { sessionStore } from './SessionStore';
 import { authStore } from '@/stores/authStore';
@@ -11,7 +11,9 @@ import { Session } from './Session';
 export class SessionManager {
     sessions = null;
     initialized = false;
-    constructor() {
+    userId;
+    constructor(userId) {
+        this.userId = userId;
     }
     /**
      * Initialize - Load all sessions from database on startup
@@ -40,7 +42,7 @@ export class SessionManager {
      * Get current user ID
      */
     getCurrentUserId() {
-        const user = authStore.getUser();
+        const user = authStore.getUser(this.userId);
         return user ? String(user.id) : null;
     }
     /**
@@ -251,5 +253,10 @@ export class SessionManager {
         return sessionStore.getMigrationStatus();
     }
 }
-// Export singleton
-export const sessionManager = new SessionManager();
+export let sessionManagerMap = new Map();
+export function getSessionManager(userId) {
+    if (!sessionManagerMap.has(userId)) {
+        sessionManagerMap.set(userId, new SessionManager(userId));
+    }
+    return sessionManagerMap.get(userId);
+}

package/dist/services/session/SessionStore.js

@@ -1,7 +1,7 @@
 import Database from 'better-sqlite3';
 import path from 'path';
 import fs from 'fs';
-import { getLogger } from '@pocketclaw/shared';
+import { getLogger } from '@myassis/shared';
 import { MigrationManager } from './MigrationManager';
 const logger = getLogger('SessionStore');
 /**
@@ -22,36 +22,10 @@ export class SessionStore {
         const migrationsDir = path.join(process.cwd(), 'migrations');
         this.migrationManager = new MigrationManager(this.db, migrationsDir);
         this.migrationManager.migrate();
-        this.ensureColumns();
         logger.info(`Initialized at ${dbPath}`);
         // 注册到全局供 Session 使用
         global.__sessionStore = this;
     }
-    /**
-     * Ensure required columns exist
-     */
-    ensureColumns() {
-        // Ensure agent_id column in sessions
-        const sessionTableInfo = this.db.prepare("PRAGMA table_info(sessions)").all();
-        const hasAgentId = sessionTableInfo.some(col => col.name === 'agent_id');
-        if (!hasAgentId) {
-            this.db.exec("ALTER TABLE sessions ADD COLUMN agent_id TEXT");
-            logger.info('Added agent_id column to sessions');
-        }
-        // Ensure tool_calls column in messages
-        const msgTableInfo = this.db.prepare("PRAGMA table_info(messages)").all();
-        const hasToolCalls = msgTableInfo.some(col => col.name === 'tool_calls');
-        if (!hasToolCalls) {
-            this.db.exec("ALTER TABLE messages ADD COLUMN tool_calls TEXT");
-            logger.info('Added tool_calls column to messages');
-        }
-        // Ensure feedback column in messages
-        const hasFeedback = msgTableInfo.some(col => col.name === 'feedback');
-        if (!hasFeedback) {
-            this.db.exec("ALTER TABLE messages ADD COLUMN feedback TEXT");
-            logger.info('Added feedback column to messages');
-        }
-    }
     /**
      * Provide transaction method
      */

package/dist/services/session/index.js

@@ -1,3 +1,3 @@
 // 导出 SessionManager（业务逻辑层）
-export { SessionManager, sessionManager } from './SessionManager';
+export { SessionManager, getSessionManager } from './SessionManager';
 export { Session } from './Session';