fix: 增加thinking disabled

修改退群监控模板
定时日报保存日报历史
This commit is contained in:
Wxw-Gu
2026-09-01 11:40:14 +08:00
parent 25b72c3d76
commit 1c8bbae1f0
21 changed files with 343 additions and 34 deletions
@@ -37,6 +37,44 @@ const provider = (id: string, modelId: string, vision: boolean): AIProviderConfi
describe('AI provider vision routing', () => {
afterAll(() => rmSync(root, { recursive: true, force: true }))
it('sends the disabled thinking control only when it is selected', async () => {
const service = new AIProviderService()
const configured = provider('thinking-provider', 'deepseek-v4-flash-vision-exp', false)
configured.advanced.thinking = 'disabled'
expect(service.save(configured).success).toBe(true)
expect(service.setDefault(configured.id).success).toBe(true)
const fetchMock = vi
.fn()
.mockResolvedValue(
new Response(
JSON.stringify({ choices: [{ message: { content: 'ok' }, finish_reason: 'stop' }] }),
{ headers: { 'content-type': 'application/json' } }
)
)
vi.stubGlobal('fetch', fetchMock)
try {
const result = await service.chat([{ role: 'user', content: 'hello' }])
expect(result).toMatchObject({ success: true, data: 'ok', finishReason: 'stop' })
const body = JSON.parse(fetchMock.mock.calls[0][1].body as string) as Record<string, unknown>
expect(body).toMatchObject({
model: 'deepseek-v4-flash-vision-exp',
thinking: { type: 'disabled' }
})
configured.advanced.thinking = 'default'
expect(service.save(configured).success).toBe(true)
await service.chat([{ role: 'user', content: 'hello' }])
const defaultBody = JSON.parse(fetchMock.mock.calls[1][1].body as string) as Record<
string,
unknown
>
expect(defaultBody).not.toHaveProperty('thinking')
} finally {
vi.unstubAllGlobals()
}
})
it('keeps DeepSeek as the text model while routing images to a verified vision model', () => {
const service = new AIProviderService()
expect(service.save(provider('deepseek', 'deepseek-chat', false)).success).toBe(true)
+13 -1
View File
@@ -1,4 +1,4 @@
import { mkdtempSync, readJsonSync, rmSync } from 'fs-extra'
import { mkdtempSync, readJsonSync, rmSync, writeJsonSync } from 'fs-extra'
import { tmpdir } from 'os'
import { join } from 'path'
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
@@ -32,6 +32,7 @@ vi.mock('../../src/main/services/personal-wechat-send-service', () => ({
}))
import { GroupExitMonitorService } from '../../src/main/services/group-exit-monitor-service'
import { GROUP_EXIT_NOTIFICATION_TEMPLATE } from '../../src/shared/group-exit-monitor'
const member = { wxid: 'wxid_member', wechatNickname: '微信名', groupNickname: '群内名' }
@@ -77,6 +78,17 @@ describe('GroupExitMonitorService', () => {
expect(() => service.setNotificationTemplate('用户: {unknown}')).toThrow('不支持的占位符')
})
it('migrates the previous default notification template', () => {
writeJsonSync(join(mocks.userData, 'group-exit-monitor.json'), {
notificationTemplate:
'[退群监测]\n\n用户: {user}\n\n群备注: {groupRemark}\n\n微信号: {wxid}\n\n退群时间: {time}'
})
const service = new GroupExitMonitorService()
expect(service.getState().notificationTemplate).toBe(GROUP_EXIT_NOTIFICATION_TEMPLATE)
})
it('keeps the previous baseline when a native snapshot contains null members', async () => {
installGroupDb()
mocks.chat.getGroupSnapshotAsync.mockResolvedValueOnce({
+3 -2
View File
@@ -65,6 +65,7 @@ describe('group exit monitor member diff', () => {
expect(message).toContain('用户: 微信名')
expect(message).toContain('群备注: 未设置')
expect(message).toContain('微信号: wxid_fixture')
expect(message).toContain('人数: 240 -> 239')
expect(message).toContain('退群时间: 2026-08-31 00:19:07')
})
@@ -97,8 +98,8 @@ describe('group exit monitor member diff', () => {
message: '群昵称退出了测试群',
detectedAt: new Date(2026, 7, 31, 0, 19, 7).getTime()
},
'{user}|{groupRemark}|{wxid}|{time}'
'{user}|{groupRemark}|{wxid}|{previousCount}|{currentCount}|{time}'
)
expect(message).toBe('微信名|群内昵称|wxid_fixture|2026-08-31 00:19:07')
expect(message).toBe('微信名|群内昵称|wxid_fixture|2|1|2026-08-31 00:19:07')
})
})
+22
View File
@@ -1,6 +1,7 @@
import { afterEach, describe, expect, it, vi } from 'vitest'
import {
buildGroupReportInput,
getSummaryDateRangeAt,
parseGroupDailyReport
} from '../../src/renderer/src/utils/group-report'
import { summaryContent } from '../../src/renderer/src/utils/group-report-facts'
@@ -324,3 +325,24 @@ describe('group report parsing', () => {
expect(input.media.gallery).toEqual([])
})
})
describe('summary date ranges', () => {
it('uses local calendar days for today, yesterday, and seven-day ranges', () => {
const now = new Date(2026, 0, 1, 12, 34, 56)
const today = new Date(2026, 0, 1)
const yesterday = new Date(2025, 11, 31)
const sevenDaysAgo = new Date(2025, 11, 26)
expect(getSummaryDateRangeAt('today', now)).toEqual({
startTime: today.getTime() / 1000,
endTime: now.getTime() / 1000
})
expect(getSummaryDateRangeAt('yesterday', now)).toEqual({
startTime: yesterday.getTime() / 1000,
endTime: today.getTime() / 1000 - 1
})
expect(getSummaryDateRangeAt('7days', now)).toEqual({
startTime: sevenDaysAgo.getTime() / 1000,
endTime: now.getTime() / 1000
})
})
})
@@ -71,6 +71,7 @@ describe('generated report template history', () => {
dateRange: '今天',
messageCount: 10,
generatedAt: '2026-08-12T10:00:00.000Z',
reportDate: '2026-08-11',
generatedImage: `data:image/png;base64,${Buffer.from('mobile-01').toString('base64')}`,
htmlPath: originalHtml,
reportSnapshot,
@@ -79,6 +80,7 @@ describe('generated report template history', () => {
})
expect(saved.success).toBe(true)
expect(saved.record).toBeDefined()
expect(saved.record?.reportDate).toBe('2026-08-11')
const updated = await updateGeneratedReportTemplate({
reportId: saved.record!.id,
@@ -34,6 +34,7 @@ describe('scheduled report scheduling', () => {
const generatedPath = join(storageDir, 'report.png')
let generated = 0
let sent = 0
const saveHistory = vi.fn().mockResolvedValue({ success: true })
const service = new ScheduledReportService({
storageDir,
now: () => now,
@@ -46,6 +47,7 @@ describe('scheduled report scheduling', () => {
sent += 1
return { success: true, status: capability.senderStatus }
},
saveGeneratedReport: saveHistory,
isDatabaseReady: () => true
})
const created = await service.createTask({
@@ -61,6 +63,14 @@ describe('scheduled report scheduling', () => {
expect(run.data?.status).toBe('success')
expect(generated).toBe(1)
expect(sent).toBe(1)
expect(saveHistory).toHaveBeenCalledWith(
expect.objectContaining({
contactName: '研发群',
reportDate: undefined,
pngPath: generatedPath,
messageCount: 0
})
)
expect(await service.listExecutions(taskId)).toHaveLength(1)
const restored = new ScheduledReportService({
storageDir,
@@ -75,4 +85,30 @@ describe('scheduled report scheduling', () => {
JSON.parse(await readFile(join(storageDir, 'executions.json'), 'utf8'))[0].message
).toContain('微信发送成功')
})
it('does not send a generated report when history persistence fails', async () => {
const storageDir = await mkdtemp(join(tmpdir(), 'tracememo-scheduled-report-history-'))
const generatedPath = join(storageDir, 'report.png')
const send = vi.fn().mockResolvedValue({ success: true, status: capability.senderStatus })
const service = new ScheduledReportService({
storageDir,
getCapability: async () => capability,
generateReport: async () => ({ success: true, pngPath: generatedPath }),
saveGeneratedReport: vi.fn().mockResolvedValue({ success: false, error: '磁盘不可写' }),
send,
isDatabaseReady: () => true
})
const created = await service.createTask({
name: '历史失败日报',
group: '研发群',
target: '研发群@chatroom',
scheduleTime: '09:00'
})
const result = await service.runScheduledReportNow(created.data!.id)
expect(result.success).toBe(false)
expect(result.data?.error).toContain('report_history_save_failed:磁盘不可写')
expect(send).not.toHaveBeenCalled()
})
})