@astrive-ai/planner 1.0.1

package/dist/index.d.ts

@@ -0,0 +1,4 @@
+export * from './planner.js';
+export * from './simple-planner.js';
+export * from './plan-optimizer.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC"}

package/dist/index.js

@@ -0,0 +1,4 @@
+export * from './planner.js';
+export * from './simple-planner.js';
+export * from './plan-optimizer.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC"}

package/dist/plan-optimizer.d.ts

@@ -0,0 +1,6 @@
+import type { ExecutionPlan, PlanStep } from '@astrive-ai/types';
+export declare class PlanOptimizer {
+    optimize(plan: ExecutionPlan): ExecutionPlan;
+    detectParallelizable(steps: PlanStep[]): string[][];
+}
+//# sourceMappingURL=plan-optimizer.d.ts.map

package/dist/plan-optimizer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"plan-optimizer.d.ts","sourceRoot":"","sources":["../src/plan-optimizer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAEjE,qBAAa,aAAa;IACxB,QAAQ,CAAC,IAAI,EAAE,aAAa,GAAG,aAAa;IAkB5C,oBAAoB,CAAC,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,EAAE,EAAE;CAoBpD"}

package/dist/plan-optimizer.js

@@ -0,0 +1,30 @@
+export class PlanOptimizer {
+    optimize(plan) {
+        // Basic optimization: reorder based on dependencies, mark parallel
+        const steps = [...plan.steps];
+        // Remove duplicate steps (same tool, same args)
+        const uniqueSteps = steps.filter((step, index, self) => index === self.findIndex((t) => (t.toolName === step.toolName &&
+            JSON.stringify(t.arguments) === JSON.stringify(step.arguments))));
+        return {
+            ...plan,
+            steps: uniqueSteps
+        };
+    }
+    detectParallelizable(steps) {
+        const levels = [];
+        const processed = new Set();
+        let currentLevel = steps
+            .filter(s => !s.dependsOn || s.dependsOn.length === 0)
+            .map(s => s.id);
+        while (currentLevel.length > 0) {
+            levels.push(currentLevel);
+            currentLevel.forEach(id => processed.add(id));
+            currentLevel = steps
+                .filter(s => !processed.has(s.id))
+                .filter(s => s.dependsOn && s.dependsOn.every(dep => processed.has(dep)))
+                .map(s => s.id);
+        }
+        return levels;
+    }
+}
+//# sourceMappingURL=plan-optimizer.js.map

package/dist/plan-optimizer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plan-optimizer.js","sourceRoot":"","sources":["../src/plan-optimizer.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,aAAa;IACxB,QAAQ,CAAC,IAAmB;QAC1B,mEAAmE;QACnE,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAE9B,gDAAgD;QAChD,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,CACrD,KAAK,KAAK,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAC9B,CAAC,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ;YAC5B,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAC/D,CAAC,CACH,CAAC;QAEF,OAAO;YACL,GAAG,IAAI;YACP,KAAK,EAAE,WAAW;SACnB,CAAC;IACJ,CAAC;IAED,oBAAoB,CAAC,KAAiB;QACpC,MAAM,MAAM,GAAe,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QAEpC,IAAI,YAAY,GAAG,KAAK;aACrB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC;aACrD,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAElB,OAAO,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC1B,YAAY,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YAE9C,YAAY,GAAG,KAAK;iBACjB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;iBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;iBACxE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACpB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}

package/dist/planner.d.ts

@@ -0,0 +1,12 @@
+import type { IPlanner, IProvider, ILogger, IEventEmitter, PlanRequest, ExecutionPlan, AstriveError } from '@astrive-ai/types';
+export declare class LLMPlanner implements IPlanner {
+    private provider;
+    private logger;
+    private events;
+    constructor(provider: IProvider, logger: ILogger, events: IEventEmitter);
+    plan(request: PlanRequest): Promise<ExecutionPlan>;
+    replan(plan: ExecutionPlan, error: AstriveError): Promise<ExecutionPlan>;
+    private buildPlanningPrompt;
+    private parsePlanResponse;
+}
+//# sourceMappingURL=planner.d.ts.map

