67 lines
3.1 KiB
TypeScript
67 lines
3.1 KiB
TypeScript
/** Package-owned durable clock-context invariants. @module @deepseek-ai/dsh-time-context/invariant */
|
|
|
|
import type { Context } from 'cordis'
|
|
import type { Session, SessionEvent } from '@deepseek-ai/dsh-session'
|
|
import type { InvariantFailure, InvariantInstaller } from '@deepseek-ai/dsh-invariants'
|
|
|
|
const PACKAGE_NAME = '@deepseek-ai/dsh-time-context'
|
|
const SOURCE_NAME = 'time-context'
|
|
const READING = new RegExp(
|
|
'^Time sampled while preparing turn (\\d+), step (\\d+): '
|
|
+ '(\\d{4}-\\d{2}-\\d{2}T\\d{2}:\\d{2}:\\d{2}(?:Z|[+-]\\d{2}:\\d{2})\\[[^\\]]+\\])\\n'
|
|
+ 'Elapsed since the preceding (model-visible message|step context): '
|
|
+ '(?:unavailable|(?:(?:\\d+d )?(?:\\d+h )?(?:\\d+m )?\\d+s))\\.$',
|
|
)
|
|
|
|
/** Cordis companion plugin name. */
|
|
export const name = 'time-context-invariant'
|
|
/** Service required before the companion can reserve package ownership. */
|
|
export const inject = ['invariants']
|
|
|
|
/** Validate one plugin-attributed time reading against its durable event timestamp. */
|
|
function validateReading(event: SessionEvent<'context/message'>, fail: InvariantFailure): void {
|
|
const [block] = event.data.content
|
|
if (event.data.content.length !== 1 || block?.type !== 'text') {
|
|
fail('time-context messages must contain exactly one text block')
|
|
}
|
|
const match = READING.exec(block.text)
|
|
if (match === null) fail('time-context message does not match the durable reading format')
|
|
const turn = Number(match[1])
|
|
const step = Number(match[2])
|
|
if (!Number.isSafeInteger(turn) || turn < 1 || !Number.isSafeInteger(step) || step < 1) {
|
|
fail('time-context turn and step must be positive safe integers')
|
|
}
|
|
const baseline = match[4]
|
|
if ((step === 1) !== (baseline === 'model-visible message')) {
|
|
fail(`time-context step ${step} uses the wrong elapsed-time baseline ${JSON.stringify(baseline)}`)
|
|
}
|
|
const rendered = match[3]
|
|
/* v8 ignore next -- the preceding fixed regexp always supplies capture group three. */
|
|
if (rendered === undefined) fail('time-context reading omitted its rendered timestamp')
|
|
const renderedTime = Date.parse(rendered.replace(/\[[^\]]+\]$/, ''))
|
|
if (!Number.isFinite(renderedTime) || !Number.isSafeInteger(event.time)
|
|
|| event.time < renderedTime || event.time - renderedTime >= 1_000) {
|
|
fail('time-context rendered timestamp must identify the durable event second')
|
|
}
|
|
}
|
|
|
|
/** Install validation for plugin-attributed context readings. */
|
|
const install: InvariantInstaller = (ctx, fail) => {
|
|
ctx.on('internal/dispatch', (_mode, eventName, args) => {
|
|
if (eventName !== 'session/event') return
|
|
const event = (args as [Session, SessionEvent])[1]
|
|
if (event.type !== 'context/message'
|
|
|| event.data.source.kind !== 'plugin'
|
|
|| event.data.source.plugin !== SOURCE_NAME) return
|
|
validateReading(event, fail)
|
|
}, { global: true })
|
|
}
|
|
|
|
/**
|
|
* Register the time-context invariant companion.
|
|
* @param ctx - Cordis context carrying the invariant service.
|
|
* @returns the installed registration's disposer after setup succeeds.
|
|
*/
|
|
export const apply = (ctx: Context): Promise<() => void> =>
|
|
Promise.resolve(ctx.invariants.register(PACKAGE_NAME, install))
|