86 lines
2.7 KiB
TypeScript
86 lines
2.7 KiB
TypeScript
|
import type { AIPermissionMode } from '../types';
|
||
|
|
|
||
|
|
export interface PromptContextSnapshot {
|
||
|
|
version: 2;
|
||
|
|
audience: 'sidebar';
|
||
|
|
providerId?: string;
|
||
|
|
modelId?: string;
|
||
|
|
permissionMode: AIPermissionMode;
|
||
|
|
scopeType: 'terminal' | 'workspace';
|
||
|
|
scopeLabel?: string;
|
||
|
|
toolNames: string[];
|
||
|
|
selectedSkillSlugs: string[];
|
||
|
|
systemPromptChars: number;
|
||
|
|
systemPromptHash: string;
|
||
|
|
injections: Array<{
|
||
|
|
order: number;
|
||
|
|
source: 'system-prompt' | 'capability-catalog' | 'user-skills' | 'terminal-scope' | 'web-search';
|
||
|
|
itemCount: number;
|
||
|
|
chars?: number;
|
||
|
|
hash: string;
|
||
|
|
}>;
|
||
|
|
webSearchEnabled: boolean;
|
||
|
|
hostSessionIds: string[];
|
||
|
|
/** Dynamic request estimate used by the Catty context meter. */
|
||
|
|
contextWindow?: number;
|
||
|
|
estimatedInputTokens?: number;
|
||
|
|
builtAt: number;
|
||
|
|
}
|
||
|
|
|
||
|
|
export function buildPromptContextSnapshot(input: {
|
||
|
|
providerId?: string;
|
||
|
|
modelId?: string;
|
||
|
|
permissionMode: AIPermissionMode;
|
||
|
|
scopeType: 'terminal' | 'workspace';
|
||
|
|
scopeLabel?: string;
|
||
|
|
toolNames: string[];
|
||
|
|
selectedSkillSlugs?: string[];
|
||
|
|
systemPrompt: string;
|
||
|
|
webSearchEnabled: boolean;
|
||
|
|
hostSessionIds: string[];
|
||
|
|
builtAt?: number;
|
||
|
|
}): PromptContextSnapshot {
|
||
|
|
const toolNames = [...input.toolNames].sort();
|
||
|
|
const selectedSkillSlugs = [...(input.selectedSkillSlugs ?? [])].sort();
|
||
|
|
const hostSessionIds = [...input.hostSessionIds];
|
||
|
|
const injectionValues = [
|
||
|
|
{ source: 'system-prompt' as const, values: [input.systemPrompt], chars: input.systemPrompt.length },
|
||
|
|
{ source: 'capability-catalog' as const, values: toolNames },
|
||
|
|
{ source: 'user-skills' as const, values: selectedSkillSlugs },
|
||
|
|
{ source: 'terminal-scope' as const, values: hostSessionIds },
|
||
|
|
{ source: 'web-search' as const, values: [String(input.webSearchEnabled)] },
|
||
|
|
];
|
||
|
|
return {
|
||
|
|
version: 2,
|
||
|
|
audience: 'sidebar',
|
||
|
|
providerId: input.providerId,
|
||
|
|
modelId: input.modelId,
|
||
|
|
permissionMode: input.permissionMode,
|
||
|
|
scopeType: input.scopeType,
|
||
|
|
scopeLabel: input.scopeLabel,
|
||
|
|
toolNames,
|
||
|
|
selectedSkillSlugs,
|
||
|
|
systemPromptChars: input.systemPrompt.length,
|
||
|
|
systemPromptHash: hashPromptPart(input.systemPrompt),
|
||
|
|
injections: injectionValues.map((entry, order) => ({
|
||
|
|
order,
|
||
|
|
source: entry.source,
|
||
|
|
itemCount: entry.values.length,
|
||
|
|
...(entry.chars != null ? { chars: entry.chars } : {}),
|
||
|
|
hash: hashPromptPart(entry.values.join('\n')),
|
||
|
|
})),
|
||
|
|
webSearchEnabled: input.webSearchEnabled,
|
||
|
|
hostSessionIds,
|
||
|
|
builtAt: input.builtAt ?? Date.now(),
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
function hashPromptPart(value: string): string {
|
||
|
|
let hash = 0x811c9dc5;
|
||
|
|
for (let index = 0; index < value.length; index += 1) {
|
||
|
|
hash ^= value.charCodeAt(index);
|
||
|
|
hash = Math.imul(hash, 0x01000193);
|
||
|
|
}
|
||
|
|
return `fnv1a-${(hash >>> 0).toString(16).padStart(8, '0')}`;
|
||
|
|
}
|