package/dist/planner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"planner.d.ts","sourceRoot":"","sources":["../src/planner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAY,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAGzI,qBAAa,UAAW,YAAW,QAAQ;IAEvC,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,MAAM;gBAFN,QAAQ,EAAE,SAAS,EACnB,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,aAAa;IAGzB,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,aAAa,CAAC;IAiClD,MAAM,CAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;IA2B9E,OAAO,CAAC,mBAAmB;IAsB3B,OAAO,CAAC,iBAAiB;CAiB1B"}

package/dist/planner.js

@@ -0,0 +1,100 @@
+import { randomUUID } from 'node:crypto';
+export class LLMPlanner {
+    provider;
+    logger;
+    events;
+    constructor(provider, logger, events) {
+        this.provider = provider;
+        this.logger = logger;
+        this.events = events;
+    }
+    async plan(request) {
+        this.events.emit('plan:start', { request });
+        this.logger.debug('Starting planning phase', { request });
+        const prompt = this.buildPlanningPrompt(request);
+        const response = await this.provider.chat({
+            messages: [{ role: 'user', content: prompt }]
+        });
+        const parsed = this.parsePlanResponse(response.content);
+        const steps = parsed.steps.map(step => ({
+            id: randomUUID(),
+            toolName: step.toolName,
+            arguments: step.arguments,
+            description: step.description,
+            dependsOn: step.dependsOn || [],
+            status: 'pending'
+        }));
+        const plan = {
+            id: randomUUID(),
+            steps,
+            reasoning: parsed.reasoning,
+            status: 'pending',
+            estimatedDuration: 0
+        };
+        this.logger.debug('Planning complete', { plan });
+        this.events.emit('plan:end', { plan });
+        return plan;
+    }
+    async replan(plan, error) {
+        this.logger.debug('Replanning due to error', { error: error.message });
+        const prompt = `The previous plan failed with error: ${error.message}.\n\nOriginal Plan:\n${JSON.stringify(plan.steps, null, 2)}\n\nPlease provide a revised JSON plan that avoids this error.`;
+        const response = await this.provider.chat({
+            messages: [{ role: 'user', content: prompt }]
+        });
+        const parsed = this.parsePlanResponse(response.content);
+        const steps = parsed.steps.map(step => ({
+            id: randomUUID(),
+            toolName: step.toolName,
+            arguments: step.arguments,
+            description: step.description,
+            dependsOn: step.dependsOn || [],
+            status: 'pending'
+        }));
+        return {
+            id: randomUUID(),
+            steps,
+            reasoning: parsed.reasoning,
+            status: 'pending',
+            estimatedDuration: 0
+        };
+    }
+    buildPlanningPrompt(request) {
+        return `You are an AI planner. Your task is to analyze the user request and return a structured JSON plan.
+Available Tools:
+${JSON.stringify(request.tools, null, 2)}
+
+User Request: ${request.prompt}
+
+Return ONLY a JSON object with the following structure:
+{
+  "steps": [
+    {
+      "toolName": "name_of_tool",
+      "arguments": { "arg1": "value" },
+      "description": "What this step does",
+      "dependsOn": [] // Optional array of step IDs this step depends on
+    }
+  ],
+  "reasoning": "Explanation of the plan"
+}
+`;
+    }
+    parsePlanResponse(response) {
+        let jsonStr = response;
+        const match = response.match(/```(?:json)?\s*([\s\S]*?)```/);
+        if (match) {
+            jsonStr = match[1];
+        }
+        try {
+            const data = JSON.parse(jsonStr.trim());
+            return {
+                steps: Array.isArray(data.steps) ? data.steps : [],
+                reasoning: data.reasoning || ''
+            };
+        }
+        catch (err) {
+            throw new Error(`Failed to parse plan response: ${err}`);
+        }
+    }
+}
+//# sourceMappingURL=planner.js.map

