mirror of
https://wget.la/https://github.com/Wxw-Gu/WechatExplorer
synced 2026-10-04 03:01:42 +08:00
Wxw-Gu
ab3b3731b7
统一问问微信与 Agent Hub 的查询链路 完善查询覆盖度、新鲜度和部分结果表达，避免索引滞后产生错误结论 基于会话实现真正的增量追新与历史补齐 支持后台同步、取消恢复、重启续传以及同步期间继续查询 优化知识库跨会话检索、同步状态、进度展示和侧栏布局
41 lines
1.8 KiB
TypeScript
41 lines
1.8 KiB
TypeScript
import { describe, expect, it, vi } from 'vitest'
|
|
import { createLocalQueryToolExecutor } from '../../src/main/services/local-query-tool-executor'
|
|
import type { LocalQueryApiService } from '../../src/main/services/local-query-api-service'
|
|
|
|
function fakeQueryApi(): LocalQueryApiService {
|
|
return {
|
|
messages: vi.fn(async () => ({ status: 'completed', returnedCount: 0 })),
|
|
search: vi.fn(async () => ({ status: 'completed', evidenceCount: 0 })),
|
|
context: vi.fn(async () => ({ status: 'completed' })),
|
|
overview: vi.fn(async () => ({ status: 'completed' }))
|
|
} as unknown as LocalQueryApiService
|
|
}
|
|
|
|
describe('createLocalQueryToolExecutor', () => {
|
|
it('把四个 Tool 映射到对应的 Local Query API 能力', async () => {
|
|
const api = fakeQueryApi()
|
|
const execute = createLocalQueryToolExecutor(api)
|
|
|
|
await execute('query_messages', { target: { query: 'BOBO' }, timeRange: { kind: 'all' } })
|
|
await execute('search_messages', { query: '房租' })
|
|
await execute('message_context', { messageRef: 'ref' })
|
|
await execute('conversation_overview', { target: { query: '群' }, timeRange: { kind: 'all' } })
|
|
|
|
expect(api.messages).toHaveBeenCalledWith({ target: { query: 'BOBO' }, timeRange: { kind: 'all' } })
|
|
expect(api.search).toHaveBeenCalledWith({ query: '房租' })
|
|
expect(api.context).toHaveBeenCalledWith({ messageRef: 'ref' })
|
|
expect(api.overview).toHaveBeenCalledWith({
|
|
target: { query: '群' },
|
|
timeRange: { kind: 'all' }
|
|
})
|
|
})
|
|
|
|
it('拒绝未知工具名（不静默落到任意查询）', async () => {
|
|
const api = fakeQueryApi()
|
|
const execute = createLocalQueryToolExecutor(api)
|
|
|
|
await expect(execute('raw_sql', {})).rejects.toThrow('不允许的工具')
|
|
expect(api.messages).not.toHaveBeenCalled()
|
|
})
|
|
})
|