ys-code-agent 2.0.0

package/dist/src/config/index.js

@@ -0,0 +1,398 @@
+import { readFileSync, writeFileSync, existsSync, mkdirSync } from 'fs';
+import { join, dirname } from 'path';
+import { homedir } from 'os';
+const DEFAULT_CONFIG_PATH = join(homedir(), '.ys-code-agent', 'config.json');
+const DEFAULT_CONFIG = {
+    version: '1.0.0',
+    providers: [
+        {
+            type: 'openai',
+            name: 'openai',
+            apiKey: '',
+            baseUrl: 'https://api.openai.com/v1',
+            models: ['gpt-4o', 'gpt-4o-mini', 'gpt-4-turbo', 'gpt-3.5-turbo'],
+            defaultModel: 'gpt-4o',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'anthropic',
+            name: 'anthropic',
+            apiKey: '',
+            baseUrl: 'https://api.anthropic.com/v1',
+            models: ['claude-sonnet-4-20250514', 'claude-3-opus-20240229', 'claude-3-sonnet-20240229', 'claude-3-haiku-20240307'],
+            defaultModel: 'claude-sonnet-4-20250514',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'gemini',
+            name: 'gemini',
+            apiKey: '',
+            baseUrl: 'https://generativelanguage.googleapis.com/v1beta',
+            models: ['gemini-2.0-flash', 'gemini-1.5-pro', 'gemini-1.5-flash'],
+            defaultModel: 'gemini-2.0-flash',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'openrouter',
+            name: 'openrouter',
+            apiKey: '',
+            baseUrl: 'https://openrouter.ai/api/v1',
+            models: ['anthropic/claude-sonnet-4-20250514', 'openai/gpt-4o', 'google/gemini-2.0-flash'],
+            defaultModel: 'anthropic/claude-sonnet-4-20250514',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'deepseek',
+            name: 'deepseek',
+            apiKey: '',
+            baseUrl: 'https://api.deepseek.com/v1',
+            models: ['deepseek-chat', 'deepseek-coder'],
+            defaultModel: 'deepseek-chat',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'groq',
+            name: 'groq',
+            apiKey: '',
+            baseUrl: 'https://api.groq.com/openai/v1',
+            models: ['llama-3.3-70b-versatile', 'mixtral-8x7b-32768', 'gemma2-9b-it'],
+            defaultModel: 'llama-3.3-70b-versatile',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+        {
+            type: 'ollama',
+            name: 'ollama',
+            apiKey: '',
+            baseUrl: 'http://localhost:11434/v1',
+            models: ['llama3.3', 'codellama', 'mistral', 'qwen2.5-coder'],
+            defaultModel: 'llama3.3',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 120000,
+        },
+        {
+            type: 'lmstudio',
+            name: 'lmstudio',
+            apiKey: '',
+            baseUrl: 'http://localhost:1234/v1',
+            models: ['local-model'],
+            defaultModel: 'local-model',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 120000,
+        },
+        {
+            type: 'custom',
+            name: 'custom',
+            apiKey: '',
+            baseUrl: '',
+            models: ['custom-model'],
+            defaultModel: 'custom-model',
+            maxRetries: 3,
+            rateLimit: 10,
+            timeout: 60000,
+        },
+    ],
+    activeProvider: 'openai',
+    model: {
+        provider: 'openai',
+        model: 'gpt-4o',
+        temperature: 0.7,
+        maxTokens: 4096,
+        topP: 0.9,
+        frequencyPenalty: 0,
+        presencePenalty: 0,
+        stop: [],
+    },
+    permissions: {
+        autoApprove: false,
+        allowedCommands: [],
+        deniedCommands: ['rm -rf /', 'sudo', 'shutdown', 'reboot', 'init', 'dd', 'mkfs'],
+        allowedPaths: [],
+        deniedPaths: [],
+        maxFileSize: 10485760,
+        askForConfirmation: true,
+    },
+    theme: {
+        mode: 'dark',
+        primaryColor: '#0066FF',
+        secondaryColor: '#00CC66',
+        backgroundColor: '#1a1b26',
+        textColor: '#c0caf5',
+        accentColor: '#7dcfff',
+        fontFamily: 'monospace',
+        fontSize: 14,
+    },
+    tools: {
+        enabledTools: [],
+        disabledTools: [],
+        toolTimeout: 30000,
+        maxToolRetries: 3,
+    },
+    session: {
+        autoSave: true,
+        saveInterval: 60000,
+        maxSessions: 50,
+        sessionDir: join(homedir(), '.ys-code-agent', 'sessions'),
+    },
+    memory: {
+        shortTermSize: 100,
+        longTermEnabled: true,
+        dbPath: join(homedir(), '.ys-code-agent', 'memory.db'),
+        autoSummarize: true,
+        summarizationThreshold: 50,
+    },
+    logging: {
+        level: 'info',
+        file: join(homedir(), '.ys-code-agent', 'logs', 'agent.log'),
+        maxSize: 10485760,
+        maxFiles: 5,
+        consoleOutput: true,
+    },
+    security: {
+        sandboxMode: false,
+        readOnlyMode: false,
+        dangerousCommandDetection: true,
+        maxCommandLength: 10000,
+        allowedEnvVars: ['PATH', 'HOME', 'SHELL', 'USER', 'NODE_ENV'],
+    },
+    fileSystem: {
+        maxFileSize: 10485760,
+        encoding: 'utf-8',
+        cacheEnabled: true,
+        cacheSize: 100,
+        ignorePatterns: ['node_modules/**', '.git/**', 'dist/**', 'build/**', '.next/**', '__pycache__/**'],
+    },
+    git: {
+        autoCommit: false,
+        commitMessagePrefix: 'YS: ',
+        signCommits: false,
+        defaultBranch: 'main',
+    },
+    context: {
+        maxTokens: 128000,
+        compressionEnabled: true,
+        summarizationEnabled: true,
+        relevanceThreshold: 0.3,
+        maxFiles: 50,
+    },
+};
+export class ConfigManager {
+    config;
+    configPath;
+    envPrefix = 'YS_CODE_AGENT_';
+    constructor(configPath) {
+        this.configPath = configPath || process.env.YS_CONFIG_PATH || DEFAULT_CONFIG_PATH;
+        this.config = this.load();
+    }
+    load() {
+        try {
+            if (existsSync(this.configPath)) {
+                const data = readFileSync(this.configPath, 'utf-8');
+                const parsed = JSON.parse(data);
+                return this.mergeDefaults(parsed);
+            }
+        }
+        catch {
+            console.warn(`Failed to load config from ${this.configPath}, using defaults`);
+        }
+        return this.mergeDefaults({});
+    }
+    mergeDefaults(custom) {
+        const merged = {
+            ...DEFAULT_CONFIG,
+            ...custom,
+            providers: custom.providers || DEFAULT_CONFIG.providers,
+            model: { ...DEFAULT_CONFIG.model, ...custom.model },
+            permissions: { ...DEFAULT_CONFIG.permissions, ...custom.permissions },
+            theme: { ...DEFAULT_CONFIG.theme, ...custom.theme },
+            tools: { ...DEFAULT_CONFIG.tools, ...custom.tools },
+            session: { ...DEFAULT_CONFIG.session, ...custom.session },
+            memory: { ...DEFAULT_CONFIG.memory, ...custom.memory },
+            logging: { ...DEFAULT_CONFIG.logging, ...custom.logging },
+            security: { ...DEFAULT_CONFIG.security, ...custom.security },
+            fileSystem: { ...DEFAULT_CONFIG.fileSystem, ...custom.fileSystem },
+            git: { ...DEFAULT_CONFIG.git, ...custom.git },
+            context: { ...DEFAULT_CONFIG.context, ...custom.context },
+        };
+        if (custom.providers) {
+            merged.providers = DEFAULT_CONFIG.providers.map((defaultProv) => {
+                const customProv = custom.providers?.find((p) => p.name === defaultProv.name);
+                return customProv ? { ...defaultProv, ...customProv } : defaultProv;
+            });
+            const customNames = new Set(custom.providers.map((p) => p.name));
+            const defaultNames = new Set(DEFAULT_CONFIG.providers.map((p) => p.name));
+            custom.providers.forEach((p) => {
+                if (!defaultNames.has(p.name)) {
+                    merged.providers.push(p);
+                }
+            });
+        }
+        return merged;
+    }
+    applyEnvOverrides() {
+        for (const key of Object.keys(process.env)) {
+            if (key.startsWith(this.envPrefix)) {
+                const configKey = key.slice(this.envPrefix.length).toLowerCase();
+                const value = process.env[key];
+                this.setNested(configKey, value);
+            }
+        }
+        for (const provider of this.config.providers) {
+            const envKey = `${provider.type.toUpperCase()}_API_KEY`;
+            const envName = `${provider.name.toUpperCase()}_API_KEY`;
+            const genericKey = `API_KEY_${provider.name.toUpperCase()}`;
+            if (process.env[envKey]) {
+                provider.apiKey = process.env[envKey];
+            }
+            if (process.env[envName]) {
+                provider.apiKey = process.env[envName];
+            }
+            if (process.env[genericKey]) {
+                provider.apiKey = process.env[genericKey];
+            }
+        }
+    }
+    setNested(key, value) {
+        const parts = key.split('_');
+        let current = this.config;
+        for (let i = 0; i < parts.length - 1; i++) {
+            if (!current[parts[i]] || typeof current[parts[i]] !== 'object') {
+                current[parts[i]] = {};
+            }
+            current = current[parts[i]];
+        }
+        const lastKey = parts[parts.length - 1];
+        if (!isNaN(Number(value))) {
+            current[lastKey] = Number(value);
+        }
+        else if (value === 'true' || value === 'false') {
+            current[lastKey] = value === 'true';
+        }
+        else {
+            current[lastKey] = value;
+        }
+    }
+    getConfig() {
+        this.applyEnvOverrides();
+        return { ...this.config };
+    }
+    get(key) {
+        this.applyEnvOverrides();
+        const keys = key.split('.');
+        let current = this.config;
+        for (const k of keys) {
+            if (current && typeof current === 'object' && k in current) {
+                current = current[k];
+            }
+            else {
+                return undefined;
+            }
+        }
+        return current;
+    }
+    set(key, value) {
+        const keys = key.split('.');
+        let current = this.config;
+        for (let i = 0; i < keys.length - 1; i++) {
+            if (current && typeof current === 'object') {
+                if (!(current[keys[i]])) {
+                    current[keys[i]] = {};
+                }
+                current = current[keys[i]];
+            }
+        }
+        const lastKey = keys[keys.length - 1];
+        if (current && typeof current === 'object') {
+            current[lastKey] = value;
+        }
+        this.save();
+    }
+    save() {
+        try {
+            const dir = dirname(this.configPath);
+            if (!existsSync(dir)) {
+                mkdirSync(dir, { recursive: true });
+            }
+            writeFileSync(this.configPath, JSON.stringify(this.config, null, 2), 'utf-8');
+        }
+        catch (error) {
+            console.error('Failed to save config:', error);
+        }
+    }
+    getProvider(name) {
+        return this.config.providers.find((p) => p.name === name);
+    }
+    getActiveProvider() {
+        const provider = this.config.providers.find((p) => p.name === this.config.activeProvider);
+        return provider || this.config.providers[0];
+    }
+    getModelConfig() {
+        return { ...this.config.model };
+    }
+    setActiveProvider(name) {
+        const provider = this.config.providers.find((p) => p.name === name);
+        if (provider) {
+            this.config.activeProvider = name;
+            this.config.model.provider = provider.type;
+            if (provider.models.includes(this.config.model.model)) {
+                this.config.model.model = provider.defaultModel;
+            }
+            this.save();
+        }
+    }
+    setApiKey(providerName, apiKey) {
+        const provider = this.config.providers.find((p) => p.name === providerName);
+        if (provider) {
+            provider.apiKey = apiKey;
+            this.save();
+        }
+    }
+    reset() {
+        this.config = { ...DEFAULT_CONFIG };
+        this.save();
+    }
+    exportConfig() {
+        const safe = { ...this.config };
+        safe.providers = safe.providers.map((p) => ({
+            ...p,
+            apiKey: p.apiKey ? '***REDACTED***' : '',
+        }));
+        return JSON.stringify(safe, null, 2);
+    }
+    importConfig(json) {
+        try {
+            const parsed = JSON.parse(json);
+            const clean = { ...parsed };
+            if (clean.providers) {
+                clean.providers = clean.providers.map((p) => {
+                    if (p.apiKey === '***REDACTED***') {
+                        const existing = this.getProvider(p.name);
+                        return { ...p, apiKey: existing?.apiKey || '' };
+                    }
+                    return p;
+                });
+            }
+            this.config = this.mergeDefaults(clean);
+            this.save();
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+}
+export const configManager = new ConfigManager();
+//# sourceMappingURL=index.js.map

package/dist/src/config/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/src/config/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAG7B,MAAM,mBAAmB,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,gBAAgB,EAAE,aAAa,CAAC,CAAC;AAE7E,MAAM,cAAc,GAAW;IAC7B,OAAO,EAAE,OAAO;IAChB,SAAS,EAAE;QACT;YACE,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,2BAA2B;YACpC,MAAM,EAAE,CAAC,QAAQ,EAAE,aAAa,EAAE,aAAa,EAAE,eAAe,CAAC;YACjE,YAAY,EAAE,QAAQ;YACtB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,8BAA8B;YACvC,MAAM,EAAE,CAAC,0BAA0B,EAAE,wBAAwB,EAAE,0BAA0B,EAAE,yBAAyB,CAAC;YACrH,YAAY,EAAE,0BAA0B;YACxC,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,kDAAkD;YAC3D,MAAM,EAAE,CAAC,kBAAkB,EAAE,gBAAgB,EAAE,kBAAkB,CAAC;YAClE,YAAY,EAAE,kBAAkB;YAChC,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,8BAA8B;YACvC,MAAM,EAAE,CAAC,oCAAoC,EAAE,eAAe,EAAE,yBAAyB,CAAC;YAC1F,YAAY,EAAE,oCAAoC;YAClD,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,UAAU;YAChB,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,6BAA6B;YACtC,MAAM,EAAE,CAAC,eAAe,EAAE,gBAAgB,CAAC;YAC3C,YAAY,EAAE,eAAe;YAC7B,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,MAAM;YACZ,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,gCAAgC;YACzC,MAAM,EAAE,CAAC,yBAAyB,EAAE,oBAAoB,EAAE,cAAc,CAAC;YACzE,YAAY,EAAE,yBAAyB;YACvC,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;QACD;YACE,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,2BAA2B;YACpC,MAAM,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,CAAC;YAC7D,YAAY,EAAE,UAAU;YACxB,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,MAAM;SAChB;QACD;YACE,IAAI,EAAE,UAAU;YAChB,IAAI,EAAE,UAAU;YAChB,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,0BAA0B;YACnC,MAAM,EAAE,CAAC,aAAa,CAAC;YACvB,YAAY,EAAE,aAAa;YAC3B,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,MAAM;SAChB;QACD;YACE,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;YACV,OAAO,EAAE,EAAE;YACX,MAAM,EAAE,CAAC,cAAc,CAAC;YACxB,YAAY,EAAE,cAAc;YAC5B,UAAU,EAAE,CAAC;YACb,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;SACf;KACF;IACD,cAAc,EAAE,QAAQ;IACxB,KAAK,EAAE;QACL,QAAQ,EAAE,QAAQ;QAClB,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,GAAG;QAChB,SAAS,EAAE,IAAI;QACf,IAAI,EAAE,GAAG;QACT,gBAAgB,EAAE,CAAC;QACnB,eAAe,EAAE,CAAC;QAClB,IAAI,EAAE,EAAE;KACT;IACD,WAAW,EAAE;QACX,WAAW,EAAE,KAAK;QAClB,eAAe,EAAE,EAAE;QACnB,cAAc,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC;QAChF,YAAY,EAAE,EAAE;QAChB,WAAW,EAAE,EAAE;QACf,WAAW,EAAE,QAAQ;QACrB,kBAAkB,EAAE,IAAI;KACzB;IACD,KAAK,EAAE;QACL,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,SAAS;QACvB,cAAc,EAAE,SAAS;QACzB,eAAe,EAAE,SAAS;QAC1B,SAAS,EAAE,SAAS;QACpB,WAAW,EAAE,SAAS;QACtB,UAAU,EAAE,WAAW;QACvB,QAAQ,EAAE,EAAE;KACb;IACD,KAAK,EAAE;QACL,YAAY,EAAE,EAAE;QAChB,aAAa,EAAE,EAAE;QACjB,WAAW,EAAE,KAAK;QAClB,cAAc,EAAE,CAAC;KAClB;IACD,OAAO,EAAE;QACP,QAAQ,EAAE,IAAI;QACd,YAAY,EAAE,KAAK;QACnB,WAAW,EAAE,EAAE;QACf,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,gBAAgB,EAAE,UAAU,CAAC;KAC1D;IACD,MAAM,EAAE;QACN,aAAa,EAAE,GAAG;QAClB,eAAe,EAAE,IAAI;QACrB,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,gBAAgB,EAAE,WAAW,CAAC;QACtD,aAAa,EAAE,IAAI;QACnB,sBAAsB,EAAE,EAAE;KAC3B;IACD,OAAO,EAAE;QACP,KAAK,EAAE,MAAM;QACb,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,gBAAgB,EAAE,MAAM,EAAE,WAAW,CAAC;QAC5D,OAAO,EAAE,QAAQ;QACjB,QAAQ,EAAE,CAAC;QACX,aAAa,EAAE,IAAI;KACpB;IACD,QAAQ,EAAE;QACR,WAAW,EAAE,KAAK;QAClB,YAAY,EAAE,KAAK;QACnB,yBAAyB,EAAE,IAAI;QAC/B,gBAAgB,EAAE,KAAK;QACvB,cAAc,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC;KAC9D;IACD,UAAU,EAAE;QACV,WAAW,EAAE,QAAQ;QACrB,QAAQ,EAAE,OAAO;QACjB,YAAY,EAAE,IAAI;QAClB,SAAS,EAAE,GAAG;QACd,cAAc,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,gBAAgB,CAAC;KACpG;IACD,GAAG,EAAE;QACH,UAAU,EAAE,KAAK;QACjB,mBAAmB,EAAE,MAAM;QAC3B,WAAW,EAAE,KAAK;QAClB,aAAa,EAAE,MAAM;KACtB;IACD,OAAO,EAAE;QACP,SAAS,EAAE,MAAM;QACjB,kBAAkB,EAAE,IAAI;QACxB,oBAAoB,EAAE,IAAI;QAC1B,kBAAkB,EAAE,GAAG;QACvB,QAAQ,EAAE,EAAE;KACb;CACF,CAAC;AAEF,MAAM,OAAO,aAAa;IAChB,MAAM,CAAS;IACf,UAAU,CAAS;IACnB,SAAS,GAAG,gBAAgB,CAAC;IAErC,YAAY,UAAmB;QAC7B,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,mBAAmB,CAAC;QAClF,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC5B,CAAC;IAEO,IAAI;QACV,IAAI,CAAC;YACH,IAAI,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBAChC,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBACpD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAChC,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,IAAI,CAAC,8BAA8B,IAAI,CAAC,UAAU,kBAAkB,CAAC,CAAC;QAChF,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC;IAEO,aAAa,CAAC,MAAuB;QAC3C,MAAM,MAAM,GAAG;YACb,GAAG,cAAc;YACjB,GAAG,MAAM;YACT,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,cAAc,CAAC,SAAS;YACvD,KAAK,EAAE,EAAE,GAAG,cAAc,CAAC,KAAK,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE;YACnD,WAAW,EAAE,EAAE,GAAG,cAAc,CAAC,WAAW,EAAE,GAAG,MAAM,CAAC,WAAW,EAAE;YACrE,KAAK,EAAE,EAAE,GAAG,cAAc,CAAC,KAAK,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE;YACnD,KAAK,EAAE,EAAE,GAAG,cAAc,CAAC,KAAK,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE;YACnD,OAAO,EAAE,EAAE,GAAG,cAAc,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE;YACzD,MAAM,EAAE,EAAE,GAAG,cAAc,CAAC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE;YACtD,OAAO,EAAE,EAAE,GAAG,cAAc,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE;YACzD,QAAQ,EAAE,EAAE,GAAG,cAAc,CAAC,QAAQ,EAAE,GAAG,MAAM,CAAC,QAAQ,EAAE;YAC5D,UAAU,EAAE,EAAE,GAAG,cAAc,CAAC,UAAU,EAAE,GAAG,MAAM,CAAC,UAAU,EAAE;YAClE,GAAG,EAAE,EAAE,GAAG,cAAc,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,GAAG,EAAE;YAC7C,OAAO,EAAE,EAAE,GAAG,cAAc,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE;SAC1D,CAAC;QAEF,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,MAAM,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,EAAE;gBAC9D,MAAM,UAAU,GAAG,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC;gBAC9E,OAAO,UAAU,CAAC,CAAC,CAAC,EAAE,GAAG,WAAW,EAAE,GAAG,UAAU,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;YACtE,CAAC,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YACjE,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1E,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC7B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC9B,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,iBAAiB;QACvB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3C,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;gBACjE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;gBAChC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC;YACxD,MAAM,OAAO,GAAG,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,CAAC;YACzD,MAAM,UAAU,GAAG,WAAW,QAAQ,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;YAC5D,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;gBACxB,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAE,CAAC;YACzC,CAAC;YACD,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;gBACzB,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC;YAC1C,CAAC;YACD,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5B,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAE,CAAC;YAC7C,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,CAAC,GAAW,EAAE,KAAa;QAC1C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7B,IAAI,OAAO,GAA4B,IAAI,CAAC,MAA4C,CAAC;QACzF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAChE,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;YACzB,CAAC;YACD,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAA4B,CAAC;QACzD,CAAC;QACD,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;YACzB,OAAmC,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YAChD,OAAmC,CAAC,OAAO,CAAC,GAAG,KAAK,KAAK,MAAM,CAAC;QACnE,CAAC;aAAM,CAAC;YACL,OAAmC,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC;QACxD,CAAC;IACH,CAAC;IAED,SAAS;QACP,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,GAAG,CAAI,GAAW;QAChB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,OAAO,GAAY,IAAI,CAAC,MAAM,CAAC;QACnC,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,CAAC,IAAK,OAAmC,EAAE,CAAC;gBACxF,OAAO,GAAI,OAAmC,CAAC,CAAC,CAAC,CAAC;YACpD,CAAC;iBAAM,CAAC;gBACN,OAAO,SAAc,CAAC;YACxB,CAAC;QACH,CAAC;QACD,OAAO,OAAY,CAAC;IACtB,CAAC;IAED,GAAG,CAAI,GAAW,EAAE,KAAQ;QAC1B,MAAM,IAAI,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC5B,IAAI,OAAO,GAAY,IAAI,CAAC,MAAM,CAAC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC3C,IAAI,CAAC,CAAE,OAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;oBACpD,OAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBACrD,CAAC;gBACD,OAAO,GAAI,OAAmC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACtC,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC1C,OAAmC,CAAC,OAAO,CAAC,GAAG,KAAK,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,IAAI;QACF,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrB,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACtC,CAAC;YACD,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAChF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,WAAW,CAAC,IAAY;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IAC5D,CAAC;IAED,iBAAiB;QACf,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QAC1F,OAAO,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED,cAAc;QACZ,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAED,iBAAiB,CAAC,IAAY;QAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACpE,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,IAAI,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3C,IAAI,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;gBACtD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,GAAG,QAAQ,CAAC,YAAY,CAAC;YAClD,CAAC;YACD,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAED,SAAS,CAAC,YAAoB,EAAE,MAAc;QAC5C,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;QAC5E,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC;YACzB,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED,YAAY;QACV,MAAM,IAAI,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1C,GAAG,CAAC;YACJ,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE;SACzC,CAAC,CAAC,CAAC;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC;IAED,YAAY,CAAC,IAAY;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAChC,MAAM,KAAK,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gBACpB,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAiB,EAAE,EAAE;oBAC1D,IAAI,CAAC,CAAC,MAAM,KAAK,gBAAgB,EAAE,CAAC;wBAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;wBAC1C,OAAO,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,IAAI,EAAE,EAAE,CAAC;oBAClD,CAAC;oBACD,OAAO,CAAC,CAAC;gBACX,CAAC,CAAC,CAAC;YACL,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YACxC,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC"}

package/dist/src/context/index.d.ts

@@ -0,0 +1,31 @@
+import { AgentMessage, ContextData } from '../types.js';
+export declare class ContextEngine {
+    private maxTokens;
+    private compressionEnabled;
+    private summarizationEnabled;
+    private relevanceThreshold;
+    private maxFiles;
+    private openFiles;
+    private recentFiles;
+    private projectStructure;
+    constructor();
+    getContext(): ContextData;
+    private getEnvironment;
+    updateProjectStructure(): Promise<void>;
+    addOpenFile(filePath: string): void;
+    removeOpenFile(filePath: string): void;
+    compressMessages(messages: AgentMessage[]): Promise<AgentMessage[]>;
+    private createSummary;
+    rankFilesByRelevance(task: string, files: string[]): Array<{
+        file: string;
+        score: number;
+    }>;
+    getRelevantFiles(task: string, maxFiles?: number): Promise<string[]>;
+    getReadableContext(task: string, maxTokens?: number): string;
+    tokenCount(text: string): number;
+    isWithinLimit(text: string): boolean;
+    getMaxTokens(): number;
+    setMaxTokens(tokens: number): void;
+}
+export declare const contextEngine: ContextEngine;
+//# sourceMappingURL=index.d.ts.map

package/dist/src/context/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/src/context/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AAMlE,qBAAa,aAAa;IACxB,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,kBAAkB,CAAU;IACpC,OAAO,CAAC,oBAAoB,CAAU;IACtC,OAAO,CAAC,kBAAkB,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,WAAW,CAAgB;IACnC,OAAO,CAAC,gBAAgB,CAAkB;;IAW1C,UAAU,IAAI,WAAW;IAWzB,OAAO,CAAC,cAAc;IAWhB,sBAAsB,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ7C,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAiBnC,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKhC,gBAAgB,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAsDzE,OAAO,CAAC,aAAa;IASrB,oBAAoB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAoCrF,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,SAAK,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAYtE,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,SAAO,GAAG,MAAM;IAuC1D,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIhC,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIpC,YAAY,IAAI,MAAM;IAItB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;CAGnC;AAED,eAAO,MAAM,aAAa,eAAsB,CAAC"}

package/dist/src/context/index.js

@@ -0,0 +1,208 @@
+import { getLogger } from '../logger/index.js';
+import { configManager } from '../config/index.js';
+import { countTokens, truncate } from '../utils/index.js';
+import { fileSystem } from '../filesystem/index.js';
+const logger = getLogger('context');
+export class ContextEngine {
+    maxTokens;
+    compressionEnabled;
+    summarizationEnabled;
+    relevanceThreshold;
+    maxFiles;
+    openFiles = [];
+    recentFiles = [];
+    projectStructure = [];
+    constructor() {
+        const config = configManager.getConfig();
+        this.maxTokens = config.context.maxTokens;
+        this.compressionEnabled = config.context.compressionEnabled;
+        this.summarizationEnabled = config.context.summarizationEnabled;
+        this.relevanceThreshold = config.context.relevanceThreshold;
+        this.maxFiles = config.context.maxFiles;
+    }
+    getContext() {
+        return {
+            currentDirectory: fileSystem.getBasePath(),
+            openFiles: [...this.openFiles],
+            recentFiles: [...this.recentFiles],
+            projectStructure: [...this.projectStructure],
+            environment: this.getEnvironment(),
+            tokenCount: 0,
+        };
+    }
+    getEnvironment() {
+        const env = {};
+        const allowedVars = ['HOME', 'USER', 'SHELL', 'PATH', 'NODE_ENV', 'PWD'];
+        for (const key of allowedVars) {
+            if (process.env[key]) {
+                env[key] = process.env[key];
+            }
+        }
+        return env;
+    }
+    async updateProjectStructure() {
+        try {
+            this.projectStructure = fileSystem.listDirectory('.');
+        }
+        catch (error) {
+            logger.error('Failed to update project structure', error);
+        }
+    }
+    addOpenFile(filePath) {
+        const normalized = filePath.replace(/\\/g, '/');
+        this.openFiles = this.openFiles.filter((f) => f !== normalized);
+        this.openFiles.push(normalized);
+        if (this.openFiles.length > this.maxFiles) {
+            this.openFiles.shift();
+        }
+        this.recentFiles = this.recentFiles.filter((f) => f !== normalized);
+        this.recentFiles.unshift(normalized);
+        if (this.recentFiles.length > 20) {
+            this.recentFiles.pop();
+        }
+    }
+    removeOpenFile(filePath) {
+        const normalized = filePath.replace(/\\/g, '/');
+        this.openFiles = this.openFiles.filter((f) => f !== normalized);
+    }
+    async compressMessages(messages) {
+        if (!this.compressionEnabled)
+            return messages;
+        const totalTokens = messages.reduce((acc, m) => acc + countTokens(m.content), 0);
+        if (totalTokens <= this.maxTokens)
+            return messages;
+        logger.info(`Compressing ${messages.length} messages (${totalTokens} tokens > ${this.maxTokens} tokens)`);
+        const compressed = [];
+        let currentTokens = 0;
+        const systemMessages = messages.filter((m) => m.role === 'system');
+        for (const msg of systemMessages) {
+            compressed.push(msg);
+            currentTokens += countTokens(msg.content);
+        }
+        const latestMessages = messages.filter((m) => m.role !== 'system').slice(-20);
+        for (const msg of latestMessages) {
+            compressed.push(msg);
+            currentTokens += countTokens(msg.content);
+        }
+        const remaining = messages.filter((m) => m.role !== 'system').slice(0, -20);
+        if (this.summarizationEnabled && remaining.length > 0) {
+            const summary = this.createSummary(remaining);
+            const summaryTokens = countTokens(summary);
+            if (currentTokens + summaryTokens <= this.maxTokens) {
+                compressed.unshift({
+                    role: 'system',
+                    content: `[Previous conversation summary]:\n${summary}`,
+                    timestamp: Date.now(),
+                });
+                currentTokens += summaryTokens;
+            }
+        }
+        if (currentTokens > this.maxTokens) {
+            while (compressed.length > 0 && currentTokens > this.maxTokens) {
+                const removed = compressed.shift();
+                if (removed && removed.role !== 'system') {
+                    currentTokens -= countTokens(removed.content);
+                }
+            }
+        }
+        logger.info(`Compressed to ${compressed.length} messages (${currentTokens} tokens)`);
+        return compressed;
+    }
+    createSummary(messages) {
+        const lines = [];
+        for (const msg of messages.slice(-30)) {
+            const truncated = truncate(msg.content.replace(/\n/g, ' '), 200);
+            lines.push(`[${msg.role}] ${truncated}`);
+        }
+        return lines.join('\n');
+    }
+    rankFilesByRelevance(task, files) {
+        const taskTokens = task.toLowerCase().split(/\s+/);
+        const taskSet = new Set(taskTokens);
+        const taskWords = [...taskSet].filter((w) => w.length > 3);
+        const scored = [];
+        const stopWords = new Set(['the', 'and', 'for', 'are', 'but', 'not', 'you', 'all', 'can', 'had', 'her', 'was', 'one', 'our', 'out', 'has', 'have', 'been', 'some', 'them', 'than', 'that', 'this', 'very', 'just', 'with', 'without', 'from', 'about', 'into', 'over', 'also', 'other', 'more', 'then']);
+        for (const filePath of files) {
+            let score = 0;
+            const fileName = filePath.split('/').pop() || '';
+            const fileLower = filePath.toLowerCase();
+            const nameLower = fileName.toLowerCase();
+            for (const word of taskWords) {
+                if (stopWords.has(word))
+                    continue;
+                if (nameLower.includes(word)) {
+                    score += 3;
+                }
+                if (fileLower.includes(word)) {
+                    score += 1;
+                }
+            }
+            if (fileLower.includes('test'))
+                score -= 0.5;
+            if (fileLower.includes('node_modules'))
+                score -= 5;
+            scored.push({ file: filePath, score });
+        }
+        return scored
+            .filter((s) => s.score >= this.relevanceThreshold)
+            .sort((a, b) => b.score - a.score);
+    }
+    async getRelevantFiles(task, maxFiles = 10) {
+        try {
+            const allFiles = await fileSystem.globSearch('**/*.{ts,js,tsx,jsx,json,md,py,java,cpp,c,h,hpp,rs,go,yaml,yml,toml,xml,html,css,scss,less,sql,sh,bash}');
+            const ranked = this.rankFilesByRelevance(task, allFiles);
+            return ranked.slice(0, maxFiles).map((r) => r.file);
+        }
+        catch (error) {
+            logger.error('Failed to get relevant files', error);
+            return [];
+        }
+    }
+    getReadableContext(task, maxTokens = 4000) {
+        const context = this.getContext();
+        const lines = [];
+        lines.push(`Working Directory: ${context.currentDirectory}`);
+        lines.push('');
+        if (context.projectStructure.length > 0) {
+            lines.push('Project Structure:');
+            const tree = fileSystem.getDirectoryTree('.', 3);
+            lines.push(tree);
+            lines.push('');
+        }
+        if (context.openFiles.length > 0) {
+            lines.push(`Open Files: ${context.openFiles.join(', ')}`);
+            lines.push('');
+        }
+        try {
+            const relevantFiles = this.rankFilesByRelevance(task, context.recentFiles.length > 0 ? context.recentFiles : ['package.json', 'tsconfig.json']);
+            if (relevantFiles.length > 0) {
+                lines.push('Relevant Files:');
+                for (const { file, score } of relevantFiles.slice(0, 5)) {
+                    lines.push(`  ${file} (relevance: ${score.toFixed(2)})`);
+                }
+                lines.push('');
+            }
+        }
+        catch {
+        }
+        let result = lines.join('\n');
+        if (countTokens(result) > maxTokens) {
+            result = truncate(result, maxTokens * 4);
+        }
+        return result;
+    }
+    tokenCount(text) {
+        return countTokens(text);
+    }
+    isWithinLimit(text) {
+        return countTokens(text) <= this.maxTokens;
+    }
+    getMaxTokens() {
+        return this.maxTokens;
+    }
+    setMaxTokens(tokens) {
+        this.maxTokens = tokens;
+    }
+}
+export const contextEngine = new ContextEngine();
+//# sourceMappingURL=index.js.map