feat(ipuaro): add context configuration

- Add ContextConfigSchema with systemPromptTokens, maxContextUsage, autoCompressAt, compressionMethod
- Update ContextManager to read compression threshold from config
- Update HandleMessage and useSession to pass context config
- Add 40 unit tests (32 schema + 8 integration)
- Coverage: 97.63% lines, 91.34% branches
This commit is contained in:
imfozilbek
2025-12-02 02:02:34 +05:00
parent 98b365bd94
commit fa647c41aa
8 changed files with 371 additions and 9 deletions

View File

@@ -2,6 +2,7 @@ import type { ContextState, Session } from "../../domain/entities/Session.js"
import type { ILLMClient } from "../../domain/services/ILLMClient.js"
import { type ChatMessage, createSystemMessage } from "../../domain/value-objects/ChatMessage.js"
import { CONTEXT_COMPRESSION_THRESHOLD, CONTEXT_WINDOW_SIZE } from "../../domain/constants/index.js"
import type { ContextConfig } from "../../shared/constants/config.js"
/**
* File in context with token count.
@@ -39,9 +40,13 @@ export class ContextManager {
private readonly filesInContext = new Map<string, FileContext>()
private currentTokens = 0
private readonly contextWindowSize: number
private readonly compressionThreshold: number
private readonly compressionMethod: "llm-summary" | "truncate"
constructor(contextWindowSize: number = CONTEXT_WINDOW_SIZE) {
constructor(contextWindowSize: number = CONTEXT_WINDOW_SIZE, config?: ContextConfig) {
this.contextWindowSize = contextWindowSize
this.compressionThreshold = config?.autoCompressAt ?? CONTEXT_COMPRESSION_THRESHOLD
this.compressionMethod = config?.compressionMethod ?? "llm-summary"
}
/**
@@ -97,7 +102,7 @@ export class ContextManager {
* Check if compression is needed.
*/
needsCompression(): boolean {
return this.getUsage() > CONTEXT_COMPRESSION_THRESHOLD
return this.getUsage() > this.compressionThreshold
}
/**

View File

@@ -70,6 +70,7 @@ export interface HandleMessageOptions {
maxToolCalls?: number
maxHistoryMessages?: number
saveInputHistory?: boolean
contextConfig?: import("../../shared/constants/config.js").ContextConfig
}
const DEFAULT_MAX_TOOL_CALLS = 20
@@ -98,13 +99,14 @@ export class HandleMessage {
llm: ILLMClient,
tools: IToolRegistry,
projectRoot: string,
contextConfig?: import("../../shared/constants/config.js").ContextConfig,
) {
this.storage = storage
this.sessionStorage = sessionStorage
this.llm = llm
this.tools = tools
this.projectRoot = projectRoot
this.contextManager = new ContextManager(llm.getContextWindowSize())
this.contextManager = new ContextManager(llm.getContextWindowSize(), contextConfig)
this.executeTool = new ExecuteTool(storage, sessionStorage, tools, projectRoot)
}

View File

@@ -106,6 +106,16 @@ export const SessionConfigSchema = z.object({
saveInputHistory: z.boolean().default(true),
})
/**
* Context configuration schema.
*/
export const ContextConfigSchema = z.object({
systemPromptTokens: z.number().int().positive().default(2000),
maxContextUsage: z.number().min(0).max(1).default(0.8),
autoCompressAt: z.number().min(0).max(1).default(0.8),
compressionMethod: z.enum(["llm-summary", "truncate"]).default("llm-summary"),
})
/**
* Full configuration schema.
*/
@@ -119,6 +129,7 @@ export const ConfigSchema = z.object({
input: InputConfigSchema.default({}),
display: DisplayConfigSchema.default({}),
session: SessionConfigSchema.default({}),
context: ContextConfigSchema.default({}),
})
/**
@@ -134,6 +145,7 @@ export type EditConfig = z.infer<typeof EditConfigSchema>
export type InputConfig = z.infer<typeof InputConfigSchema>
export type DisplayConfig = z.infer<typeof DisplayConfigSchema>
export type SessionConfig = z.infer<typeof SessionConfigSchema>
export type ContextConfig = z.infer<typeof ContextConfigSchema>
/**
* Default configuration.

View File

@@ -109,6 +109,7 @@ async function initializeSession(
deps.llm,
deps.tools,
deps.projectRoot,
deps.config?.context,
)
if (deps.projectStructure) {
handleMessage.setProjectStructure(deps.projectStructure)
@@ -117,6 +118,7 @@ async function initializeSession(
autoApply: options.autoApply,
maxHistoryMessages: deps.config?.session.maxHistoryMessages,
saveInputHistory: deps.config?.session.saveInputHistory,
contextConfig: deps.config?.context,
})
handleMessage.setEvents(createEventHandlers(setters, options))
refs.current.handleMessage = handleMessage