SaylorTwift

/

kimi-code

Like 0

import { describe, expect, it } from 'vitest';

import { filterOpsForGrade, isAppendOnly, redactSnapshotForGrade } from '#/granularity/filterOps';

import { detachGrades, gradeFor, needsResetOnTransition } from '#/granularity/grade';

import { paginateTurns } from '#/pagination/paginate';

import { ViewRegistry } from '#/view/registry';

import { projectTranscriptUserOrigin } from '#/contract/origin';

import { groupMessagesIntoSnapshot, type HistoryContentPart } from '#/history/groupTurns';

import { foldWireRecordFacts, type HistoryWireRecord } from '#/history/foldFacts';

import {

transcriptOperationSchema,

transcriptQuerySchema,

transcriptResponseSchema,

transcriptGradeSpecSchema,

transcriptUserOriginSchema,

} from '#/contract/schema';

import type { TranscriptItem } from '#/model/item';

import type { AgentTranscriptSnapshot, TranscriptOperation } from '#/ops/operation';

const idLabel = (i: TranscriptItem): string =>

i.kind === 'turn' ? i.turnId : i.kind === 'marker' ? i.markerId : i.refId;

describe('client metadata in transcript user origins', () => {

it('retains user-invoked single skill frame metadata without exposing model-triggered activations as user input', () => {

const origin = { kind: 'skill_activation', trigger: 'user-slash', skillName: 'example-skill', skillArgs: 'args', clientMetadata: [{ display_text: 'Save button' }] };

expect(transcriptUserOriginSchema.parse(projectTranscriptUserOrigin(origin))).toEqual(origin);

expect(projectTranscriptUserOrigin({ ...origin, trigger: 'model-tool' })).toBeUndefined();

});

it('rebuilds a user turn payload without server-local paths', () => {

const clientMetadata = [{ display_text: 'Visible prompt' }];

const origin = {

kind: 'user',

clientMetadata,

skillActivations: [{ activationId: 'a1', skillName: 'deploy', skillArgs: 'now', skillPath: '/private/deploy/SKILL.md' }],

attachments: [{ name: 'notes.pdf', mediaType: 'application/pdf', size: 42, path: '/private/notes.pdf' }],

};

const snapshot = groupMessagesIntoSnapshot([

{ role: 'user', content: [{ type: 'text', text: 'rendered skill' }, { type: 'text', text: 'visible prompt' }], toolCalls: [], origin },

{ role: 'assistant', content: [{ type: 'text', text: 'reply' }], toolCalls: [] },

]);

const turn = snapshot.items.find((item) => item.kind === 'turn');

expect(turn?.origin).toEqual({ kind: 'user', payload: { kind: 'user', clientMetadata, skillActivations: [{ skillName: 'deploy', skillArgs: 'now' }] } });

expect(JSON.stringify(turn)).not.toContain('/private/');

expect(JSON.stringify(snapshot.attachments)).not.toContain('/private/');

});

it('keeps opening prompt metadata when rebuilding history turns', () => {

const clientMetadata = [{ kimi_code_composer: { version: 1, doc: { type: 'doc' } } }];

const origin = { kind: 'user', clientMetadata };

const snapshot = groupMessagesIntoSnapshot([

{ role: 'user', content: [{ type: 'text', text: 'visible prompt' }], toolCalls: [], origin },

{ role: 'assistant', content: [{ type: 'text', text: 'reply' }], toolCalls: [] },

]);

const turn = snapshot.items.find((item) => item.kind === 'turn');

expect(turn?.origin).toEqual({ kind: 'user', payload: { kind: 'user', clientMetadata } });

expect(turn?.prompt).toBe('visible prompt');

});

it('projects and validates independent document snapshots without losing their nested fields', () => {

const clientMetadata = [

{ kimi_code_composer: { version: 1, doc: { type: 'doc', content: [{ type: 'paragraph' }] }, captureIds: ['capture-a'] } },

{ kimi_code_composer: { version: 1, captureIds: ['capture-b'] } },

];

const projected = projectTranscriptUserOrigin({ kind: 'user', clientMetadata });

expect(transcriptUserOriginSchema.parse(projected)).toEqual({ kind: 'user', clientMetadata });

expect(projectTranscriptUserOrigin({ kind: 'user' })).toStrictEqual({ kind: 'user' });

expect(projectTranscriptUserOrigin({ kind: 'injection', clientMetadata })).toBeUndefined();

});

});

describe('user slash skill activations as transcript origins', () => {

it('projects a user-invoked activation and rejects a model-triggered one', () => {

const origin = { kind: 'skill_activation', trigger: 'user-slash', skillName: 'example-skill', skillArgs: 'args' };

expect(transcriptUserOriginSchema.parse(projectTranscriptUserOrigin(origin))).toEqual(origin);

expect(projectTranscriptUserOrigin({ ...origin, trigger: 'model-tool' })).toBeUndefined();

expect(projectTranscriptUserOrigin({ ...origin, skillName: '' })).toBeUndefined();

expect(projectTranscriptUserOrigin({ kind: 'user' })).toStrictEqual({ kind: 'user' });

});

});

const turnOp = (n: number): TranscriptOperation => ({

op: 'turn.upsert',

turn: { kind: 'turn', turnId: `t${n}`, ordinal: n, state: 'running', origin: { kind: 'user' } },

});

const stepOp: TranscriptOperation = {

op: 'step.upsert',

turnId: 't1',

step: { kind: 'step', stepId: 't1.1', turnId: 't1', ordinal: 1, state: 'running' },

};

const frameOp: TranscriptOperation = {

op: 'frame.upsert',

turnId: 't1',

stepId: 't1.1',

frame: { kind: 'text', frameId: 't1.1.f1', role: 'assistant', text: 'full' },

};

const appendOp: TranscriptOperation = {