@researchcomputer/agents-sdk 0.1.0

package/dist/factory.js

@@ -0,0 +1,282 @@
+import { createHash, randomUUID } from 'node:crypto';
+import { Agent } from '@mariozechner/pi-agent-core';
+import { createRunContext } from './context/run-context.js';
+import { convertToLlm } from './context/converter.js';
+import { buildSystemPrompt } from './context/system-prompt.js';
+import { createCompressionMiddleware } from './context/compression.js';
+import { createPermissionMiddleware } from './middleware/permission-middleware.js';
+import { composePipeline } from './middleware/pipeline.js';
+import { runLifecycleHooks } from './middleware/hooks.js';
+import { createCostTracker } from './observability/cost-tracker.js';
+import { createMemoryManager } from './memory/memory.js';
+import { createSessionManager } from './session/session.js';
+import { createMcpManager } from './mcp/client.js';
+import { getAllTools } from './tools/index.js';
+import { createSwarmManager } from './agents/swarm.js';
+import { createSwarmTools } from './agents/tools.js';
+import { composeCodingAgentConfig } from './skills.js';
+/**
+ * Main factory function. Creates a fully wired CodingAgent.
+ */
+export async function createCodingAgent(config) {
+    const cwd = config.cwd ?? process.cwd();
+    const permissionMode = config.permissionMode ?? 'default';
+    const enableMemory = config.enableMemory ?? true;
+    // 1. Create RunContext
+    const runContext = createRunContext({
+        cwd,
+        sessionId: config.sessionId,
+    });
+    // 2. Load memories
+    const memoryManager = createMemoryManager();
+    const memoryDir = config.memoryDir ?? `${process.env.HOME ?? '~'}/.rc-agents/memory`;
+    let memories = [];
+    let memorySelections = [];
+    if (enableMemory) {
+        try {
+            memories = await memoryManager.load(memoryDir);
+            memorySelections = memoryManager.retrieve(memories, { query: '' });
+        }
+        catch {
+            // Memory loading is non-critical
+        }
+    }
+    // 3. Create tools
+    const defaultTools = getAllTools({
+        cwd,
+        onQuestion: config.onQuestion,
+    });
+    const composedConfig = composeCodingAgentConfig(config, { defaultTools });
+    const tools = composedConfig.tools;
+    const hooks = composedConfig.hooks;
+    const permissionRules = composedConfig.permissionRules;
+    // 4. Connect MCP servers
+    const mcp = createMcpManager();
+    if (composedConfig.mcpServers.length > 0) {
+        for (const serverConfig of composedConfig.mcpServers) {
+            try {
+                await mcp.connect(serverConfig);
+            }
+            catch {
+                // Log error per server but continue
+            }
+        }
+    }
+    // Merge MCP tools with built-in tools
+    const allTools = [...tools, ...mcp.getTools()];
+    // 5. Build system prompt
+    const basePrompt = composedConfig.systemPrompt ?? 'You are a coding agent that helps users with software development tasks.';
+    const systemPrompt = buildSystemPrompt({
+        basePrompt,
+        skills: composedConfig.skills,
+        tools: allTools,
+        memories: memorySelections,
+        permissionMode,
+    });
+    // 6. Create permission middleware
+    const permissionGate = createPermissionMiddleware({
+        mode: permissionMode,
+        rules: permissionRules,
+        tools: allTools,
+        runContext,
+        onAsk: config.onPermissionAsk,
+    });
+    // 7. Compose pipeline
+    const pipeline = composePipeline({
+        hooks,
+        permissionGate,
+        runContext,
+    });
+    // 8. Create compression middleware
+    const maxContextTokens = config.maxContextTokens ??
+        Math.max(4000, Math.floor((config.model.contextWindow || 128000) * 0.8));
+    const transformContext = createCompressionMiddleware({
+        maxTokens: maxContextTokens,
+        strategy: config.compressionStrategy ?? 'truncate',
+        model: config.model,
+    });
+    // 9. Create cost tracker
+    const costTracker = createCostTracker();
+    // 10. Create Agent
+    const agent = new Agent({
+        initialState: {
+            systemPrompt,
+            model: config.model,
+            tools: allTools,
+            thinkingLevel: config.thinkingLevel ?? 'off',
+        },
+        convertToLlm,
+        transformContext,
+        beforeToolCall: pipeline.beforeToolCall,
+        afterToolCall: pipeline.afterToolCall,
+        getApiKey: config.getApiKey,
+        toolExecution: config.toolExecution ?? 'parallel',
+        streamFn: config.streamFn,
+    });
+    // Wire cost tracker to agent events
+    agent.subscribe((event) => {
+        if (event.type === 'message_end') {
+            const msg = event.message;
+            if (msg.role === 'assistant' && msg.usage) {
+                costTracker.record(msg.usage, config.model.id);
+            }
+        }
+    });
+    // 11. Create session/memory managers
+    const sessionDir = config.sessionDir ?? `${process.env.HOME ?? '~'}/.rc-agents/sessions`;
+    const sessions = createSessionManager(sessionDir);
+    // 12. Optionally create swarm
+    let swarm;
+    if (config.enableSwarm) {
+        swarm = createSwarmManager({
+            model: config.model,
+            tools: allTools,
+            convertToLlm,
+            getApiKey: config.getApiKey,
+            beforeToolCall: pipeline.beforeToolCall,
+            afterToolCall: pipeline.afterToolCall,
+            transformContext,
+        });
+        // Create default team and swarm tools
+        swarm.createTeam({ name: 'default', model: config.model });
+        const swarmTools = createSwarmTools('default', swarm);
+        const currentTools = [...allTools, ...swarmTools];
+        agent.state.tools = currentTools;
+    }
+    // 13. Resume session if sessionId provided
+    if (config.sessionId) {
+        try {
+            const snapshot = await sessions.load(config.sessionId);
+            if (snapshot) {
+                agent.state.messages = snapshot.messages;
+            }
+        }
+        catch {
+            // Session loading is non-critical
+        }
+    }
+    // 14. Run SessionStart hooks
+    await runLifecycleHooks(hooks, 'SessionStart', runContext);
+    // Compute system prompt hash for session snapshots
+    const systemPromptHash = createHash('sha256')
+        .update(systemPrompt)
+        .digest('hex')
+        .slice(0, 16);
+    async function _spawnChildren(baseMessages, message, n) {
+        if (n < 0)
+            throw new RangeError(`fork: n must be >= 0, got ${n}`);
+        if (n === 0)
+            return [];
+        const childConfig = {
+            ...config,
+            sessionId: undefined,
+            autoFork: undefined,
+        };
+        const children = await Promise.all(Array.from({ length: n }, () => createCodingAgent(childConfig)));
+        await Promise.all(children.map((child) => {
+            child.agent.replaceMessages(structuredClone(baseMessages));
+            return child.prompt(message);
+        }));
+        return children;
+    }
+    const codingAgent = {
+        agent,
+        mcp,
+        sessions,
+        memory: memoryManager,
+        swarm,
+        costTracker,
+        async prompt(message, images) {
+            await agent.prompt(message, images);
+        },
+        async dispose() {
+            // Run SessionEnd hooks
+            await runLifecycleHooks(hooks, 'SessionEnd', runContext);
+            // Save session snapshot
+            try {
+                await sessions.save({
+                    version: 1,
+                    id: runContext.sessionId,
+                    messages: agent.state.messages,
+                    modelId: config.model.id,
+                    providerName: config.model.provider ?? 'unknown',
+                    systemPromptHash,
+                    memoryRefs: memories.map(m => m.name),
+                    createdAt: Date.now(),
+                    updatedAt: Date.now(),
+                });
+            }
+            catch {
+                // Session save is non-critical
+            }
+            // Disconnect MCP servers
+            for (const conn of mcp.getConnections()) {
+                try {
+                    await mcp.disconnect(conn.name);
+                }
+                catch {
+                    // MCP disconnect is non-critical
+                }
+            }
+            // Destroy swarm if present
+            if (swarm) {
+                try {
+                    await swarm.destroyTeam('default');
+                }
+                catch {
+                    // Swarm cleanup is non-critical
+                }
+            }
+        },
+        snapshot() {
+            if (agent.state.isStreaming) {
+                throw new Error('snapshot: cannot snapshot while agent is streaming');
+            }
+            return {
+                id: randomUUID(),
+                messages: structuredClone(agent.state.messages),
+                createdAt: Date.now(),
+            };
+        },
+        restore(snapshot) {
+            if (agent.state.isStreaming) {
+                throw new Error('restore: cannot restore while agent is streaming');
+            }
+            agent.replaceMessages(structuredClone(snapshot.messages));
+        },
+        async fork(message, n) {
+            return _spawnChildren(structuredClone(agent.state.messages), message, n);
+        },
+        async promptFork(message, n) {
+            return _spawnChildren(structuredClone(agent.state.messages), message, n);
+        },
+        async forkFrom(snapshot, message, n) {
+            return _spawnChildren(structuredClone(snapshot.messages), message, n);
+        },
+    };
+    // Wire auto-fork if configured
+    if (config.autoFork) {
+        const autoFork = config.autoFork;
+        let lastUserMessage;
+        agent.subscribe((event) => {
+            if (event.type === 'message_start') {
+                const msg = event.message;
+                if (msg.role === 'user') {
+                    const textPart = msg.content.find((p) => p.type === 'text');
+                    if (textPart)
+                        lastUserMessage = textPart.text;
+                }
+            }
+            if (event.type === 'turn_end' && lastUserMessage) {
+                const message = lastUserMessage;
+                lastUserMessage = undefined;
+                codingAgent
+                    .fork(message, autoFork.branches)
+                    .then(children => autoFork.onBranches(children))
+                    .catch(() => { });
+            }
+        });
+    }
+    return codingAgent;
+}
+//# sourceMappingURL=factory.js.map