package/dist/planner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"planner.js","sourceRoot":"","sources":["../src/planner.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,OAAO,UAAU;IAEX;IACA;IACA;IAHV,YACU,QAAmB,EACnB,MAAe,EACf,MAAqB;QAFrB,aAAQ,GAAR,QAAQ,CAAW;QACnB,WAAM,GAAN,MAAM,CAAS;QACf,WAAM,GAAN,MAAM,CAAe;IAC5B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAE1D,MAAM,MAAM,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAEjD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACxC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;SAC9C,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAExD,MAAM,KAAK,GAAe,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClD,EAAE,EAAE,UAAU,EAAE;YAChB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE;YAC/B,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,GAAkB;YAC1B,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM,EAAE,SAAS;YACjB,iBAAiB,EAAE,CAAC;SACrB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,IAAmB,EAAE,KAAmB;QACnD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,wCAAwC,KAAK,CAAC,OAAO,wBAAwB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,gEAAgE,CAAC;QAEhM,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACxC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;SAC9C,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAExD,MAAM,KAAK,GAAe,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAClD,EAAE,EAAE,UAAU,EAAE;YAChB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE;YAC/B,MAAM,EAAE,SAAS;SAClB,CAAC,CAAC,CAAC;QAEJ,OAAO;YACL,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM,EAAE,SAAS;YACjB,iBAAiB,EAAE,CAAC;SACrB,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,OAAoB;QAC9C,OAAO;;EAET,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;;gBAExB,OAAO,CAAC,MAAM;;;;;;;;;;;;;;CAc7B,CAAC;IACA,CAAC;IAEO,iBAAiB,CAAC,QAAgB;QACxC,IAAI,OAAO,GAAG,QAAQ,CAAC;QACvB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAC7D,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YACxC,OAAO;gBACL,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;gBAClD,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,EAAE;aAChC,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,kCAAkC,GAAG,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;CACF"}

package/dist/simple-planner.d.ts

@@ -0,0 +1,7 @@
+import type { IPlanner, PlanRequest, ExecutionPlan } from '@astrive-ai/types';
+export declare class SimplePlanner implements IPlanner {
+    plan(request: PlanRequest): Promise<ExecutionPlan>;
+    replan(plan: ExecutionPlan, error: any): Promise<ExecutionPlan>;
+    private extractUrl;
+}
+//# sourceMappingURL=simple-planner.d.ts.map

package/dist/simple-planner.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple-planner.d.ts","sourceRoot":"","sources":["../src/simple-planner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAY,MAAM,mBAAmB,CAAC;AAGxF,qBAAa,aAAc,YAAW,QAAQ;IACtC,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,aAAa,CAAC;IA4DlD,MAAM,CAAC,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,aAAa,CAAC;IAIrE,OAAO,CAAC,UAAU;CAKnB"}

package/dist/simple-planner.js

@@ -0,0 +1,78 @@
+import { randomUUID } from 'node:crypto';
+export class SimplePlanner {
+    async plan(request) {
+        const input = request.prompt.toLowerCase();
+        const steps = [];
+        let reasoning = 'Direct response';
+        const addStep = (toolName, description, args = {}) => {
+            steps.push({
+                id: randomUUID(),
+                toolName,
+                arguments: args,
+                description,
+                dependsOn: [],
+                status: 'pending'
+            });
+        };
+        if (input.includes('download')) {
+            addStep('downloader', 'Download media', { url: this.extractUrl(input) });
+            reasoning = 'User wants to download media';
+        }
+        else if (input.includes('image') || input.includes('generate') || input.includes('create image')) {
+            addStep('texttoimage', 'Generate image', { prompt: request.prompt });
+            reasoning = 'User wants to generate an image';
+        }
+        else if (input.includes('remove background') || input.includes('removebg')) {
+            addStep('removebg', 'Remove background', { image: this.extractUrl(input) });
+            reasoning = 'User wants to remove background';
+        }
+        else if (input.includes('upscale')) {
+            addStep('upscaler', 'Upscale image', { image: this.extractUrl(input) });
+            reasoning = 'User wants to upscale an image';
+        }
+        else if (input.includes('face swap')) {
+            addStep('faceswap', 'Swap faces', { faceImage: '', targetImage: '' });
+            reasoning = 'User wants to swap faces';
+        }
+        else if (input.includes('search') || input.includes('find') || input.includes('look up')) {
+            addStep('websearch', 'Search web', { query: request.prompt });
+            reasoning = 'User wants to search the web';
+        }
+        else if (input.includes('speech') || input.includes('say') || input.includes('speak') || input.includes('tts')) {
+            addStep('texttospeech', 'Text to speech', { text: request.prompt });
+            reasoning = 'User wants text to speech';
+        }
+        else if (input.includes('transcribe') || input.includes('listen') || input.includes('stt')) {
+            addStep('speechtotext', 'Transcribe audio', { audio: '' });
+            reasoning = 'User wants audio transcription';
+        }
+        else if (input.includes('vision') || input.includes('describe image') || input.includes('analyze image')) {
+            addStep('vision', 'Analyze image', { image: this.extractUrl(input) });
+            reasoning = 'User wants to analyze an image';
+        }
+        else if (input.includes('ocr') || input.includes('read text') || input.includes('extract text')) {
+            addStep('ocr', 'Extract text', { image: this.extractUrl(input) });
+            reasoning = 'User wants to extract text from image';
+        }
+        else if (input.includes('edit image')) {
+            addStep('img2img', 'Edit image', { image: this.extractUrl(input), prompt: request.prompt });
+            reasoning = 'User wants to edit an image';
+        }
+        return {
+            id: randomUUID(),
+            steps,
+            reasoning,
+            status: 'pending',
+            estimatedDuration: 0
+        };
+    }
+    async replan(plan, error) {
+        return plan;
+    }
+    extractUrl(text) {
+        const urlRegex = /(https?:\/\/[^\s]+)/g;
+        const match = text.match(urlRegex);
+        return match ? match[0] : '';
+    }
+}
+//# sourceMappingURL=simple-planner.js.map

