Version 1.0.4 is expected to be released.

Author: PythonnotJava

CHANGELOG.md

@@ -10,7 +10,10 @@
 - 优化了代码高亮文本.
 - 使用[archify](https://github.com/tt-a1i/archify)对对话进行流程图总结输出.
 - 更为可靠的版本工作流.
-
+- 知识库功能.
+- 优化了MCP服务器的载入和启动.
+- 优化了Agent自主生成文件默认在工作目录下.
+- 
 
 ## en
 - Optimized the issue of potential stuttering during long conversations.

assets/default_skills/toolshell/tools.json

@@ -20,11 +20,11 @@
     "type": "function",
     "function": {
       "name": "toolshell_write",
-      "description": "写入文件。支持新建、覆盖、追加。自动创建父目录。",
+      "description": "写入文件。支持新建、覆盖、追加。自动创建父目录。除非用户明确要求写到其他位置，否则默认使用相对于项目根目录的路径，不要自行拼接绝对路径。",
       "parameters": {
         "type": "object",
         "properties": {
-          "path": {"type": "string", "description": "目标文件路径"},
+          "path": {"type": "string", "description": "目标文件路径(相对于项目根目录；无特殊要求时不要使用绝对路径)"},
           "content": {"type": "string", "description": "写入内容"},
           "mode": {"type": "string", "enum": ["create", "overwrite", "append"]}
         },
@@ -40,7 +40,7 @@
       "parameters": {
         "type": "object",
         "properties": {
-          "path": {"type": "string", "description": "目标路径"},
+          "path": {"type": "string", "description": "目标路径(相对于项目根目录)"},
           "recursive": {"type": "boolean", "default": false}
         },
         "required": ["path"]

lib/app.dart

@@ -8,9 +8,11 @@ import 'package:google_fonts/google_fonts.dart';
 import 'package:tray_manager/tray_manager.dart';
 import 'package:window_manager/window_manager.dart';
 import 'core/font/custom_font_loader.dart';
+import 'core/logger/app_logger.dart';
 import 'core/shortcuts/shortcut_config.dart';
 import 'l10n/app_localizations.dart';
 import 'core/l10n/l10n_ext.dart';
+import 'providers/mcp_provider.dart';
 import 'providers/settings_provider.dart';
 import 'providers/api_server_provider.dart';
 import 'features/online_service/online_service_provider.dart';
@@ -288,8 +290,14 @@ class _WindowWrapperState extends ConsumerState<_WindowWrapper>
         _toggleOnlineService();
         break;
       case 'exit':
-        windowManager.setPreventClose(false);
-        windowManager.close();
+        // 这里先关闭 preventClose 再 close()，走的是"直接原生关闭"路径，
+        // 不保证 onWindowClose 监听器一定会被触发去做清理，所以清理动作
+        // 显式放在这里做一遍 (与 onWindowClose 里的清理逻辑重复调用是安全的，
+        // disconnectAll 对已清空的状态是幂等操作)。
+        _cleanupBeforeExit().whenComplete(() {
+          windowManager.setPreventClose(false);
+          windowManager.close();
+        });
         break;
     }
   }
@@ -302,12 +310,29 @@ class _WindowWrapperState extends ConsumerState<_WindowWrapper>
     if (shouldMinimize) {
       await windowManager.hide();
     } else {
+      await _cleanupBeforeExit();
       await trayManager.destroy();
       await windowManager.setPreventClose(false);
       await windowManager.close();
     }
   }
 
+  /// 真正退出前的收尾清理——目前主要是 MCP 连接：stdio 模式的子进程、
+  /// SSE/HTTP 模式下应用代为拉起的本地进程，若不在这里显式断开，
+  /// 单靠 Riverpod 容器 dispose 并不保证会在窗口关闭流程里被同步触发，
+  /// 遗留下来会变成占着端口/常驻后台的僵尸进程。加超时兜底，避免某个
+  /// 进程杀不掉时把整个退出流程卡住。
+  Future<void> _cleanupBeforeExit() async {
+    try {
+      await ref
+          .read(mcpConnectionsProvider.notifier)
+          .disconnectAll()
+          .timeout(const Duration(seconds: 5));
+    } catch (e) {
+      AppLogger.instance.log('[App] 退出前清理 MCP 连接异常(已忽略，继续退出): $e');
+    }
+  }
+
   Future<bool> _showCloseDialog() async {
     final s = context.s;
     final result = await showDialog<bool>(

lib/core/agent/agent_context.dart

@@ -4,8 +4,8 @@ import 'dart:io';
 import 'package:flutter/services.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
 import 'package:path/path.dart' as p;
-import 'package:path_provider/path_provider.dart';
 
+import '../settings/app_settings.dart';
 import '../db/daos/memory_dao.dart';
 import '../llm/llm_client.dart';
 import '../llm/llm_provider.dart';
@@ -23,6 +23,7 @@ import '../toolshell/executor.dart';
 import '../toolshell/worktree_manager.dart';
 import '../../providers/database_provider.dart';
 import '../../providers/experts_provider.dart';
+import '../../providers/kb_provider.dart';
 import '../../providers/mcp_provider.dart';
 import '../../providers/search_provider.dart';
 import '../../providers/settings_provider.dart';
@@ -35,6 +36,7 @@ import 'tool_middleware.dart';
 import 'tool_pipeline.dart';
 import 'hooks/memory_recall_hook.dart';
 import 'hooks/memory_store_hook.dart';
+import 'hooks/kb_recall_hook.dart';
 import 'hooks/schedule_hook.dart';
 import 'hooks/system_probe_hook.dart';
 import 'middleware/logging_middleware.dart';
@@ -129,8 +131,8 @@ class AgentContextBuilder {
     var workDir = _ref.read(workingDirectoryProvider);
     final hasRealWorkDir = workDir.isNotEmpty;
     if (workDir.isEmpty) {
-      final documentsDir = await getApplicationDocumentsDirectory();
-      workDir = p.join(documentsDir.path, '.RemindAI', 'workspace');
+      final root = await AppSettings.getRootDir();
+      workDir = p.join(root, 'workspace');
     }
 
     // ─── Worktree 隔离: 若存在有效的活跃隔离工作树，Executor 的实际
@@ -253,7 +255,11 @@ class AgentContextBuilder {
     // ─── 系统提示词 ───
     // 拆成稳定前缀（专家+元技能）与用户技能区两段，
     // 便于会话中途技能变化时只刷新技能区，保护 prompt cache。
-    final systemPromptPrefix = await _buildSystemPromptPrefix();
+    // 传入 effectiveRoot（而非原始 workDir）确保 worktree 隔离生效时
+    // 模型看到的"当前工作目录"与 Executor 实际使用的根目录一致。
+    final systemPromptPrefix = await _buildSystemPromptPrefix(
+      workDirOverride: effectiveRoot,
+    );
     final skillsSection = await _buildSkillsSection(
       userInput: userInput,
       existingMessages: existingMessages,
@@ -307,13 +313,16 @@ class AgentContextBuilder {
 
     // ─── Hooks ───
     final hasWorkDir = _ref.read(workingDirectoryProvider).isNotEmpty;
+    final selectedKbIds = _ref.read(sessionKnowledgeBasesProvider);
     final hooks = <AgentHook>[
       if (effectiveRecall && memoryManager != null && memoryCollection != null)
         MemoryRecallHook(
           manager: memoryManager,
           collection: memoryCollection,
           useQdrant: useQdrant,
         ),
+      if (selectedKbIds.isNotEmpty)
+        KbRecallHook(dao: _ref.read(kbDaoProvider), kbIds: selectedKbIds),
       if (effectiveStore && memoryManager != null && memoryCollection != null)
         MemoryStoreHook(
           manager: memoryManager,
@@ -645,9 +654,13 @@ class AgentContextBuilder {
 
   /// 构建 system prompt 的稳定前缀：领域专家 + 元技能（或全局模式提示）。
   /// 注意：领域专家为一次性消费（读取后置空），仅在新会话首次构建时注入。
-  Future<String> _buildSystemPromptPrefix() async {
+  /// [workDirOverride] 实际生效的根目录（考虑 worktree 隔离重定向后的值）。
+  /// 不传时回退到 workingDirectoryProvider 的原始值，用于外部无该上下文的调用场景。
+  Future<String> _buildSystemPromptPrefix({String? workDirOverride}) async {
     final parts = <String>[];
-    final hasWorkspace = _ref.read(workingDirectoryProvider).isNotEmpty;
+    final String currentWorkDir =
+        workDirOverride ?? _ref.read(workingDirectoryProvider);
+    final hasWorkspace = currentWorkDir.isNotEmpty;
 
     // 领域专家
     final activeExpert = _ref.read(activeExpertProvider);
@@ -658,6 +671,18 @@ class AgentContextBuilder {
     }
 
     if (hasWorkspace) {
+      // 显式把当前工作目录的绝对路径告知模型，作为 ground truth。
+      // 没有这一行，模型在用户未明确指定路径时只能自行猜测一个绝对路径，
+      // 而写入工具默认不做越界拦截（allowOutsideRoot=true），猜错就会
+      // 把文件写到工作目录之外。
+      parts.add(
+        '# 当前工作目录\n'
+        '$currentWorkDir\n\n'
+        '所有文件操作工具（读取/写入/搜索/删除等）的相对路径均基于此目录解析。'
+        '除非用户明确要求写到其他位置，否则新建、修改文件默认使用相对路径，'
+        '不要自行拼接其他绝对路径。\n\n---\n',
+      );
+
       final tsPrompt = await rootBundle.loadString(
         'assets/default_skills/toolshell/SKILL.md',
       );

lib/core/agent/hooks/kb_recall_hook.dart

@@ -0,0 +1,101 @@
+import 'package:flutter/foundation.dart';
+
+import '../../db/daos/kb_dao.dart';
+import '../../logger/app_logger.dart';
+import '../../memory/memory_manager.dart';
+import '../agent_hook.dart';
+
+/// 知识库召回钩子 — 在用户消息发送前从选中的知识库中检索相关内容注入 context。
+///
+/// 与 MemoryRecallHook 的区别:
+/// - 支持多个知识库同时检索
+/// - 每个知识库使用各自快照的嵌入模型配置
+/// - 结果按知识库分组展示，标明来源文件名
+class KbRecallHook extends AgentHook {
+  final KbDao dao;
+  final List<String> kbIds;
+  final int topK;
+  final double scoreThreshold;
+
+  KbRecallHook({
+    required this.dao,
+    required this.kbIds,
+    this.topK = 5,
+    this.scoreThreshold = 0.45,
+  });
+
+  @override
+  Future<String?> onBeforeUserMessage(
+    String input,
+    List<Map<String, dynamic>> messages,
+  ) async {
+    if (kbIds.isEmpty) return null;
+
+    debugPrint(
+      '[KB Recall] 开始检索, 知识库数=${kbIds.length}, '
+      'query="${input.length > 40 ? input.substring(0, 40) : input}"',
+    );
+
+    final allResults = <String>[];
+
+    for (final kbId in kbIds) {
+      try {
+        final kb = await dao.getBase(kbId);
+        if (kb == null || !kb.hasEmbedding) {
+          debugPrint('[KB Recall] 跳过 $kbId (不存在或未配置嵌入模型)');
+          continue;
+        }
+
+        debugPrint('[KB Recall] 检索知识库: ${kb.name} (${kb.collection})');
+
+        final manager = MemoryManager(
+          embeddingBaseUrl: kb.embeddingBaseUrl,
+          embeddingApiKey: kb.embeddingApiKey,
+          embeddingModel: kb.embeddingModel,
+          memoryDao: null,
+        );
+
+        final results = await manager.recall(
+          query: input,
+          collectionName: kb.collection,
+          topK: topK,
+          scoreThreshold: scoreThreshold,
+          useQdrant: true,
+        );
+
+        debugPrint('[KB Recall] ${kb.name}: 命中 ${results.length} 条');
+
+        if (results.isNotEmpty) {
+          allResults.add('── ${kb.name} ──');
+          for (final r in results) {
+            final text = r['text'] as String? ?? '';
+            final filename = r['filename'] as String? ?? '';
+            final score = r['score'] as double? ?? 0;
+            final prefix = filename.isNotEmpty ? '[$filename] ' : '';
+            allResults.add('$prefix$text');
+            debugPrint(
+              '[KB Recall]   ✓ score=${score.toStringAsFixed(3)}, '
+              'file=$filename, len=${text.length}',
+            );
+          }
+        }
+      } catch (e) {
+        debugPrint('[KB Recall] ✗ 知识库 $kbId 检索失败: $e');
+        AppLogger.instance.log('[KB Recall] 知识库 $kbId 检索失败: $e');
+      }
+    }
+
+    if (allResults.isNotEmpty) {
+      final context = allResults.join('\n');
+      messages.add({
+        'role': 'system',
+        'content': '[以下是从知识库中检索到的相关参考内容，请结合这些信息回答用户问题]\n$context',
+      });
+      debugPrint('[KB Recall] ✓ 已注入 ${allResults.length} 条结果到 context');
+    } else {
+      debugPrint('[KB Recall] 无相关结果');
+    }
+
+    return null;
+  }
+}