package/dist/factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.js","sourceRoot":"","sources":["../src/factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AAiBpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,EAAE,2BAA2B,EAAE,MAAM,0BAA0B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,uCAAuC,CAAC;AACnF,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAsDvD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAyB;IAC/D,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IACxC,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,SAAS,CAAC;IAC1D,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC;IAEjD,uBAAuB;IACvB,MAAM,UAAU,GAAG,gBAAgB,CAAC;QAClC,GAAG;QACH,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAC,CAAC;IAEH,mBAAmB;IACnB,MAAM,aAAa,GAAG,mBAAmB,EAAE,CAAC;IAC5C,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,oBAAoB,CAAC;IACrF,IAAI,QAAQ,GAA+C,EAAE,CAAC;IAC9D,IAAI,gBAAgB,GAA0C,EAAE,CAAC;IAEjE,IAAI,YAAY,EAAE,CAAC;QACjB,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,gBAAgB,GAAG,aAAa,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACrE,CAAC;QAAC,MAAM,CAAC;YACP,iCAAiC;QACnC,CAAC;IACH,CAAC;IAED,kBAAkB;IAClB,MAAM,YAAY,GAAwB,WAAW,CAAC;QACpD,GAAG;QACH,UAAU,EAAE,MAAM,CAAC,UAAU;KAC9B,CAAC,CAAC;IACH,MAAM,cAAc,GAAG,wBAAwB,CAAC,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;IAC1E,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC;IACnC,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC;IACnC,MAAM,eAAe,GAAG,cAAc,CAAC,eAAe,CAAC;IAEvD,yBAAyB;IACzB,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,IAAI,cAAc,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzC,KAAK,MAAM,YAAY,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC;YACrD,IAAI,CAAC;gBACH,MAAM,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YAClC,CAAC;YAAC,MAAM,CAAC;gBACP,oCAAoC;YACtC,CAAC;QACH,CAAC;IACH,CAAC;IAED,sCAAsC;IACtC,MAAM,QAAQ,GAAG,CAAC,GAAG,KAAK,EAAE,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;IAE/C,yBAAyB;IACzB,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,IAAI,0EAA0E,CAAC;IAC7H,MAAM,YAAY,GAAG,iBAAiB,CAAC;QACrC,UAAU;QACV,MAAM,EAAE,cAAc,CAAC,MAAM;QAC7B,KAAK,EAAE,QAAQ;QACf,QAAQ,EAAE,gBAAgB;QAC1B,cAAc;KACf,CAAC,CAAC;IAEH,kCAAkC;IAClC,MAAM,cAAc,GAAG,0BAA0B,CAAC;QAChD,IAAI,EAAE,cAAc;QACpB,KAAK,EAAE,eAAe;QACtB,KAAK,EAAE,QAAQ;QACf,UAAU;QACV,KAAK,EAAE,MAAM,CAAC,eAAe;KAC9B,CAAC,CAAC;IAEH,sBAAsB;IACtB,MAAM,QAAQ,GAAG,eAAe,CAAC;QAC/B,KAAK;QACL,cAAc;QACd,UAAU;KACX,CAAC,CAAC;IAEH,mCAAmC;IACnC,MAAM,gBAAgB,GAAG,MAAM,CAAC,gBAAgB;QAC9C,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;IAE3E,MAAM,gBAAgB,GAAG,2BAA2B,CAAC;QACnD,SAAS,EAAE,gBAAgB;QAC3B,QAAQ,EAAE,MAAM,CAAC,mBAAmB,IAAI,UAAU;QAClD,KAAK,EAAE,MAAM,CAAC,KAAK;KACpB,CAAC,CAAC;IAEH,yBAAyB;IACzB,MAAM,WAAW,GAAG,iBAAiB,EAAE,CAAC;IAExC,mBAAmB;IACnB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;QACtB,YAAY,EAAE;YACZ,YAAY;YACZ,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,KAAK,EAAE,QAAQ;YACf,aAAa,EAAE,MAAM,CAAC,aAAa,IAAK,KAAuB;SAChE;QACD,YAAY;QACZ,gBAAgB;QAChB,cAAc,EAAE,QAAQ,CAAC,cAAc;QACvC,aAAa,EAAE,QAAQ,CAAC,aAAa;QACrC,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,UAAU;QACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ;KAC1B,CAAC,CAAC;IAEH,oCAAoC;IACpC,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;QACxB,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACjC,MAAM,GAAG,GAAG,KAAK,CAAC,OAAwE,CAAC;YAC3F,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBAC1C,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACjD,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,qCAAqC;IACrC,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,sBAAsB,CAAC;IACzF,MAAM,QAAQ,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;IAElD,8BAA8B;IAC9B,IAAI,KAA+B,CAAC;IACpC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,KAAK,GAAG,kBAAkB,CAAC;YACzB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,KAAK,EAAE,QAAQ;YACf,YAAY;YACZ,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,cAAc,EAAE,QAAQ,CAAC,cAAc;YACvC,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,gBAAgB;SACjB,CAAC,CAAC;QAEH,sCAAsC;QACtC,KAAK,CAAC,UAAU,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;QAC3D,MAAM,UAAU,GAAG,gBAAgB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QACtD,MAAM,YAAY,GAAG,CAAC,GAAG,QAAQ,EAAE,GAAG,UAAU,CAAC,CAAC;QAClD,KAAK,CAAC,KAAK,CAAC,KAAK,GAAG,YAAY,CAAC;IACnC,CAAC;IAED,2CAA2C;IAC3C,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACvD,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,CAAC,KAAK,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC;YAC3C,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,kCAAkC;QACpC,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,MAAM,iBAAiB,CAAC,KAAK,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;IAE3D,mDAAmD;IACnD,MAAM,gBAAgB,GAAG,UAAU,CAAC,QAAQ,CAAC;SAC1C,MAAM,CAAC,YAAY,CAAC;SACpB,MAAM,CAAC,KAAK,CAAC;SACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEhB,KAAK,UAAU,cAAc,CAC3B,YAA4B,EAC5B,OAAe,EACf,CAAS;QAET,IAAI,CAAC,GAAG,CAAC;YAAE,MAAM,IAAI,UAAU,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC;QAClE,IAAI,CAAC,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEvB,MAAM,WAAW,GAAsB;YACrC,GAAG,MAAM;YACT,SAAS,EAAE,SAAS;YACpB,QAAQ,EAAE,SAAS;SACpB,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,GAAG,CAChC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC,CAChE,CAAC;QAEF,MAAM,OAAO,CAAC,GAAG,CACf,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACrB,KAAK,CAAC,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,YAAY,CAAmB,CAAC,CAAC;YAC7E,OAAO,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CACH,CAAC;QAEF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,WAAW,GAAgB;QAC/B,KAAK;QACL,GAAG;QACH,QAAQ;QACR,MAAM,EAAE,aAAa;QACrB,KAAK;QACL,WAAW;QAEX,KAAK,CAAC,MAAM,CAAC,OAAe,EAAE,MAAuB;YACnD,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,KAAK,CAAC,OAAO;YACX,uBAAuB;YACvB,MAAM,iBAAiB,CAAC,KAAK,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;YAEzD,wBAAwB;YACxB,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,IAAI,CAAC;oBAClB,OAAO,EAAE,CAAC;oBACV,EAAE,EAAE,UAAU,CAAC,SAAS;oBACxB,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,QAAQ;oBAC9B,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;oBACxB,YAAY,EAAG,MAAM,CAAC,KAAa,CAAC,QAAQ,IAAI,SAAS;oBACzD,gBAAgB;oBAChB,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;oBACrC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;YAED,yBAAyB;YACzB,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,MAAM,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAClC,CAAC;gBAAC,MAAM,CAAC;oBACP,iCAAiC;gBACnC,CAAC;YACH,CAAC;YAED,2BAA2B;YAC3B,IAAI,KAAK,EAAE,CAAC;gBACV,IAAI,CAAC;oBACH,MAAM,KAAK,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;gBACrC,CAAC;gBAAC,MAAM,CAAC;oBACP,gCAAgC;gBAClC,CAAC;YACH,CAAC;QACH,CAAC;QAED,QAAQ;YACN,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;YACxE,CAAC;YACD,OAAO;gBACL,EAAE,EAAE,UAAU,EAAE;gBAChB,QAAQ,EAAE,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAC/C,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;QACJ,CAAC;QAED,OAAO,CAAC,QAAuB;YAC7B,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;YACtE,CAAC;YACD,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,OAAe,EAAE,CAAS;YACnC,OAAO,cAAc,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,OAAe,EAAE,CAAS;YACzC,OAAO,cAAc,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,KAAK,CAAC,QAAQ,CAAC,QAAuB,EAAE,OAAe,EAAE,CAAS;YAChE,OAAO,cAAc,CAAC,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QACxE,CAAC;KACF,CAAC;IAEF,+BAA+B;IAC/B,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,IAAI,eAAmC,CAAC;QAExC,KAAK,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;YACxB,IAAI,KAAK,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC;gBAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,MAAM,QAAQ,GAAI,GAAG,CAAC,OAAiB,CAAC,IAAI,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;oBAC5E,IAAI,QAAQ;wBAAE,eAAe,GAAG,QAAQ,CAAC,IAAI,CAAC;gBAChD,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,eAAe,EAAE,CAAC;gBACjD,MAAM,OAAO,GAAG,eAAe,CAAC;gBAChC,eAAe,GAAG,SAAS,CAAC;gBAC5B,WAAW;qBACR,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC;qBAChC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;qBAC/C,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACrB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC"}