package/dist/simple-planner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"simple-planner.js","sourceRoot":"","sources":["../src/simple-planner.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,OAAO,aAAa;IACxB,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAe,EAAE,CAAC;QAC7B,IAAI,SAAS,GAAG,iBAAiB,CAAC;QAElC,MAAM,OAAO,GAAG,CAAC,QAAgB,EAAE,WAAmB,EAAE,OAAY,EAAE,EAAE,EAAE;YACxE,KAAK,CAAC,IAAI,CAAC;gBACT,EAAE,EAAE,UAAU,EAAE;gBAChB,QAAQ;gBACR,SAAS,EAAE,IAAI;gBACf,WAAW;gBACX,SAAS,EAAE,EAAE;gBACb,MAAM,EAAE,SAAS;aAClB,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACzE,SAAS,GAAG,8BAA8B,CAAC;QAC7C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YACnG,OAAO,CAAC,aAAa,EAAE,gBAAgB,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YACrE,SAAS,GAAG,iCAAiC,CAAC;QAChD,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7E,OAAO,CAAC,UAAU,EAAE,mBAAmB,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC5E,SAAS,GAAG,iCAAiC,CAAC;QAChD,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YACrC,OAAO,CAAC,UAAU,EAAE,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACxE,SAAS,GAAG,gCAAgC,CAAC;QAC/C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACvC,OAAO,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;YACtE,SAAS,GAAG,0BAA0B,CAAC;QACzC,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3F,OAAO,CAAC,WAAW,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC9D,SAAS,GAAG,8BAA8B,CAAC;QAC7C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACjH,OAAO,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YACpE,SAAS,GAAG,2BAA2B,CAAC;QAC1C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7F,OAAO,CAAC,cAAc,EAAE,kBAAkB,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAC3D,SAAS,GAAG,gCAAgC,CAAC;QAC/C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAC3G,OAAO,CAAC,QAAQ,EAAE,eAAe,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACtE,SAAS,GAAG,gCAAgC,CAAC;QAC/C,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;YAClG,OAAO,CAAC,KAAK,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAClE,SAAS,GAAG,uCAAuC,CAAC;QACtD,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YACxC,OAAO,CAAC,SAAS,EAAE,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAC5F,SAAS,GAAG,6BAA6B,CAAC;QAC5C,CAAC;QAED,OAAO;YACL,EAAE,EAAE,UAAU,EAAE;YAChB,KAAK;YACL,SAAS;YACT,MAAM,EAAE,SAAS;YACjB,iBAAiB,EAAE,CAAC;SACrB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,IAAmB,EAAE,KAAU;QAC1C,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,UAAU,CAAC,IAAY;QAC7B,MAAM,QAAQ,GAAG,sBAAsB,CAAC;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnC,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/B,CAAC;CACF"}

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "@astrive-ai/planner",
+  "version": "1.0.1",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "@astrive-ai/types": "*",
+    "@astrive-ai/logger": "*",
+    "@astrive-ai/events": "*"
+  },
+  "devDependencies": {
+    "typescript": "5.4.5"
+  }
+}

package/src/index.ts

@@ -0,0 +1,3 @@
+export * from './planner.js';
+export * from './simple-planner.js';
+export * from './plan-optimizer.js';

package/src/plan-optimizer.ts

