diff --git a/packages/core/src/core/agent.ts b/packages/core/src/core/agent.ts index 5564449..8dacc55 100644 --- a/packages/core/src/core/agent.ts +++ b/packages/core/src/core/agent.ts @@ -57,8 +57,16 @@ export class Agent { baseUrl: config.baseUrl, }); + // 构建压缩配置,使用模型的 contextWindow(如果有) + const finalCompressionConfig: Partial = { + ...compressionConfig, + }; + if (config.contextWindow) { + finalCompressionConfig.contextLimit = config.contextWindow; + } + // 初始化压缩管理器 - this.compressionManager = new CompressionManager(compressionConfig); + this.compressionManager = new CompressionManager(finalCompressionConfig); // 设置主模型(作为摘要模型的后备) this.compressionManager.setModel(this.getModel(config.model)); diff --git a/packages/core/src/provider/registry.ts b/packages/core/src/provider/registry.ts index 1bed0f0..19559bb 100644 --- a/packages/core/src/provider/registry.ts +++ b/packages/core/src/provider/registry.ts @@ -300,6 +300,17 @@ export class ProviderRegistry { return [...provider.info.models, ...customModels]; } + /** + * 获取指定模型的详细信息 + * @param providerId 提供商 ID + * @param modelId 模型 ID + * @returns 模型信息,未找到时返回 undefined + */ + getModelInfo(providerId: string, modelId: string): ModelInfo | undefined { + const models = this.getModels(providerId); + return models.find((m) => m.id === modelId); + } + /** * 添加自定义模型 */ diff --git a/packages/core/src/types/index.ts b/packages/core/src/types/index.ts index 1a476f3..ee059e7 100644 --- a/packages/core/src/types/index.ts +++ b/packages/core/src/types/index.ts @@ -78,6 +78,8 @@ export interface AgentConfig { systemPrompt: string; /** 自定义 API 基础 URL(用于兼容 OpenAI API 的第三方服务,如阿里云百炼) */ baseUrl?: string; + /** 模型的上下文窗口大小(tokens),用于压缩管理 */ + contextWindow?: number; } // 会话上下文 diff --git a/packages/core/src/utils/config.ts b/packages/core/src/utils/config.ts index 6cf2854..c31a9db 100644 --- a/packages/core/src/utils/config.ts +++ b/packages/core/src/utils/config.ts @@ -121,6 +121,10 @@ export function loadConfig(): AgentConfig { // 确定 baseUrl const finalBaseUrl = storedConfig.baseUrl || providerConfig?.baseUrl; + // 获取模型的 contextWindow(从 ProviderRegistry 查询) + const modelInfo = providerRegistry.getModelInfo(finalProvider, finalModel); + const contextWindow = modelInfo?.contextWindow; + return { provider: finalProvider, apiKey: finalApiKey, @@ -128,6 +132,7 @@ export function loadConfig(): AgentConfig { maxTokens: storedConfig.maxTokens || 4096, systemPrompt: DEFAULT_SYSTEM_PROMPT, baseUrl: finalBaseUrl, + contextWindow, }; }