poe-code 0.1.26

package/dist/services/agent-session.js

@@ -0,0 +1,107 @@
+import path from "node:path";
+import * as fs from "node:fs";
+import { DefaultToolExecutor, getAvailableTools } from "./tools.js";
+import { McpManager } from "./mcp-manager.js";
+import { PoeChatService } from "./chat.js";
+import { AgentTaskRegistry } from "./agent-task-registry.js";
+import { AgentConfigManager } from "./agent-config-manager.js";
+import { createDefaultAgentRegistry } from "./agent-registry.js";
+export async function createAgentSession(options) {
+    const mcpManager = new McpManager(options.fs, options.homeDir);
+    try {
+        await mcpManager.connectAll();
+    }
+    catch (error) {
+        options.logger(`Failed to connect to some MCP servers: ${formatError(error)}`);
+    }
+    let systemPrompt;
+    try {
+        const systemPromptPath = path.join(options.cwd, "SYSTEM_PROMPT.md");
+        systemPrompt = await options.fs.readFile(systemPromptPath, "utf8");
+    }
+    catch {
+        systemPrompt = undefined;
+    }
+    const cwdContext = `\n\nIMPORTANT: You are working in the directory: ${options.cwd}\nWhen accessing files, use paths relative to this directory.`;
+    systemPrompt = systemPrompt ? `${systemPrompt}${cwdContext}` : cwdContext;
+    const tasksDir = path.join(options.homeDir, ".poe-code", "tasks");
+    const logsDir = path.join(options.homeDir, ".poe-code", "logs", "tasks");
+    const nativeFs = fs;
+    const taskRegistry = new AgentTaskRegistry({
+        fs: nativeFs,
+        tasksDir,
+        logsDir,
+        logger: (event, payload) => {
+            options.logger(`task:${event} ${JSON.stringify(payload ?? {})}`);
+        }
+    });
+    const awaitTasksOnDispose = options.awaitTasksOnDispose ?? false;
+    const agentRegistry = createDefaultAgentRegistry();
+    const agentConfigManager = new AgentConfigManager({
+        fs: options.fs,
+        homeDir: options.homeDir,
+        registry: agentRegistry
+    });
+    try {
+        await agentConfigManager.loadConfig();
+    }
+    catch (error) {
+        options.logger(`Failed to load agent configuration: ${formatError(error)}`);
+    }
+    const toolExecutor = new DefaultToolExecutor({
+        fs: options.fs,
+        cwd: options.cwd,
+        allowedPaths: [options.cwd, options.homeDir],
+        mcpManager,
+        taskRegistry,
+        onWriteFile: async ({ relativePath }) => {
+            options.logger(`Tool write_file -> ${relativePath}`);
+        },
+        agentRegistry,
+        agentConfigManager,
+        homeDir: options.homeDir
+    });
+    let toolCallback;
+    const chatService = new PoeChatService(options.apiKey, options.model, toolExecutor, (event) => {
+        toolCallback?.(event);
+    }, systemPrompt, taskRegistry);
+    return {
+        getModel() {
+            return chatService.getModel();
+        },
+        setToolCallCallback(callback) {
+            toolCallback = callback;
+            chatService.setToolCallCallback(callback);
+        },
+        async sendMessage(prompt, options) {
+            const tools = await getAvailableTools({
+                agentRegistry,
+                agentConfigManager,
+                mcpManager
+            });
+            return chatService.sendMessage(prompt, tools, options);
+        },
+        async waitForAllTasks() {
+            await taskRegistry.waitForAllTasks();
+        },
+        drainCompletedTasks() {
+            const completed = taskRegistry.getCompletedTasks();
+            taskRegistry.clearCompleted();
+            return completed;
+        },
+        async dispose() {
+            if (awaitTasksOnDispose) {
+                await taskRegistry.waitForAllTasks();
+            }
+            taskRegistry.dispose();
+            await mcpManager.disconnectAll();
+        }
+    };
+}
+function formatError(error) {
+    if (error instanceof Error) {
+        return error.message;
+    }
+    return String(error);
+}
+//# sourceMappingURL=agent-session.js.map