@@ -0,0 +1,42 @@
+import type { ExecutionPlan, PlanStep } from '@astrive-ai/types';
+
+export class PlanOptimizer {
+  optimize(plan: ExecutionPlan): ExecutionPlan {
+    // Basic optimization: reorder based on dependencies, mark parallel
+    const steps = [...plan.steps];
+    
+    // Remove duplicate steps (same tool, same args)
+    const uniqueSteps = steps.filter((step, index, self) =>
+      index === self.findIndex((t) => (
+        t.toolName === step.toolName && 
+        JSON.stringify(t.arguments) === JSON.stringify(step.arguments)
+      ))
+    );
+
+    return {
+      ...plan,
+      steps: uniqueSteps
+    };
+  }
+
+  detectParallelizable(steps: PlanStep[]): string[][] {
+    const levels: string[][] = [];
+    const processed = new Set<string>();
+    
+    let currentLevel = steps
+      .filter(s => !s.dependsOn || s.dependsOn.length === 0)
+      .map(s => s.id);
+      
+    while (currentLevel.length > 0) {
+      levels.push(currentLevel);
+      currentLevel.forEach(id => processed.add(id));
+      
+      currentLevel = steps
+        .filter(s => !processed.has(s.id))
+        .filter(s => s.dependsOn && s.dependsOn.every(dep => processed.has(dep)))
+        .map(s => s.id);
+    }
+    
+    return levels;
+  }
+}

package/src/planner.ts

@@ -0,0 +1,110 @@
+import type { IPlanner, IProvider, ILogger, IEventEmitter, PlanRequest, ExecutionPlan, PlanStep, AstriveError } from '@astrive-ai/types';
+import { randomUUID } from 'node:crypto';
+
+export class LLMPlanner implements IPlanner {
+  constructor(
+    private provider: IProvider,
+    private logger: ILogger,
+    private events: IEventEmitter
+  ) {}
+
+  async plan(request: PlanRequest): Promise<ExecutionPlan> {
+    this.events.emit('plan:start', { request });
+    this.logger.debug('Starting planning phase', { request });
+
+    const prompt = this.buildPlanningPrompt(request);
+    
+    const response = await this.provider.chat({
+      messages: [{ role: 'user', content: prompt }]
+    });
+    const parsed = this.parsePlanResponse(response.content);
+    
+    const steps: PlanStep[] = parsed.steps.map(step => ({
+      id: randomUUID(),
+      toolName: step.toolName,
+      arguments: step.arguments,
+      description: step.description,
+      dependsOn: step.dependsOn || [],
+      status: 'pending'
+    }));
+
+    const plan: ExecutionPlan = {
+      id: randomUUID(),
+      steps,
+      reasoning: parsed.reasoning,
+      status: 'pending',
+      estimatedDuration: 0
+    };
+
+    this.logger.debug('Planning complete', { plan });
+    this.events.emit('plan:end', { plan });
+    return plan;
+  }
+
+  async replan(plan: ExecutionPlan, error: AstriveError): Promise<ExecutionPlan> {
+    this.logger.debug('Replanning due to error', { error: error.message });
+    const prompt = `The previous plan failed with error: ${error.message}.\n\nOriginal Plan:\n${JSON.stringify(plan.steps, null, 2)}\n\nPlease provide a revised JSON plan that avoids this error.`;
+    
+    const response = await this.provider.chat({
+      messages: [{ role: 'user', content: prompt }]
+    });
+    const parsed = this.parsePlanResponse(response.content);
+    
+    const steps: PlanStep[] = parsed.steps.map(step => ({
+      id: randomUUID(),
+      toolName: step.toolName,
+      arguments: step.arguments,
+      description: step.description,
+      dependsOn: step.dependsOn || [],
+      status: 'pending'
+    }));
+
+    return {
+      id: randomUUID(),
+      steps,
+      reasoning: parsed.reasoning,
+      status: 'pending',
+      estimatedDuration: 0
+    };
+  }
+
+  private buildPlanningPrompt(request: PlanRequest): string {
+    return `You are an AI planner. Your task is to analyze the user request and return a structured JSON plan.
+Available Tools:
+${JSON.stringify(request.tools, null, 2)}
+
+User Request: ${request.prompt}
+
+Return ONLY a JSON object with the following structure:
+{
+  "steps": [
+    {
+      "toolName": "name_of_tool",
+      "arguments": { "arg1": "value" },
+      "description": "What this step does",
+      "dependsOn": [] // Optional array of step IDs this step depends on
+    }
+  ],
+  "reasoning": "Explanation of the plan"
+}
+`;
+  }
+
+  private parsePlanResponse(response: string): { steps: any[], reasoning: string } {
+    let jsonStr = response;
+    const match = response.match(/```(?:json)?\s*([\s\S]*?)```/);
+    if (match) {
+      jsonStr = match[1];
+    }
+    
+    try {
+      const data = JSON.parse(jsonStr.trim());
+      return {
+        steps: Array.isArray(data.steps) ? data.steps : [],
+        reasoning: data.reasoning || ''
+      };
+    } catch (err) {
+      throw new Error(`Failed to parse plan response: ${err}`);
+    }
+  }
+}

