test: 暂存代码

2026-08-17 19:47:08 +08:00 · 2026-08-06 20:29:25 +08:00
parent 307d247660
commit ad4b3a8074
43 changed files with 9978 additions and 512 deletions
@@ -0,0 +1,77 @@
+import { describe, expect, it } from 'vitest'
+import {
+  buildLocalAiSearchPlan,
+  includesExplicitAiSearchAlias,
+  inferAiSearchTimeRange
+} from '../../src/shared/ai-search'
+
+const NOW = new Date('2026-08-05T12:00:00+08:00')
+
+describe('AI search natural-language time ranges', () => {
+  it('tightens an all-history selection when the user says 最近', () => {
+    expect(inferAiSearchTimeRange('我和张三最近聊了什么？', 'all', NOW)).toMatchObject({
+      label: '近 30 天',
+      source: 'query',
+      startTime: Math.floor(NOW.getTime() / 1000) - 30 * 86400
+    })
+  })
+
+  it('recognizes explicit recent days and the current year', () => {
+    expect(inferAiSearchTimeRange('我和张三最近三天聊了什么？', 'all', NOW)).toMatchObject({
+      label: '近 3 天',
+      source: 'query'
+    })
+    expect(inferAiSearchTimeRange('我和张三今年聊了什么？', 'all', NOW)).toMatchObject({
+      label: '今年',
+      startTime: Math.floor(new Date(2026, 0, 1).getTime() / 1000)
+    })
+  })
+
+  it('keeps an explicit user retry override above the word 最近 in the original question', () => {
+    expect(
+      inferAiSearchTimeRange('我和张三最近聊了什么？', 'all', NOW, {
+        label: '全部历史',
+        reason: '用户主动扩大到全部历史',
+        source: 'user_retry'
+      })
+    ).toMatchObject({
+      label: '全部历史',
+      source: 'user_retry'
+    })
+  })
+
+  it('classifies a direct person recap as conversation_recall rather than a topic FTS query', () => {
+    expect(buildLocalAiSearchPlan('我和张三最近聊了什么？')).toMatchObject({
+      intent: 'conversation_recall',
+      contactQuery: '张三'
+    })
+  })
+
+  it('keeps identity and message topic separate for a contact topic search', () => {
+    expect(buildLocalAiSearchPlan('我和张三最近聊过健身吗？')).toMatchObject({
+      intent: 'conversation_topic_search',
+      contactQuery: '张三',
+      topicQuery: '健身',
+      keywords: ['健身']
+    })
+  })
+
+  it('classifies global topics and bare conversation names without turning names into FTS terms', () => {
+    expect(buildLocalAiSearchPlan('最近谁聊过 MCP？')).toMatchObject({
+      intent: 'global_topic_search',
+      topicQuery: 'MCP',
+      keywords: ['MCP']
+    })
+    expect(buildLocalAiSearchPlan('技术交流群')).toMatchObject({
+      intent: 'conversation_name_search',
+      contactQuery: '技术交流群',
+      topicQuery: undefined
+    })
+  })
+
+  it('matches an explicitly mentioned nickname when the user omits punctuation', () => {
+    expect(includesExplicitAiSearchAlias('我和中田健身弘毅最近聊了什么？', '中田健身-弘毅')).toBe(
+      true
+    )
+  })
+})