-
Notifications
You must be signed in to change notification settings - Fork 43
feat(settings): 本地/远程模式下管理远程服务器技能 #36
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Changes from all commits
fdb2411
b5fd912
17fbded
e6cc0bf
33edd55
13b5b84
465359e
d9f3c0e
2d364d6
40d7de1
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
| 你是一个记忆整理器。给定一段对话记录,判断其中是否有值得长期记住的信息。 | ||
|
|
||
| 只提取以下三类事实,且必须是对话中明确出现的信号,不要推测、不要脑补: | ||
|
|
||
| 1. 用户偏好:用户明确表达的喜好或习惯(输出风格、沟通方式、常用工具等),跨项目通用。 | ||
| 2. 项目约定:本项目的工作方式(构建命令、目录结构、提交规范、专属配置等)。 | ||
| 3. 踩坑经验:讨论中确认的 bug 根因与修法、验证过的绕行方案。 | ||
|
|
||
| 输出严格的 JSON 数组,没有任何其他文本。每条格式: | ||
|
|
||
| [{"name": "snake_case 短名", "description": "一句话摘要(何时该读它)", "scope": "global 或 project", "content": "Markdown 正文,具体、可执行"}, ...] | ||
|
|
||
| 规则: | ||
| - 没有值得记的就输出 []。 | ||
| - name 是文件名:英文小写加下划线,不含空格和路径分隔符,不超过 64 字符。 | ||
| - content 写具体事实,不要写空话;单条不超过 500 字。 | ||
| - 宁缺毋滥:普通任务执行过程、一次性问题不要记。 | ||
| - 最多 3 条。 |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,15 @@ | ||
| # 记忆纪律(何时必须调用 memory 工具) | ||
|
|
||
| 长期记忆不是可选项。以下信号出现时,**当轮立即**用 memory 工具记录,不要等会话结束,不要等用户要求: | ||
|
|
||
| - 用户表达个人偏好(输出风格、沟通方式、习惯做法)→ scope=global; | ||
| - 用户纠正过你的做法、指出你说错的事实 → 把纠正记下来(global 或按内容归 project),避免再犯; | ||
| - 项目约定(构建方式、目录结构、分支/提交规范、专属工具链)→ scope=project; | ||
| - 定位到 bug 根因并验证修复后,把「根因 + 修法」沉淀成踩坑记忆 → scope=project。未经根因确认的猜测不记。 | ||
|
|
||
| 记录方式: | ||
|
|
||
| - 先用 memory(action=list) 确认是否已有同名/同主题记忆,再决定 save 还是 edit; | ||
| - 已有相关记忆 → 用 memory(action=edit) 局部更新正文,不要新建重复文件; | ||
| - 确属新主题 → memory(action=save),description 写清「何时该读它」; | ||
| - 同一事实已经记录过 → 不再重复记录;记忆内容过时 → 用 edit 修正或 delete 清理。 |
| Original file line number | Diff line number | Diff line change | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| @@ -0,0 +1,132 @@ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| package com.aicode.feature.agent.domain.memory | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import com.aicode.core.util.FileLogger | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import com.aicode.feature.agent.domain.memory.MemorySource | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import com.aicode.feature.agent.domain.model.AgentMessage | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import com.aicode.feature.agent.domain.provider.AIProvider | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import com.aicode.feature.agent.domain.prompt.PromptFileResolver | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import kotlinx.serialization.json.Json | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import kotlinx.serialization.json.contentOrNull | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import kotlinx.serialization.json.jsonArray | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import kotlinx.serialization.json.jsonObject | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import kotlinx.serialization.json.jsonPrimitive | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import javax.inject.Inject | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| import javax.inject.Singleton | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private const val TAG = "MemoryCurator" | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** 单次抽取的上限条数(与提示词约定一致)。 */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private const val MAX_CANDIDATES = 3 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** 一轮对话送给整理器的上下文上限(字符):只看最近发生的事,控制成本。 */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private const val MAX_TRANSCRIPT_CHARS = 12_000 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * 引擎级记忆兜底:一轮对话结束后,用轻量模型静默抽取值得长期记住的事实, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * 直接写入 [MemoryRepository]。主模型忘了调用 memory 工具时由此兜底; | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * 全程静默失败,绝不影响对话主流程。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| @Singleton | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| class MemoryCurator @Inject constructor( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private val memoryRepository: MemoryRepository, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private val promptFileResolver: PromptFileResolver | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ) { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** 提示词文件名,与 [PromptFileResolver.resolve] 的路径约定一致。 */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private fun prompt(): String = promptFileResolver.resolve("agent/memory-curator.md") | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * 抽取并落盘本轮对话的记忆。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * @param provider 由调用方解析好的轻量 provider(压缩专用模型或当前聊天模型回退)。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * @param transcript 本轮对话文本("用户: …/助手: …" 行)。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| * @return 本次实际写入的记忆条数(失败为 0)。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| suspend fun curate( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| provider: AIProvider, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| sessionId: String, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| projectRoot: String?, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| transcript: String | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ): Int = runCatching { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (transcript.isBlank()) return@runCatching 0 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val systemPrompt = prompt().replace(LEADING_COMMENT, "").trim() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (systemPrompt.isEmpty()) return@runCatching 0 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val response = provider.complete( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| systemPrompt = systemPrompt, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| messages = listOf(AgentMessage.UserMessage(content = transcript.takeLast(MAX_TRANSCRIPT_CHARS))), | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| tools = emptyList() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val candidates = parseCandidates(response.content) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| var saved = 0 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| for (c in candidates) { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| // 已有同名记忆不覆盖:主模型当轮写的完整版本不该被 curator 的截断版覆盖。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val existing = memoryRepository.loadContent(c.name, projectRoot) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (existing != null) { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| FileLogger.i(TAG, "记忆「${c.name}」已存在,跳过自动覆盖") | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| continue | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| // PROJECT 但无工作区时降级为 GLOBAL(静默场景下降级比丢弃合理)。 | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val effectiveScope = if (c.scope == MemoryScope.PROJECT && projectRoot.isNullOrBlank()) { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| MemoryScope.GLOBAL | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } else { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| c.scope | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val ok = memoryRepository.saveMemory( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| name = c.name, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| description = c.description, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| content = c.content, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| scope = effectiveScope, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| projectRoot = projectRoot | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (ok) saved++ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (saved > 0) { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| FileLogger.i(TAG, "会话 $sessionId 自动沉淀 $saved 条记忆") | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| saved | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| }.onFailure { e -> | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| FileLogger.w(TAG, "自动记忆整理失败(静默忽略): ${e.message}", e) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| }.getOrDefault(0) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
Comment on lines
+48
to
+88
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 🩺 Stability & Availability | 🟠 Major | ⚡ Quick win Rethrow
🐛 Proposed fix }.onFailure { e ->
+ if (e is kotlinx.coroutines.CancellationException) throw e
FileLogger.w(TAG, "自动记忆整理失败(静默忽略): ${e.message}", e)
}.getOrDefault(0)Based on learnings: "avoid using runCatching ... inside suspend functions, because it also catches CancellationException." 📝 Committable suggestion
Suggested change
🤖 Prompt for AI AgentsSource: Learnings |
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| /** 解析整理器输出;格式不合法/越界条目一律丢弃,宁缺毋滥。 */ | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private fun parseCandidates(content: String): List<Candidate> { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val start = content.indexOf('[') | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val end = content.lastIndexOf(']') | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (start < 0 || end <= start) return emptyList() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val arr = runCatching { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| Json.parseToJsonElement(content.substring(start, end + 1)).jsonArray | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| }.getOrNull() ?: return emptyList() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| return arr.asSequence() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| .mapNotNull { el -> runCatching { el.jsonObject }.getOrNull() } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| .mapNotNull { obj -> | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| runCatching { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val name = obj["name"]?.jsonPrimitive?.contentOrNull?.trim().orEmpty() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val description = obj["description"]?.jsonPrimitive?.contentOrNull?.trim().orEmpty() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val body = obj["content"]?.jsonPrimitive?.contentOrNull?.trim().orEmpty() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val isProject = obj["scope"]?.jsonPrimitive?.contentOrNull?.trim() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ?.equals("project", ignoreCase = true) == true | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| if (name.isEmpty() || body.isEmpty() || description.isEmpty()) return@runCatching null | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| Candidate( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| name = name, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| description = description.take(200), | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| content = body.take(2000), | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| scope = if (isProject) MemoryScope.PROJECT else MemoryScope.GLOBAL | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| }.getOrNull() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| .filter { MemorySource.sanitizeName(it.name).isNotEmpty() } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| .take(MAX_CANDIDATES) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| .toList() | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private data class Candidate( | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val name: String, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val description: String, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val content: String, | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val scope: MemoryScope | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| ) | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
|
|
||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| private companion object { | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| val LEADING_COMMENT = Regex("(?s)^\\s*<!--.*?-->\\s*") | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| } | ||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,54 @@ | ||
| package com.aicode.feature.agent.domain.prompt | ||
|
|
||
| import com.aicode.core.util.FileLogger | ||
| import com.aicode.feature.agent.domain.container.ContainerInstaller | ||
| import dagger.hilt.android.qualifiers.ApplicationContext | ||
| import android.content.Context | ||
| import java.io.File | ||
| import javax.inject.Inject | ||
| import javax.inject.Singleton | ||
|
|
||
| /** | ||
| * 单个提示词片段的文件解析器,供 [SystemPromptProvider] 与无循环依赖需求的 | ||
| * 其它组件(如 MemoryCurator)共用: | ||
| * | ||
| * - 名字是顶层 `<NN>-*.md`:先按数字身份在 `prompts.custom/` 顶层找覆盖(尾部名称可自由改), | ||
| * - 其余名字(含 `agent/` 子目录):按精确同名在 `prompts.custom/<name>` 找覆盖; | ||
| * - 再落到 `prompts/<name>`(本地默认副本),最后 assets(内置兜底)。 | ||
| * | ||
| * 本地副本由 [ContainerInstaller.extractPrompts] 在启动时全量释放,App 升级后随之更新。 | ||
| */ | ||
| @Singleton | ||
| class PromptFileResolver @Inject constructor( | ||
| @param:ApplicationContext private val context: Context, | ||
| private val containerInstaller: ContainerInstaller | ||
| ) { | ||
| private val customDir: File | ||
| get() = File(containerInstaller.aicodeDir, "prompts.custom") | ||
|
|
||
| private val customFragmentsByNumber: Map<Int, File> by lazy { | ||
| PromptFragmentResolver.numberedFragments(customDir).toMap() | ||
| } | ||
|
|
||
| fun resolve(name: String): String { | ||
| PromptFragmentResolver.parseNumber(name) | ||
| ?.let { number -> readFileOrNull(customFragmentsByNumber[number])?.let { return it } } | ||
| readFileOrNull(File(customDir, name))?.let { return it } | ||
| readFileOrNull(File(File(containerInstaller.aicodeDir, "prompts"), name))?.let { return it } | ||
| return context.assets.open("prompts/$name").bufferedReader().use { it.readText() } | ||
| } | ||
|
|
||
| private fun readFileOrNull(file: File?): String? { | ||
| if (file == null || !file.isFile) return null | ||
| return try { | ||
| file.bufferedReader().use { it.readText() } | ||
| } catch (e: Exception) { | ||
| FileLogger.w(TAG, "读取提示词失败 ${file.name}: ${e.message}", e) | ||
| null | ||
| } | ||
| } | ||
|
|
||
| private companion object { | ||
| const val TAG = "PromptFileResolver" | ||
| } | ||
| } |
| Original file line number | Diff line number | Diff line change | ||||||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
|
|
@@ -32,6 +32,7 @@ class SystemPromptProvider @Inject constructor( | |||||||||||||||||||
| @param:ApplicationContext private val context: Context, | ||||||||||||||||||||
| private val skillRepository: SkillRepository, | ||||||||||||||||||||
| private val memoryRepository: MemoryRepository, | ||||||||||||||||||||
| private val promptFileResolver: PromptFileResolver, | ||||||||||||||||||||
| private val containerInstaller: ContainerInstaller, | ||||||||||||||||||||
| private val agentDefinitionRepository: AgentDefinitionRepository | ||||||||||||||||||||
| ) { | ||||||||||||||||||||
|
|
@@ -194,7 +195,8 @@ class SystemPromptProvider @Inject constructor( | |||||||||||||||||||
| val memories = try { memoryRepository.listMemories(ctx.projectRoot) } catch (e: Exception) { return null } | ||||||||||||||||||||
| if (memories.isEmpty()) { | ||||||||||||||||||||
| cachedByKey[key] = "" | ||||||||||||||||||||
| return null | ||||||||||||||||||||
| // 空清单也要注入纪律:首次会话正是建立记忆的起点。 | ||||||||||||||||||||
| return memoryDiscipline() | ||||||||||||||||||||
|
Comment on lines
196
to
+199
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 🎯 Functional Correctness | 🟠 Major | ⚡ Quick win Cache the discipline text for an empty memory list. When the memory list is empty, the code caches 🐛 Proposed fix if (memories.isEmpty()) {
- cachedByKey[key] = ""
- // 空清单也要注入纪律:首次会话正是建立记忆的起点。
- return memoryDiscipline()
+ val discipline = memoryDiscipline()
+ cachedByKey[key] = discipline.orEmpty()
+ return discipline
}📝 Committable suggestion
Suggested change
🤖 Prompt for AI Agents |
||||||||||||||||||||
| } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| val globalMemories = memories.filter { it.scope == MemoryScope.GLOBAL } | ||||||||||||||||||||
|
|
@@ -212,14 +214,26 @@ class SystemPromptProvider @Inject constructor( | |||||||||||||||||||
| } | ||||||||||||||||||||
| }.trimEnd() | ||||||||||||||||||||
|
|
||||||||||||||||||||
| cachedByKey[key] = content | ||||||||||||||||||||
| // 记忆纪律紧跟清单注入:清单告诉模型「有什么」,纪律告诉它「何时必须写」。 | ||||||||||||||||||||
| val full = listOf(content, memoryDiscipline()).mapNotNull { it }.joinToString("\n\n") | ||||||||||||||||||||
| if (full.isEmpty()) return null | ||||||||||||||||||||
|
|
||||||||||||||||||||
| cachedByKey[key] = full | ||||||||||||||||||||
| trimIfNeeded() | ||||||||||||||||||||
| return content | ||||||||||||||||||||
| return full | ||||||||||||||||||||
| } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| /** 记忆纪律正文(无记忆清单时单独注入)。 */ | ||||||||||||||||||||
| private fun memoryDiscipline(): String? = | ||||||||||||||||||||
| resolvePrompt(MEMORY_DISCIPLINE_FILE).replace(LEADING_COMMENT, "").trim().ifEmpty { null } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| private fun trimIfNeeded() { | ||||||||||||||||||||
| if (cachedByKey.size > SOURCE_CACHE_LIMIT) cachedByKey.clear() | ||||||||||||||||||||
| } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| fun invalidate(key: SourceCacheKey) { | ||||||||||||||||||||
| cachedByKey.remove(key) | ||||||||||||||||||||
| } | ||||||||||||||||||||
| } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| /** 会话级缓存 key:同一会话同一工作区共享一份快照,避免每轮重扫磁盘导致 system prompt 变化。 */ | ||||||||||||||||||||
|
|
@@ -397,22 +411,15 @@ class SystemPromptProvider @Inject constructor( | |||||||||||||||||||
| private fun currentDate(): String = | ||||||||||||||||||||
| java.time.ZonedDateTime.now().format(java.time.format.DateTimeFormatter.ofPattern("yyyy-MM-dd")) | ||||||||||||||||||||
|
|
||||||||||||||||||||
| /** | ||||||||||||||||||||
| * 按优先级解析单个提示词片段: | ||||||||||||||||||||
| * - 名字是顶层 `<NN>-*.md`:先按数字身份在 `prompts.custom/` 顶层找覆盖(尾部名称可自由改), | ||||||||||||||||||||
| * - 其余名字(含 `agent/` 子目录):按精确同名在 `prompts.custom/<name>` 找覆盖; | ||||||||||||||||||||
| * 再落到 `prompts/<name>`(本地默认副本),最后 assets(内置兜底)。 | ||||||||||||||||||||
| * | ||||||||||||||||||||
| * 本地副本由 [ContainerInstaller.extractPrompts] 在启动时全量释放,App 升级后随之更新。 | ||||||||||||||||||||
| */ | ||||||||||||||||||||
| fun resolvePrompt(name: String): String { | ||||||||||||||||||||
| PromptFragmentResolver.parseNumber(name) | ||||||||||||||||||||
| ?.let { number -> readFileOrNull(customFragmentsByNumber[number])?.let { return it } } | ||||||||||||||||||||
| readFileOrNull(File(customDir, name))?.let { return it } | ||||||||||||||||||||
| readFileOrNull(File(File(containerInstaller.aicodeDir, "prompts"), name))?.let { return it } | ||||||||||||||||||||
| return context.assets.open("prompts/$name").bufferedReader().use { it.readText() } | ||||||||||||||||||||
| /** 按优先级解析单个提示词片段,见 [PromptFileResolver.resolve]。保留本方法以兼容现有调用点。 */ | ||||||||||||||||||||
| fun resolvePrompt(name: String): String = promptFileResolver.resolve(name) | ||||||||||||||||||||
|
|
||||||||||||||||||||
| /** 失效记忆清单的会话级缓存:curator 写入新记忆后调用,让下一轮 system prompt 看到新清单。 */ | ||||||||||||||||||||
| fun invalidateMemoryCache(sessionId: String?, projectRoot: String?) { | ||||||||||||||||||||
| memoryListSource.invalidate(SourceCacheKey(sessionId, projectRoot)) | ||||||||||||||||||||
| } | ||||||||||||||||||||
|
|
||||||||||||||||||||
| /** 直接读本地文件内容;失败返回 null。供静态基线与自定义片段合并时使用。 */ | ||||||||||||||||||||
| private fun readFileOrNull(file: File?): String? { | ||||||||||||||||||||
| if (file == null || !file.isFile) return null | ||||||||||||||||||||
| return try { | ||||||||||||||||||||
|
|
@@ -428,6 +435,7 @@ class SystemPromptProvider @Inject constructor( | |||||||||||||||||||
| const val AGENTS_FILE = "AGENTS.md" | ||||||||||||||||||||
| const val CLAUDE_FILE = "CLAUDE.md" | ||||||||||||||||||||
| const val SUBAGENT_BASE_FILE = "agent/subagent-base.md" | ||||||||||||||||||||
| const val MEMORY_DISCIPLINE_FILE = "agent/memory-discipline.md" | ||||||||||||||||||||
| const val MAX_AGENTS_CHARS = 32_000 | ||||||||||||||||||||
| /** 会话级缓存 key 数量上限:超过后整体清空,仅防长期累积;正常会话数远小于此。 */ | ||||||||||||||||||||
| const val SOURCE_CACHE_LIMIT = 32 | ||||||||||||||||||||
|
|
||||||||||||||||||||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🩺 Stability & Availability | 🟡 Minor | ⚡ Quick win
connect()now swallows failures, butapplyProfilestill depends on the exception to log errors.SettingsViewModel.applyProfilewrapsremoteSshConnection.connect(...)inrunCatching { }.onFailure { ... }.connectno longer throws. TheonFailurebranch is now dead, so profile-switch failures only show up inconnectionState. The catch at Line 106 also catchesCancellationException, becauseExceptionis its supertype. When the coroutine is cancelled, the state becomesFAILEDand cancellation does not propagate. RethrowCancellationExceptionbefore the state update.Proposed fix
} catch (e: Exception) { + if (e is kotlinx.coroutines.CancellationException) throw e _connectionState.value = ConnectionState.FAILEDThis follows the retrieved learning: CancellationException must be rethrown in coroutine code.
🤖 Prompt for AI Agents
Source: Learnings