mirror of
https://wget.la/https://github.com/Wxw-Gu/WechatExplorer
synced 2026-10-04 03:01:42 +08:00
Wxw-Gu
b97e1f4bc6
- 问问微信：Host 分配稳定 citationId 并注入模型上下文，正文 [E#] 与证据按钮同号 - 问问微信：回答返回前校验引用，幻觉编号移除并提示，不再出现不可信的引用按钮 - Agent Hub：收紧「最近会话」快捷路由，内容查询交回 Query Agent
107 lines
4.0 KiB
TypeScript
107 lines
4.0 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import {
|
|
askWechatToolLabels,
|
|
formatAskWechatStats,
|
|
mapAskWechatEvidence
|
|
} from '../../src/renderer/src/components/search/askWechatPresentation'
|
|
import type { AskWechatStats } from '../../src/shared/query-agent'
|
|
|
|
const stats = (patch: Partial<AskWechatStats> = {}): AskWechatStats => ({
|
|
tools: ['query_messages'],
|
|
reads: { messageCount: 0, matchedCount: 0, overviewSourceCount: 0, evidenceCount: 0 },
|
|
scope: { kind: 'all', label: '所有聊天记录' },
|
|
modelCallCount: 2,
|
|
toolCallCount: 1,
|
|
totalMs: 3343,
|
|
...patch
|
|
})
|
|
|
|
describe('mapAskWechatEvidence', () => {
|
|
it('群消息证据保留群名与成员，且不泄露 md5', () => {
|
|
const [item] = mapAskWechatEvidence([
|
|
{
|
|
citationId: 'E1',
|
|
messageRef: 'ref-1',
|
|
conversationName: 'TraceMemo 交流群',
|
|
conversationType: 'group',
|
|
sender: '张三',
|
|
timestamp: 1_787_650_302_000,
|
|
messageType: 'text',
|
|
text: '最近重新开始健身了',
|
|
source: 'search_messages'
|
|
}
|
|
])
|
|
expect(item.evidenceId).toBe('E1')
|
|
expect(item.contact.m_nsNickName).toBe('TraceMemo 交流群')
|
|
expect(item.contact.type).toBe('group')
|
|
expect(item.message.name).toBe('张三')
|
|
expect(item.message.content).toBe('最近重新开始健身了')
|
|
expect(item.message.createTime).toBe(1_787_650_302)
|
|
expect(item.contact.md5).not.toMatch(/^[0-9a-f]{32}$/)
|
|
})
|
|
|
|
it('证据编号直接消费 Host 的 citationId，不用数组下标合成', () => {
|
|
const mapped = mapAskWechatEvidence([
|
|
{ citationId: 'E3', messageRef: 'ref-3', source: 'search_messages' },
|
|
{ citationId: 'E7', messageRef: 'ref-7', source: 'search_messages' }
|
|
])
|
|
// 下标是 0/1；若还在自行编号就会得到 E1/E2 —— 那会让正文 [E3] 指向别的证据。
|
|
expect(mapped.map((entry) => entry.evidenceId)).toEqual(['E3', 'E7'])
|
|
})
|
|
|
|
it('缺少群名时给出可读回退，而不是把群消息归成"群聊"', () => {
|
|
const [item] = mapAskWechatEvidence([
|
|
{ citationId: 'E1', messageRef: 'ref-2', source: 'conversation_overview' }
|
|
])
|
|
expect(item.contact.m_nsNickName).toBe('未命名会话')
|
|
})
|
|
})
|
|
|
|
describe('formatAskWechatStats', () => {
|
|
it('query_messages：读取条数 + 证据条数 + 模型调用 + 耗时', () => {
|
|
const chips = formatAskWechatStats(
|
|
stats({
|
|
tools: ['query_messages'],
|
|
reads: { messageCount: 20, matchedCount: 0, overviewSourceCount: 0, evidenceCount: 20 }
|
|
})
|
|
)
|
|
expect(chips).toContain('所有聊天记录内查询')
|
|
expect(chips).toContain('读取 20 条消息')
|
|
expect(chips).toContain('使用 20 条证据')
|
|
expect(chips).toContain('2 次模型调用')
|
|
expect(chips).toContain('3.3 s')
|
|
})
|
|
|
|
it('overview：覆盖源消息 + 代表证据；search 无证据时说明命中为 0 的事实', () => {
|
|
const overview = formatAskWechatStats(
|
|
stats({
|
|
tools: ['conversation_overview'],
|
|
scope: { kind: 'current', label: '当前会话：TraceMemo 交流群' },
|
|
reads: { messageCount: 0, matchedCount: 0, overviewSourceCount: 1200, evidenceCount: 60 }
|
|
})
|
|
)
|
|
expect(overview).toContain('当前会话：TraceMemo 交流群内查询')
|
|
expect(overview).toContain('覆盖 1200 条消息')
|
|
expect(overview).toContain('使用 60 条证据')
|
|
|
|
const search = formatAskWechatStats(
|
|
stats({
|
|
tools: ['search_messages'],
|
|
scope: { kind: 'groups', label: '群聊专属' },
|
|
reads: { messageCount: 0, matchedCount: 12, overviewSourceCount: 0, evidenceCount: 0 }
|
|
})
|
|
)
|
|
expect(search).toContain('命中 12 条相关消息')
|
|
expect(search.some((chip) => chip.includes('知识库已收录'))).toBe(false)
|
|
})
|
|
})
|
|
|
|
describe('askWechatToolLabels', () => {
|
|
it('把 Tool 名映射成中文能力标签并去重', () => {
|
|
expect(askWechatToolLabels(['query_messages', 'search_messages', 'search_messages'])).toEqual([
|
|
'精确读取',
|
|
'关键词检索'
|
|
])
|
|
})
|
|
})
|