package/dist/factory.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=factory.test.d.ts.map

package/dist/factory.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.test.d.ts","sourceRoot":"","sources":["../src/factory.test.ts"],"names":[],"mappings":""}

package/dist/factory.test.js

@@ -0,0 +1,63 @@
+import { describe, it, expect } from 'vitest';
+import { getModel } from '@mariozechner/pi-ai';
+import { createCodingAgent } from './factory.js';
+describe('createCodingAgent', () => {
+    it('returns an object with all expected properties', async () => {
+        const model = getModel('openai', 'gpt-4o-mini');
+        const agent = await createCodingAgent({
+            model,
+            permissionMode: 'allowAll',
+        });
+        // Verify all expected properties exist
+        expect(agent).toHaveProperty('agent');
+        expect(agent).toHaveProperty('mcp');
+        expect(agent).toHaveProperty('sessions');
+        expect(agent).toHaveProperty('memory');
+        expect(agent).toHaveProperty('costTracker');
+        expect(agent).toHaveProperty('prompt');
+        expect(agent).toHaveProperty('dispose');
+        // Verify types
+        expect(typeof agent.prompt).toBe('function');
+        expect(typeof agent.dispose).toBe('function');
+        // No swarm by default
+        expect(agent.swarm).toBeUndefined();
+        // Cost tracker works
+        expect(agent.costTracker.total()).toEqual({ tokens: 0, cost: 0 });
+        // Clean up (don't actually call LLM)
+        await agent.dispose();
+    });
+    it('creates swarm when enableSwarm is true', async () => {
+        const model = getModel('openai', 'gpt-4o-mini');
+        const agent = await createCodingAgent({
+            model,
+            permissionMode: 'allowAll',
+            enableSwarm: true,
+        });
+        expect(agent.swarm).toBeDefined();
+        await agent.dispose();
+    });
+    it('uses custom tools when provided', async () => {
+        const model = getModel('openai', 'gpt-4o-mini');
+        const { Type } = await import('@sinclair/typebox');
+        const customTool = {
+            name: 'custom',
+            label: 'Custom Tool',
+            description: 'A custom tool',
+            parameters: Type.Object({}),
+            capabilities: [],
+            async execute() {
+                return { content: [{ type: 'text', text: 'done' }], details: {} };
+            },
+        };
+        const agent = await createCodingAgent({
+            model,
+            permissionMode: 'allowAll',
+            tools: [customTool],
+        });
+        // Agent should have the custom tool
+        const toolNames = agent.agent.state.tools.map(t => t.name);
+        expect(toolNames).toContain('custom');
+        await agent.dispose();
+    });
+});
+//# sourceMappingURL=factory.test.js.map