package/dist/services/agent-session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-session.js","sourceRoot":"","sources":["../../src/services/agent-session.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAE9B,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EACL,cAAc,EAGf,MAAM,WAAW,CAAC;AACnB,OAAO,EACL,iBAAiB,EAGlB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAwBjE,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAA4B;IAE5B,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/D,IAAI,CAAC;QACH,MAAM,UAAU,CAAC,UAAU,EAAE,CAAC;IAChC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,MAAM,CACZ,0CAA0C,WAAW,CAAC,KAAK,CAAC,EAAE,CAC/D,CAAC;IACJ,CAAC;IAED,IAAI,YAAgC,CAAC;IACrC,IAAI,CAAC;QACH,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;QACpE,YAAY,GAAG,MAAM,OAAO,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC;IACrE,CAAC;IAAC,MAAM,CAAC;QACP,YAAY,GAAG,SAAS,CAAC;IAC3B,CAAC;IAED,MAAM,UAAU,GAAG,oDAAoD,OAAO,CAAC,GAAG,+DAA+D,CAAC;IAClJ,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,GAAG,YAAY,GAAG,UAAU,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;IAE1E,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IAClE,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACzE,MAAM,QAAQ,GAAG,EAAuB,CAAC;IACzC,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC;QACzC,EAAE,EAAE,QAAQ;QACZ,QAAQ;QACR,OAAO;QACP,MAAM,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;YACzB,OAAO,CAAC,MAAM,CAAC,QAAQ,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;KACF,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,IAAI,KAAK,CAAC;IAEjE,MAAM,aAAa,GAAG,0BAA0B,EAAE,CAAC;IACnD,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,CAAC;QAChD,EAAE,EAAE,OAAO,CAAC,EAAE;QACd,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,QAAQ,EAAE,aAAa;KACxB,CAAC,CAAC;IACH,IAAI,CAAC;QACH,MAAM,kBAAkB,CAAC,UAAU,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,MAAM,CACZ,uCAAuC,WAAW,CAAC,KAAK,CAAC,EAAE,CAC5D,CAAC;IACJ,CAAC;IAED,MAAM,YAAY,GAAG,IAAI,mBAAmB,CAAC;QAC3C,EAAE,EAAE,OAAO,CAAC,EAAE;QACd,GAAG,EAAE,OAAO,CAAC,GAAG;QAChB,YAAY,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,OAAO,CAAC;QAC5C,UAAU;QACV,YAAY;QACZ,WAAW,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;YACtC,OAAO,CAAC,MAAM,CAAC,sBAAsB,YAAY,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,aAAa;QACb,kBAAkB;QAClB,OAAO,EAAE,OAAO,CAAC,OAAO;KACzB,CAAC,CAAC;IAEH,IAAI,YAA0C,CAAC;IAC/C,MAAM,WAAW,GAAG,IAAI,cAAc,CACpC,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,KAAK,EACb,YAAY,EACZ,CAAC,KAAK,EAAE,EAAE;QACR,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC,EACD,YAAY,EACZ,YAAY,CACb,CAAC;IAEF,OAAO;QACL,QAAQ;YACN,OAAO,WAAW,CAAC,QAAQ,EAAE,CAAC;QAChC,CAAC;QACD,mBAAmB,CAAC,QAA0B;YAC5C,YAAY,GAAG,QAAQ,CAAC;YACxB,WAAW,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC5C,CAAC;QACD,KAAK,CAAC,WAAW,CACf,MAAc,EACd,OAAqE;YAErE,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;gBACpC,aAAa;gBACb,kBAAkB;gBAClB,UAAU;aACX,CAAC,CAAC;YACH,OAAO,WAAW,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACzD,CAAC;QACD,KAAK,CAAC,eAAe;YACnB,MAAM,YAAY,CAAC,eAAe,EAAE,CAAC;QACvC,CAAC;QACD,mBAAmB;YACjB,MAAM,SAAS,GAAG,YAAY,CAAC,iBAAiB,EAAE,CAAC;YACnD,YAAY,CAAC,cAAc,EAAE,CAAC;YAC9B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,KAAK,CAAC,OAAO;YACX,IAAI,mBAAmB,EAAE,CAAC;gBACxB,MAAM,YAAY,CAAC,eAAe,EAAE,CAAC;YACvC,CAAC;YACD,YAAY,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,UAAU,CAAC,aAAa,EAAE,CAAC;QACnC,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,KAAc;IACjC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;AACvB,CAAC"}

package/dist/services/agent-task-registry.d.ts

