ys-code-agent 2.0.0

package/dist/session/index.js

@@ -0,0 +1,203 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync, readdirSync, unlinkSync } from 'fs';
+import { join } from 'path';
+import { getLogger } from '../logger/index.js';
+import { configManager } from '../config/index.js';
+import { generateId, formatDate } from '../utils/index.js';
+const logger = getLogger('session');
+export class SessionManager {
+    sessionDir;
+    maxSessions;
+    autoSave;
+    saveInterval;
+    currentSession = null;
+    sessions = new Map();
+    saveTimer = null;
+    constructor() {
+        const config = configManager.getConfig();
+        this.sessionDir = config.session.sessionDir;
+        this.maxSessions = config.session.maxSessions;
+        this.autoSave = config.session.autoSave;
+        this.saveInterval = config.session.saveInterval;
+        this.init();
+    }
+    init() {
+        if (!existsSync(this.sessionDir)) {
+            mkdirSync(this.sessionDir, { recursive: true });
+        }
+        this.loadSessions();
+        if (this.autoSave) {
+            this.saveTimer = setInterval(() => {
+                this.saveCurrentSession();
+            }, this.saveInterval);
+        }
+        logger.info(`Session manager initialized (${this.sessions.size} sessions)`);
+    }
+    loadSessions() {
+        try {
+            const files = readdirSync(this.sessionDir)
+                .filter((f) => f.endsWith('.json'))
+                .sort()
+                .reverse();
+            for (const file of files.slice(0, this.maxSessions)) {
+                try {
+                    const content = readFileSync(join(this.sessionDir, file), 'utf-8');
+                    const session = JSON.parse(content);
+                    this.sessions.set(session.id, session);
+                }
+                catch (err) {
+                    logger.warn(`Failed to load session file: ${file}`, err);
+                }
+            }
+        }
+        catch (error) {
+            logger.error('Failed to load sessions', error);
+        }
+    }
+    createSession(name) {
+        const session = {
+            id: generateId(24),
+            name: name || `Session ${formatDate(Date.now())}`,
+            state: this.createEmptyState(),
+            createdAt: Date.now(),
+            updatedAt: Date.now(),
+        };
+        this.sessions.set(session.id, session);
+        this.currentSession = session;
+        this.saveSession(session);
+        logger.info(`Created session: ${session.id} (${session.name})`);
+        return session;
+    }
+    createEmptyState() {
+        return {
+            task: '',
+            plan: [],
+            currentStep: 0,
+            messages: [],
+            context: {
+                currentDirectory: process.cwd(),
+                openFiles: [],
+                recentFiles: [],
+                projectStructure: [],
+                environment: {},
+                tokenCount: 0,
+            },
+            memory: {
+                shortTerm: { messages: [], maxSize: 100 },
+                longTerm: { summaries: [], preferences: {}, previousTasks: [] },
+            },
+            status: 'idle',
+            startTime: Date.now(),
+        };
+    }
+    getCurrentSession() {
+        return this.currentSession;
+    }
+    setCurrentSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (session) {
+            this.currentSession = session;
+            return session;
+        }
+        return null;
+    }
+    getSession(sessionId) {
+        return this.sessions.get(sessionId) || null;
+    }
+    getAllSessions() {
+        return [...this.sessions.values()]
+            .sort((a, b) => b.updatedAt - a.updatedAt);
+    }
+    updateSessionState(state) {
+        if (this.currentSession) {
+            this.currentSession.state = state;
+            this.currentSession.updatedAt = Date.now();
+        }
+    }
+    saveCurrentSession() {
+        if (this.currentSession) {
+            this.saveSession(this.currentSession);
+        }
+    }
+    saveSession(session) {
+        try {
+            const filePath = join(this.sessionDir, `${session.id}.json`);
+            writeFileSync(filePath, JSON.stringify(session, null, 2), 'utf-8');
+        }
+        catch (error) {
+            logger.error('Failed to save session', error);
+        }
+    }
+    deleteSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session)
+            return false;
+        try {
+            const filePath = join(this.sessionDir, `${sessionId}.json`);
+            if (existsSync(filePath)) {
+                unlinkSync(filePath);
+            }
+            this.sessions.delete(sessionId);
+            if (this.currentSession?.id === sessionId) {
+                this.currentSession = null;
+            }
+            logger.info(`Deleted session: ${sessionId}`);
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to delete session', error);
+            return false;
+        }
+    }
+    renameSession(sessionId, newName) {
+        const session = this.sessions.get(sessionId);
+        if (session) {
+            session.name = newName;
+            session.updatedAt = Date.now();
+            this.saveSession(session);
+            return true;
+        }
+        return false;
+    }
+    exportSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (session) {
+            return JSON.stringify(session, null, 2);
+        }
+        return null;
+    }
+    importSession(json) {
+        try {
+            const session = JSON.parse(json);
+            session.id = generateId(24);
+            session.createdAt = Date.now();
+            session.updatedAt = Date.now();
+            this.sessions.set(session.id, session);
+            this.saveSession(session);
+            logger.info(`Imported session: ${session.id} (${session.name})`);
+            return session;
+        }
+        catch (error) {
+            logger.error('Failed to import session', error);
+            return null;
+        }
+    }
+    getSessionCount() {
+        return this.sessions.size;
+    }
+    listSessions() {
+        return this.getAllSessions().map((s) => ({
+            id: s.id,
+            name: s.name,
+            createdAt: formatDate(s.createdAt),
+            messageCount: s.state.messages.length,
+        }));
+    }
+    destroy() {
+        if (this.saveTimer) {
+            clearInterval(this.saveTimer);
+        }
+        this.saveCurrentSession();
+    }
+}
+export const sessionManager = new SessionManager();
+//# sourceMappingURL=index.js.map

