mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
ec601ca13d
Machine-produced by `pnpm run rescope-vendor --apply` plus the regeneration it prints: `pnpm install` for the lockfile, `pnpm run gen-third-party-notices`, `verify-translation-pairing --write` for the touched bilingual pairs, `gen-doc-graphs`, and one typert snapshot whose ids embed character offsets. `pnpm run rescope-vendor --check` verifies the result. Renames nine vendored packages (cordis, cosmokit, schemastery and the six @cordisjs plugins) and every reference that resolves them: manifest names and dependency keys, module specifiers including declare-module merges, cordis.yml plugin names, tsconfig paths, every Markdown fence, and `docs/` prose. Directory names, upstream versions, and dependency ranges are unchanged, so vendor/README.md still reads as an upstream snapshot; its manifest table gains an upstream-name column so THIRD_PARTY_NOTICES keeps MIT attribution pointed at each fork's origin. The tutorial tier follows the rename end to end: its yaml fences named plugins the Loader can no longer resolve, its `ts ignore-check` fences disagreed with the compiled fences beside them, and its prose quoted both. The contracts that told readers to keep upstream names — the root convention and the vendoring cookbook's tree comment and manifest invariant — now say to rescope instead. Two rules read `@deepseek-ai/` as "another workspace plugin": the client bundle purity gate now names the vendored libraries a browser bundle inlines, and the files where a bare `cordis` is an agent-preset id keep that product data.
87 lines
3.5 KiB
TypeScript
87 lines
3.5 KiB
TypeScript
import { Context } from '@deepseek-ai/cordis'
|
|
import { describe, expect, it, vi } from 'vitest'
|
|
import LlmService, { createUserMessage, 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, { type SessionTitleProvider } from '@deepseek-ai/dsh-session-title'
|
|
import * as providerPlugin from '@deepseek-ai/dsh-session-title-first-message-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: 'First-message 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,
|
|
provider: 'title-route',
|
|
model: 'title-model',
|
|
} as const
|
|
|
|
async function settle(): Promise<void> {
|
|
await new Promise(resolve => setTimeout(resolve, 0))
|
|
}
|
|
|
|
describe('first-message LLM title provider', () => {
|
|
it('rejects an impossible empty provider request at its own boundary', async () => {
|
|
const ctx = new Context()
|
|
await ctx.plugin(LlmService)
|
|
await ctx.plugin(SessionStore)
|
|
await ctx.plugin(SessionTitleService, TITLE_CONFIG)
|
|
let registered: SessionTitleProvider | undefined
|
|
vi.spyOn(ctx.sessionTitle, 'register').mockImplementation((provider) => {
|
|
registered = provider
|
|
return async () => undefined
|
|
})
|
|
providerPlugin.apply(ctx, LLM_CONFIG)
|
|
|
|
await expect(registered!.generate({
|
|
session: Session.create(SessionId('empty-first-provider')),
|
|
messages: [],
|
|
signal: new AbortController().signal,
|
|
})).rejects.toThrow(/requires one human message/)
|
|
})
|
|
|
|
it('always selects only the first eligible human message, including explicit refresh', async () => {
|
|
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(['title-route'], adapter)
|
|
await ctx.plugin(providerPlugin, LLM_CONFIG)
|
|
const session = ctx.sessions.create(SessionId('first-plugin'))
|
|
session.append('turn/start', { turn: 1 })
|
|
const first = session.append('user/message', createUserMessage({
|
|
content: [{ type: 'text', text: 'first input' }], source: { kind: 'user' },
|
|
}), { surfaceOp: 'append' })
|
|
await settle()
|
|
session.append('request/header', {
|
|
header: { config: { provider: 'main', model: 'main-model' } }, reason: 'initial',
|
|
})
|
|
await settle()
|
|
session.append('user/message', createUserMessage({
|
|
content: [{ type: 'text', text: 'second input must be ignored' }], source: { kind: 'user' },
|
|
}), { surfaceOp: 'append' })
|
|
|
|
await ctx.sessionTitle.refresh(session)
|
|
|
|
expect(adapter.requests).toHaveLength(2)
|
|
for (const options of adapter.requests) {
|
|
const content = options.messages[0]?.content[0]
|
|
expect(content?.type === 'text' && content.text).toContain('first input')
|
|
expect(content?.type === 'text' && content.text).not.toContain('second input must be ignored')
|
|
}
|
|
expect(ctx.sessionTitle.get(session)).toMatchObject({ messageSeqs: [first.seq] })
|
|
})
|
|
})
|