package/src/simple-planner.ts

@@ -0,0 +1,74 @@
+import type { IPlanner, PlanRequest, ExecutionPlan, PlanStep } from '@astrive-ai/types';
+import { randomUUID } from 'node:crypto';
+
+export class SimplePlanner implements IPlanner {
+  async plan(request: PlanRequest): Promise<ExecutionPlan> {
+    const input = request.prompt.toLowerCase();
+    const steps: PlanStep[] = [];
+    let reasoning = 'Direct response';
+    
+    const addStep = (toolName: string, description: string, args: any = {}) => {
+      steps.push({
+        id: randomUUID(),
+        toolName,
+        arguments: args,
+        description,
+        dependsOn: [],
+        status: 'pending'
+      });
+    };
+
+    if (input.includes('download')) {
+      addStep('downloader', 'Download media', { url: this.extractUrl(input) });
+      reasoning = 'User wants to download media';
+    } else if (input.includes('image') || input.includes('generate') || input.includes('create image')) {
+      addStep('texttoimage', 'Generate image', { prompt: request.prompt });
+      reasoning = 'User wants to generate an image';
+    } else if (input.includes('remove background') || input.includes('removebg')) {
+      addStep('removebg', 'Remove background', { image: this.extractUrl(input) });
+      reasoning = 'User wants to remove background';
+    } else if (input.includes('upscale')) {
+      addStep('upscaler', 'Upscale image', { image: this.extractUrl(input) });
+      reasoning = 'User wants to upscale an image';
+    } else if (input.includes('face swap')) {
+      addStep('faceswap', 'Swap faces', { faceImage: '', targetImage: '' });
+      reasoning = 'User wants to swap faces';
+    } else if (input.includes('search') || input.includes('find') || input.includes('look up')) {
+      addStep('websearch', 'Search web', { query: request.prompt });
+      reasoning = 'User wants to search the web';
+    } else if (input.includes('speech') || input.includes('say') || input.includes('speak') || input.includes('tts')) {
+      addStep('texttospeech', 'Text to speech', { text: request.prompt });
+      reasoning = 'User wants text to speech';
+    } else if (input.includes('transcribe') || input.includes('listen') || input.includes('stt')) {
+      addStep('speechtotext', 'Transcribe audio', { audio: '' });
+      reasoning = 'User wants audio transcription';
+    } else if (input.includes('vision') || input.includes('describe image') || input.includes('analyze image')) {
+      addStep('vision', 'Analyze image', { image: this.extractUrl(input) });
+      reasoning = 'User wants to analyze an image';
+    } else if (input.includes('ocr') || input.includes('read text') || input.includes('extract text')) {
+      addStep('ocr', 'Extract text', { image: this.extractUrl(input) });
+      reasoning = 'User wants to extract text from image';
+    } else if (input.includes('edit image')) {
+      addStep('img2img', 'Edit image', { image: this.extractUrl(input), prompt: request.prompt });
+      reasoning = 'User wants to edit an image';
+    }
+
+    return {
+      id: randomUUID(),
+      steps,
+      reasoning,
+      status: 'pending',
+      estimatedDuration: 0
+    };
+  }
+  
+  async replan(plan: ExecutionPlan, error: any): Promise<ExecutionPlan> {
+    return plan;
+  }
+  
+  private extractUrl(text: string): string {
+    const urlRegex = /(https?:\/\/[^\s]+)/g;
+    const match = text.match(urlRegex);
+    return match ? match[0] : '';
+  }
+}

package/tsconfig.json

@@ -0,0 +1,8 @@
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "rootDir": "./src"
+  },
+  "include": ["src/**/*"]
+}