package/dist/session/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/session/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACjG,OAAO,EAAE,IAAI,EAAW,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE3D,MAAM,MAAM,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;AAEpC,MAAM,OAAO,cAAc;IACjB,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,QAAQ,CAAU;IAClB,YAAY,CAAS;IACrB,cAAc,GAAuB,IAAI,CAAC;IAC1C,QAAQ,GAA6B,IAAI,GAAG,EAAE,CAAC;IAC/C,SAAS,GAA0C,IAAI,CAAC;IAEhE;QACE,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,EAAE,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;QAC9C,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;QACxC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC;QAEhD,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YACjC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpB,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;gBAChC,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,CAAC,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;QACxB,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,gCAAgC,IAAI,CAAC,QAAQ,CAAC,IAAI,YAAY,CAAC,CAAC;IAC9E,CAAC;IAEO,YAAY;QAClB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC;iBACvC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;iBAClC,IAAI,EAAE;iBACN,OAAO,EAAE,CAAC;YAEb,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;gBACpD,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;oBACnE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;oBACnD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;gBACzC,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,CAAC,gCAAgC,IAAI,EAAE,EAAE,GAAG,CAAC,CAAC;gBAC3D,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,aAAa,CAAC,IAAa;QACzB,MAAM,OAAO,GAAgB;YAC3B,EAAE,EAAE,UAAU,CAAC,EAAE,CAAC;YAClB,IAAI,EAAE,IAAI,IAAI,WAAW,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE;YACjD,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE;YAC9B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;QAC9B,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAE1B,MAAM,CAAC,IAAI,CAAC,oBAAoB,OAAO,CAAC,EAAE,KAAK,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;QAChE,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB;QACtB,OAAO;YACL,IAAI,EAAE,EAAE;YACR,IAAI,EAAE,EAAE;YACR,WAAW,EAAE,CAAC;YACd,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE;gBACP,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE;gBAC/B,SAAS,EAAE,EAAE;gBACb,WAAW,EAAE,EAAE;gBACf,gBAAgB,EAAE,EAAE;gBACpB,WAAW,EAAE,EAAE;gBACf,UAAU,EAAE,CAAC;aACd;YACD,MAAM,EAAE;gBACN,SAAS,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE;gBACzC,QAAQ,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;aAChE;YACD,MAAM,EAAE,MAAM;YACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;IACJ,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,iBAAiB,CAAC,SAAiB;QACjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC;YAC9B,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC;IAC9C,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;aAC/B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/C,CAAC;IAED,kBAAkB,CAAC,KAAiB;QAClC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,cAAc,CAAC,KAAK,GAAG,KAAK,CAAC;YAClC,IAAI,CAAC,cAAc,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,kBAAkB;QAChB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,OAAoB;QACtC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;YAC7D,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACrE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAE3B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;YAC5D,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACzB,UAAU,CAAC,QAAQ,CAAC,CAAC;YACvB,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAEhC,IAAI,IAAI,CAAC,cAAc,EAAE,EAAE,KAAK,SAAS,EAAE,CAAC;gBAC1C,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC7B,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,oBAAoB,SAAS,EAAE,CAAC,CAAC;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YAChD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,aAAa,CAAC,SAAiB,EAAE,OAAe;QAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,GAAG,OAAO,CAAC;YACvB,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC/B,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAC1C,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,aAAa,CAAC,IAAY;QACxB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAgB,CAAC;YAChD,OAAO,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC;YAC5B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC/B,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE/B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACvC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;YAE1B,MAAM,CAAC,IAAI,CAAC,qBAAqB,OAAO,CAAC,EAAE,KAAK,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC;YACjE,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YAChD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;IAC5B,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACvC,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;YAClC,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM;SACtC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,OAAO;QACL,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC"}

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

