mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
74 lines
3.1 KiB
TypeScript
74 lines
3.1 KiB
TypeScript
import { Context } from 'cordis'
|
|
import { describe, expect, it } from 'vitest'
|
|
import LlmService, { LlmAdapter } from '@deepseek-ai/dsh-llm'
|
|
import type { GenerateOptions, StreamChunk } from '@deepseek-ai/dsh-llm'
|
|
import SessionStore, { Session, SessionId } from '@deepseek-ai/dsh-session'
|
|
import SessionTitleService from '@deepseek-ai/dsh-session-title'
|
|
import * as providerPlugin from '@deepseek-ai/dsh-session-title-all-messages-llm'
|
|
|
|
class RecordingAdapter extends LlmAdapter {
|
|
readonly requests: GenerateOptions[] = []
|
|
|
|
override async * stream(options: GenerateOptions): AsyncIterable<StreamChunk> {
|
|
this.requests.push(options)
|
|
yield { type: 'text-delta', index: 0, text: 'All messages model title' }
|
|
yield { type: 'finish', reason: { kind: 'stop' } }
|
|
}
|
|
}
|
|
|
|
const TITLE_CONFIG = { fallbackMaxWords: 5, fallbackMaxBytes: 40, maxTitleBytes: 80 } as const
|
|
const LLM_CONFIG = {
|
|
targetWords: 5,
|
|
targetCjkCharacters: 10,
|
|
maxInputBytes: 1_000,
|
|
maxOutputTokens: 32,
|
|
timeoutMs: 1_000,
|
|
} as const
|
|
|
|
async function settle(): Promise<void> {
|
|
await new Promise(resolve => setTimeout(resolve, 0))
|
|
}
|
|
|
|
describe('all-messages LLM title provider', () => {
|
|
it('includes seeded history and the latest prompt while inheriting the logged request route', async () => {
|
|
const seeded = new Session(SessionId('seed-source'))
|
|
seeded.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
|
|
const inherited = seeded.append('user/message', {
|
|
content: [{ type: 'text', text: 'inherited prompt' }], source: { kind: 'user' },
|
|
}, { surfaceOp: 'append' })
|
|
seeded.append('session/title', {
|
|
title: 'Inherited fallback', messageSeqs: [inherited.seq], source: { kind: 'fallback' },
|
|
})
|
|
seeded.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
|
|
|
|
const ctx = new Context()
|
|
await ctx.plugin(LlmService)
|
|
await ctx.plugin(SessionStore)
|
|
await ctx.plugin(SessionTitleService, TITLE_CONFIG)
|
|
const adapter = new RecordingAdapter()
|
|
ctx.llm.registerAdapter(['current-route'], adapter)
|
|
await ctx.plugin(providerPlugin, LLM_CONFIG)
|
|
const session = ctx.sessions.create(SessionId('all-plugin'), {
|
|
seed: seeded.events,
|
|
meta: { parentSession: seeded.id, seedLength: seeded.seq },
|
|
})
|
|
session.append('turn/start', { turn: 2, trigger: { kind: 'message', source: { kind: 'user' } } })
|
|
const latest = session.append('user/message', {
|
|
content: [{ type: 'text', text: 'latest prompt' }], source: { kind: 'user' },
|
|
}, { surfaceOp: 'append' })
|
|
await settle()
|
|
session.append('request/header', {
|
|
header: { config: { provider: 'current-route', model: 'current-model' } }, reason: 'resume',
|
|
})
|
|
await settle()
|
|
|
|
expect(adapter.requests[0]).toMatchObject({ provider: 'current-route', model: 'current-model' })
|
|
const content = adapter.requests[0]?.messages[0]?.content[0]
|
|
expect(content?.type === 'text' && content.text).toContain('inherited prompt')
|
|
expect(content?.type === 'text' && content.text).toContain('latest prompt')
|
|
expect(ctx.sessionTitle.get(session)).toMatchObject({
|
|
messageSeqs: [inherited.seq, latest.seq],
|
|
})
|
|
})
|
|
})
|