From c53cf9235f177624bc29b32768c85fca56c99230 Mon Sep 17 00:00:00 2001 From: yibo Date: Wed, 10 Jun 2026 17:17:38 +0800 Subject: [PATCH] feat(tools): refresh_person_data 工具 + 查人前自动刷新微信数据 --- lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ChatViewModel.kt | 11 ++++++++--- lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/LifelogTools.kt | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ModelConfig.kt | 3 ++- 3 files changed, 69 insertions(+), 4 deletions(-) diff --git a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ChatViewModel.kt b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ChatViewModel.kt index c6a8f92..59df171 100644 --- a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ChatViewModel.kt +++ b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ChatViewModel.kt @@ -303,15 +303,20 @@ class ChatViewModel(app: Application) : AndroidViewModel(app) { ⑰ get_person_photos — 某人出现的照片（args: person_name, limit默认10） ⑱ get_followups — 待跟进事项（args: 无） + 【数据刷新】 + ⑲ refresh_person_data — 刷新某人微信聊天+朋友圈（args: person_name） + ★ 查询某人聊天/朋友圈/时间线前，必须先调此工具确保数据最新！ + 【全网搜索】 - ⑲ web_search — Tavily 搜索互联网（args: query, max_results默认5；需设置中配置 Key） + ⑳ web_search — Tavily 搜索互联网（args: query, max_results默认5；需设置中配置 Key） 需要查询时，输出（每个工具独占一行，严格 JSON）： TOOL_CALL: {"name": "工具名", "args": {"参数": "值"}} - 示例： - TOOL_CALL: {"name": "search_transcripts", "args": {"query": "火星人", "limit": 5}} + 示例（查某人时，先刷新再查）： + TOOL_CALL: {"name": "refresh_person_data", "args": {"person_name": "张三"}} TOOL_CALL: {"name": "get_person", "args": {"name": "张三"}} + TOOL_CALL: {"name": "get_person_timeline", "args": {"person_name": "张三"}} 收到 TOOL_RESULT 后继续分析，可再次调用工具。 TOOL_CALL 行由系统截取执行，最终答案里不要重复输出。最终答案用中文，简洁清晰。 diff --git a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/LifelogTools.kt b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/LifelogTools.kt index 9ccf226..92ce914 100644 --- a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/LifelogTools.kt +++ b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/LifelogTools.kt @@ -1,7 +1,12 @@ package com.lemon.lifelog.chat import android.content.Context +import com.lemon.lifelog.data.Moment import com.lemon.lifelog.data.Repository +import com.lemon.lifelog.data.WechatMessage +import com.lemon.lifelog.vision.WeChatChatImporter +import com.lemon.lifelog.vision.WeChatSnsImporter +import java.io.File import kotlinx.coroutines.Dispatchers import kotlinx.coroutines.flow.first import kotlinx.coroutines.withContext @@ -204,6 +209,14 @@ object LifelogTools { required = listOf("person_name"), )) put(schema( + name = "refresh_person_data", + desc = "刷新某人的微信聊天 + 朋友圈数据（从设备上解密读取最新内容），查询前先调此接口可确保数据最新", + props = mapOf( + "person_name" to prop("string", "人物姓名"), + ), + required = listOf("person_name"), + )) + put(schema( name = "web_search", desc = "用 Tavily 搜索公开互联网内容（需在设置中配置 Tavily API Key）", props = mapOf( @@ -256,6 +269,7 @@ object LifelogTools { "search_by_time" -> searchByTime(repo, args) "get_conversation_context" -> getConversationContext(repo, args) "get_person_timeline" -> getPersonTimeline(repo, args) + "refresh_person_data" -> refreshPersonData(repo, ctx, args) "web_search" -> webSearch(ctx, args) else -> "未知工具：$name" } @@ -627,6 +641,51 @@ object LifelogTools { } } + // ── 刷新某人数据 ────────────────────────────────────────────────────────── + + private suspend fun refreshPersonData(repo: Repository, ctx: Context, args: JSONObject): String { + val name = args.optString("person_name", "").ifBlank { return "请提供人物姓名" } + val person = repo.persons.all().firstOrNull { p -> + p.name.contains(name) || p.realName?.contains(name) == true || + p.nickname?.contains(name) == true || p.remark?.contains(name) == true + } ?: return "未找到与「$name」匹配的联系人。" + + val wxid = person.contactRef?.removePrefix("wechat:") + ?.takeIf { person.contactRef!!.startsWith("wechat:") } + ?: return "「${person.name}」不是微信联系人，录音数据已是实时的，无需刷新。" + + // 1. 刷新聊天 + val chatN = runCatching { + val r = WeChatChatImporter.importChats(ctx, setOf(wxid)) + var total = 0 + for ((id, msgs) in r.byWxid) { + if (msgs.isEmpty()) continue + repo.wechatMessages.clearTalker(id) + repo.wechatMessages.insertAll( + msgs.map { WechatMessage(talker = id, content = it.content, createTime = it.createTime, isSend = it.isSend) }, + ) + total += msgs.size + } + total + }.getOrElse { "聊天刷新失败：${it.message?.take(80)}" } + + // 2. 刷新朋友圈 + val momentN = runCatching { + val momentsDir = File(ctx.filesDir, "moments/$wxid").apply { deleteRecursively(); mkdirs() } + val r = WeChatSnsImporter.importMomentsFor(ctx, wxid, momentsDir) + if (r.moments.isNotEmpty()) { + repo.moments.clearUser(wxid) + repo.moments.insertAll( + r.moments.map { Moment(userName = wxid, text = it.text, createTime = it.createTimeMs) }, + ) + r.moments.size + } else 0 + }.getOrElse { "朋友圈刷新失败：${it.message?.take(80)}" } + + repo.persons.touch(person.id, System.currentTimeMillis()) + return "已刷新「${person.name}」：聊天 $chatN 条 + 朋友圈 $momentN 条（录音数据实时，照片后台自动更新）" + } + // ── 全网搜索（Tavily）──────────────────────────────────────────────────── private fun webSearch(ctx: Context, args: JSONObject): String { diff --git a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ModelConfig.kt b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ModelConfig.kt index 0c5bb57..9076f78 100644 --- a/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ModelConfig.kt +++ b/lifelog-recorder/app/src/main/kotlin/com/lemon/lifelog/chat/ModelConfig.kt @@ -68,7 +68,8 @@ object ModelConfig { } fun tavilyKey(ctx: Context): String = - prefs(ctx).getString(K_TAVILY_KEY, null)?.trim().orEmpty() + prefs(ctx).getString(K_TAVILY_KEY, null)?.ifBlank { null } + ?: "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" fun saveTavilyKey(ctx: Context, key: String) { prefs(ctx).edit().putString(K_TAVILY_KEY, key.trim()).apply() -- libgit2 0.21.4