@@ -0,0 +1,29 @@
+import { AgentState, AgentMessage } from '../types.js';
+export declare class Agent {
+    private state;
+    private provider;
+    private tools;
+    private maxIterations;
+    private currentIteration;
+    constructor();
+    private createInitialState;
+    private initProvider;
+    switchProvider(providerName: string): boolean;
+    private think;
+    private plan;
+    private parsePlanSteps;
+    private executeStep;
+    private resolveParameters;
+    private analyzeResults;
+    private fixErrors;
+    private isTaskComplete;
+    private generateWithRetry;
+    executeTask(task: string): Promise<AgentMessage>;
+    chat(message: string): Promise<AgentMessage>;
+    getState(): AgentState;
+    getMessages(): AgentMessage[];
+    reset(): void;
+    setMaxIterations(max: number): void;
+}
+export declare const agent: Agent;
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/src/agent/index.ts"],"names":[],"mappings":"AAWA,OAAO,EACL,UAAU,EAAE,YAAY,EAEzB,MAAM,aAAa,CAAC;AA2BrB,qBAAa,KAAK;IAChB,OAAO,CAAC,KAAK,CAAa;IAC1B,OAAO,CAAC,QAAQ,CAAkD;IAClE,OAAO,CAAC,KAAK,CAA8B;IAC3C,OAAO,CAAC,aAAa,CAAM;IAC3B,OAAO,CAAC,gBAAgB,CAAK;;IAO7B,OAAO,CAAC,kBAAkB;IAe1B,OAAO,CAAC,YAAY;IAapB,cAAc,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO;YAuB/B,KAAK;YAgCL,IAAI;IA8BlB,OAAO,CAAC,cAAc;YAiCR,WAAW;IA0CzB,OAAO,CAAC,iBAAiB;YAwBX,cAAc;YA6Bd,SAAS;IAqBvB,OAAO,CAAC,cAAc;YAkBR,iBAAiB;IAsDzB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAwIhD,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;IAgDlD,QAAQ,IAAI,UAAU;IAItB,WAAW,IAAI,YAAY,EAAE;IAI7B,KAAK,IAAI,IAAI;IAOb,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;CAGpC;AAED,eAAO,MAAM,KAAK,OAAc,CAAC"}