@@ -0,0 +1,131 @@
+import type { FSWatcher, Stats } from "node:fs";
+export type AgentTaskStatus = "running" | "completed" | "failed" | "killed";
+export interface AgentTask {
+    id: string;
+    toolName: string;
+    args: Record<string, unknown>;
+    status: AgentTaskStatus;
+    startTime: number;
+    endTime?: number;
+    result?: string;
+    error?: string;
+    logFile: string;
+    progressFile: string;
+    pid?: number;
+    command?: string;
+    worktreePath?: string;
+}
+export interface ProgressUpdate {
+    type: "progress" | "complete" | "error";
+    message?: string;
+    result?: string;
+    error?: string;
+    timestamp: number;
+}
+type MinimalStats = Pick<Stats, "size" | "mtimeMs">;
+export interface FsLike {
+    promises: {
+        mkdir(path: string, options?: {
+            recursive?: boolean;
+        }): Promise<void>;
+        readFile(path: string, encoding: BufferEncoding): Promise<string>;
+        writeFile(path: string, data: string | Uint8Array, options?: {
+            encoding?: BufferEncoding;
+        }): Promise<void>;
+        appendFile(path: string, data: string | Uint8Array, options?: {
+            encoding?: BufferEncoding;
+        }): Promise<void>;
+        readdir(path: string): Promise<string[]>;
+        stat(path: string): Promise<MinimalStats>;
+        unlink(path: string): Promise<void>;
+        rename(oldPath: string, newPath: string): Promise<void>;
+    };
+    watch(filename: string, listener: (event: string, filename: string | null) => void): FSWatcher;
+    existsSync(path: string): boolean;
+    mkdirSync(path: string, options?: {
+        recursive?: boolean;
+    }): void;
+    readFileSync(path: string, encoding: BufferEncoding): string;
+    writeFileSync(path: string, data: string, options?: {
+        encoding?: BufferEncoding;
+    }): void;
+    appendFileSync(path: string, data: string): void;
+    statSync(path: string): MinimalStats;
+    unlinkSync(path: string): void;
+    renameSync(oldPath: string, newPath: string): void;
+    readdirSync(path: string): string[];
+}
+export interface AgentTaskRegistryOptions {
+    fs: FsLike;
+    tasksDir: string;
+    logsDir: string;
+    now?: () => number;
+    maxTasks?: number;
+    archiveAfterMs?: number;
+    logRetentionMs?: number;
+    completionRetentionMs?: number;
+    watchFactory?: (dir: string, listener: (event: string, filename: string | null) => void) => FSWatcher;
+    debounceMs?: number;
+    logger?: (event: string, payload?: Record<string, unknown>) => void;
+}
+export declare class AgentTaskRegistry {
+    private readonly fs;
+    private readonly tasksDir;
+    private readonly logsDir;
+    private readonly now;
+    private readonly maxTasks;
+    private readonly archiveAfterMs;
+    private readonly logRetentionMs;
+    private readonly completionRetentionMs;
+    private readonly watchFactory?;
+    private readonly debounceMs;
+    private readonly logger?;
+    private watcher?;
+    private readonly tasks;
+    private readonly taskOrder;
+    private readonly completionCallbacks;
+    private readonly progressCallbacks;
+    private readonly completedQueue;
+    private readonly completedSet;
+    private readonly pendingDebounce;
+    constructor(options: AgentTaskRegistryOptions);
+    get size(): number;
+    getTasksDirectory(): string;
+    getLogsDirectory(): string;
+    registerTask(input: {
+        toolName: string;
+        args: Record<string, unknown>;
+    }): string;
+    updateTask(id: string, patch: Partial<AgentTask>): void;
+    getTask(id: string): AgentTask | undefined;
+    getAllTasks(): AgentTask[];
+    getRunningTasks(): AgentTask[];
+    getCompletedTasks(): AgentTask[];
+    clearCompleted(): void;
+    onTaskComplete(callback: (task: AgentTask) => void): void;
+    onTaskProgress(callback: (taskId: string, update: ProgressUpdate) => void): void;
+    killTask(id: string): boolean;
+    readProgress(taskId: string): ProgressUpdate[];
+    waitForTask(id: string, onUpdate: (update: ProgressUpdate) => void): Promise<AgentTask | undefined>;
+    dispose(): void;
+    waitForAllTasks(): Promise<void>;
+    private ensureDirectories;
+    private bootstrapFromDisk;
+    private loadExistingTasks;
+    private cleanupCompletedTasks;
+    private cleanupArchives;
+    private cleanupLogs;
+    private startWatching;
+    private handleFileChange;
+    private enqueueCompletion;
+    private readLatestProgress;
+    archiveTask(id: string): void;
+    private trimCache;
+    private ensureFile;
+    private persistTask;
+    private loadTask;
+    private createTaskId;
+    private isTerminal;
+    private isProcessAlive;
+}
+export {};