package/dist/factory.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"factory.test.js","sourceRoot":"","sources":["../src/factory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEjD,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;YACpC,KAAK;YACL,cAAc,EAAE,UAAU;SAC3B,CAAC,CAAC;QAEH,uCAAuC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QAExC,eAAe;QACf,MAAM,CAAC,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAE9C,sBAAsB;QACtB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QAEpC,qBAAqB;QACrB,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAElE,qCAAqC;QACrC,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;YACpC,KAAK;YACL,cAAc,EAAE,UAAU;YAC1B,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;QAElC,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAChD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAEnD,MAAM,UAAU,GAAG;YACjB,IAAI,EAAE,QAAQ;YACd,KAAK,EAAE,aAAa;YACpB,WAAW,EAAE,eAAe;YAC5B,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3B,YAAY,EAAE,EAAW;YACzB,KAAK,CAAC,OAAO;gBACX,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAC7E,CAAC;SACF,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,iBAAiB,CAAC;YACpC,KAAK;YACL,cAAc,EAAE,UAAU;YAC1B,KAAK,EAAE,CAAC,UAAU,CAAC;SACpB,CAAC,CAAC;QAEH,oCAAoC;QACpC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAEtC,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/fork.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=fork.test.d.ts.map

package/dist/fork.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fork.test.d.ts","sourceRoot":"","sources":["../src/fork.test.ts"],"names":[],"mappings":""}