substrai-lambdallm 1.0.0 → 1.2.0

package/README.md

@@ -1,41 +1,83 @@
 # substrai-lambdallm
 
-**Serverless-native LLM orchestration framework for AWS Lambda.**
+**Serverless-native LLM orchestration framework for AWS Lambda (TypeScript)**
 
-[![PyPI](https://badge.fury.io/py/substrai-lambdallm.svg)](https://pypi.org/project/substrai-lambdallm/)
+[![npm](https://badge.fury.io/js/substrai-lambdallm.svg)](https://www.npmjs.com/package/substrai-lambdallm)
 [![GitHub](https://img.shields.io/github/stars/substrai/lambdallm)](https://github.com/substrai/lambdallm)
-[![Docs](https://img.shields.io/badge/docs-substrai.github.io-blue)](https://substrai.github.io/lambdallm)
 
-> **Note:** The primary implementation is in Python. TypeScript SDK coming soon.
-
-## Install (Python)
+## Install
 
 ```bash
-pip install substrai-lambdallm[bedrock]
+npm install substrai-lambdallm @aws-sdk/client-bedrock-runtime
 ```
 
 ## Quick Start
 
-```python
-from lambdallm import handler, Prompt, Model
+```typescript
+import { handler, Model } from 'substrai-lambdallm';
 
-@handler(model=Model.CLAUDE_3_HAIKU)
-def lambda_handler(event, context):
-    result = context.invoke("Summarize: {text}", text=event["body"]["text"])
-    return {"statusCode": 200, "body": result}
+export const lambdaHandler = handler(
+  { model: Model.CLAUDE_3_HAIKU, maxRetries: 3 },
+  async (event, context) => {
+    const result = await context.invoke('Summarize: {text}', { text: event.body.text });
+    return { statusCode: 200, body: { result, cost: context.totalCost } };
+  }
+);
 ```
 
 ## Features
 
-- < 5MB package (vs 400MB+ for LangChain)
-- Cold-start optimized (lazy imports, connection pooling)
-- DynamoDB-native state (conversation memory persists)
-- Multi-step chains with checkpoint/resume
-- AI Agents with tool sandboxing
-- Cost-aware model routing
-- One-command deploy
-- A/B testing for prompts
-- Full observability (X-Ray + CloudWatch)
+- **< 50KB** package size
+- **TypeScript-first** with full type safety
+- **Bedrock provider** (Claude, Titan, Llama)
+- **Retry with backoff** and fallback models
+- **Prompt templates** with variable validation
+- **Multi-step chains** with checkpoint/resume
+- **AI Agents** with ReAct loop and tools
+- **Session state** with sliding window memory
+- **Cost tracking** per request
+
+## Chains
+
+```typescript
+import { Chain, Step } from 'substrai-lambdallm';
+
+const pipeline = new Chain({
+  name: 'analysis',
+  steps: [
+    new Step({ name: 'extract', prompt: 'Extract entities: {input}' }),
+    new Step({ name: 'classify', prompt: 'Classify: {extract.output}' }),
+  ],
+  timeoutStrategy: 'checkpoint',
+});
+
+const result = await pipeline.run(context, { input: 'document text' });
+```
+
+## Agents
+
+```typescript
+import { Agent, Tool } from 'substrai-lambdallm';
+
+const search = Tool({ description: 'Search docs' })(
+  (input: any) => `Results for ${input.query}`
+);
+
+const agent = new Agent({
+  name: 'researcher',
+  systemPrompt: 'You are a research assistant.',
+  tools: [search],
+  maxIterations: 5,
+});
+
+const result = await agent.run('What is LambdaLLM?', context);
+```
+
+## Also Available in Python
+
+```bash
+pip install substrai-lambdallm[bedrock]
+```
 
 ## Links
 
@@ -46,5 +88,3 @@ def lambda_handler(event, context):
 ## Author
 
 **Gaurav Kumar Sinha** — Founder, [SubstrAI](https://github.com/substrai)
-
-Email: gaurav@substrai.dev

package/dist/agents/agent.d.ts

@@ -0,0 +1,57 @@
+/**
+ * Agent and Tool system for LambdaLLM.
+ */
+import { LambdaLLMContext } from '../core/context';
+export interface ToolDefinition {
+    name: string;
+    description: string;
+    parameters: Array<{
+        name: string;
+        type: string;
+        description: string;
+        required: boolean;
+    }>;
+    func: (...args: any[]) => any;
+}
+/**
+ * Decorator-like function to define a tool.
+ */
+export declare function Tool(options: {
+    description: string;
+    name?: string;
+}): (func: Function) => ToolDefinition;
+export interface AgentResult {
+    answer: string;
+    status: 'completed' | 'max_iterations' | 'timeout' | 'cost_limit';
+    totalIterations: number;
+    totalToolCalls: number;
+    totalCostUsd: number;
+    totalLatencyMs: number;
+    steps: Array<{
+        thought: string;
+        toolName?: string;
+        toolOutput?: string;
+        isFinal: boolean;
+    }>;
+}
+export interface AgentOptions {
+    name: string;
+    systemPrompt: string;
+    tools: ToolDefinition[];
+    maxIterations?: number;
+    timeoutBuffer?: number;
+    maxCostUsd?: number;
+}
+export declare class Agent {
+    name: string;
+    systemPrompt: string;
+    tools: ToolDefinition[];
+    maxIterations: number;
+    timeoutBuffer: number;
+    maxCostUsd?: number;
+    constructor(options: AgentOptions);
+    run(query: string, context: LambdaLLMContext): Promise<AgentResult>;
+    private parseDecision;
+    private buildResult;
+}
+//# sourceMappingURL=agent.d.ts.map

package/dist/agents/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAC1F,IAAI,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC;CAC/B;AAED;;GAEG;AACH,wBAAgB,IAAI,CAAC,OAAO,EAAE;IAAE,WAAW,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,IACjD,MAAM,QAAQ,KAAG,cAAc,CAQjD;AAED,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,WAAW,GAAG,gBAAgB,GAAG,SAAS,GAAG,YAAY,CAAC;IAClE,eAAe,EAAE,MAAM,CAAC;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,KAAK,CAAC;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CAC7F;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,qBAAa,KAAK;IACT,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;gBAEf,OAAO,EAAE,YAAY;IAS3B,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IA+CzE,OAAO,CAAC,aAAa;IAarB,OAAO,CAAC,WAAW;CAYpB"}

package/dist/agents/agent.js

@@ -0,0 +1,93 @@
+"use strict";
+/**
+ * Agent and Tool system for LambdaLLM.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Agent = void 0;
+exports.Tool = Tool;
+/**
+ * Decorator-like function to define a tool.
+ */
+function Tool(options) {
+    return function (func) {
+        return {
+            name: options.name || func.name,
+            description: options.description,
+            parameters: [],
+            func: func,
+        };
+    };
+}
+class Agent {
+    constructor(options) {
+        this.name = options.name;
+        this.systemPrompt = options.systemPrompt;
+        this.tools = options.tools;
+        this.maxIterations = options.maxIterations || 10;
+        this.timeoutBuffer = options.timeoutBuffer || 30;
+        this.maxCostUsd = options.maxCostUsd;
+    }
+    async run(query, context) {
+        const startTime = Date.now();
+        const steps = [];
+        let toolCalls = 0;
+        const toolDescriptions = this.tools.map(t => `- ${t.name}: ${t.description}`).join('\n');
+        const systemContent = `${this.systemPrompt}\n\nTools available:\n${toolDescriptions}\n\nRespond with JSON: {"thought": "...", "tool_name": "...", "tool_input": {...}} or {"thought": "...", "final_answer": "..."}`;
+        let messages = `System: ${systemContent}\n\nUser: ${query}\n\nAssistant:`;
+        for (let i = 0; i < this.maxIterations; i++) {
+            if (context.remainingTimeMs < this.timeoutBuffer * 1000) {
+                return this.buildResult(steps, 'timeout', startTime, toolCalls, context.totalCost, 'Approaching timeout.');
+            }
+            if (this.maxCostUsd && context.totalCost >= this.maxCostUsd) {
+                return this.buildResult(steps, 'cost_limit', startTime, toolCalls, context.totalCost, 'Cost limit reached.');
+            }
+            const response = await context.invoke(messages);
+            const decision = this.parseDecision(response);
+            if (decision.final_answer) {
+                steps.push({ thought: decision.thought || '', isFinal: true });
+                return this.buildResult(steps, 'completed', startTime, toolCalls, context.totalCost, decision.final_answer);
+            }
+            if (decision.tool_name) {
+                const tool = this.tools.find(t => t.name === decision.tool_name);
+                let toolOutput = 'Tool not found';
+                if (tool) {
+                    try {
+                        toolOutput = String(tool.func(decision.tool_input || {}));
+                        toolCalls++;
+                    }
+                    catch (e) {
+                        toolOutput = `Error: ${e.message}`;
+                    }
+                }
+                steps.push({ thought: decision.thought || '', toolName: decision.tool_name, toolOutput, isFinal: false });
+                messages += `\n${response}\nUser: Tool '${decision.tool_name}' returned: ${toolOutput}\nContinue.\nAssistant:`;
+            }
+        }
+        return this.buildResult(steps, 'max_iterations', startTime, toolCalls, context.totalCost, 'Max iterations reached.');
+    }
+    parseDecision(response) {
+        try {
+            if (response.trim().startsWith('{'))
+                return JSON.parse(response.trim());
+            if (response.includes('```json')) {
+                const json = response.split('```json')[1].split('```')[0];
+                return JSON.parse(json.trim());
+            }
+            return { thought: 'Direct response', final_answer: response };
+        }
+        catch {
+            return { thought: 'Could not parse', final_answer: response };
+        }
+    }
+    buildResult(steps, status, startTime, toolCalls, cost, answer) {
+        return {
+            answer, status, steps,
+            totalIterations: steps.length,
+            totalToolCalls: toolCalls,
+            totalCostUsd: cost,
+            totalLatencyMs: Date.now() - startTime,
+        };
+    }
+}
+exports.Agent = Agent;
+//# sourceMappingURL=agent.js.map

package/dist/agents/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/agents/agent.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAeH,oBASC;AAZD;;GAEG;AACH,SAAgB,IAAI,CAAC,OAA+C;IAClE,OAAO,UAAU,IAAc;QAC7B,OAAO;YACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI;YAC/B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,UAAU,EAAE,EAAE;YACd,IAAI,EAAE,IAAW;SAClB,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAqBD,MAAa,KAAK;IAQhB,YAAY,OAAqB;QAC/B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC;QACjD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAa,EAAE,OAAyB;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAyB,EAAE,CAAC;QACvC,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzF,MAAM,aAAa,GAAG,GAAG,IAAI,CAAC,YAAY,yBAAyB,gBAAgB,iIAAiI,CAAC;QAErN,IAAI,QAAQ,GAAG,WAAW,aAAa,aAAa,KAAK,gBAAgB,CAAC;QAE1E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,IAAI,OAAO,CAAC,eAAe,GAAG,IAAI,CAAC,aAAa,GAAG,IAAI,EAAE,CAAC;gBACxD,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,sBAAsB,CAAC,CAAC;YAC7G,CAAC;YAED,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC5D,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,qBAAqB,CAAC,CAAC;YAC/G,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;YAE9C,IAAI,QAAQ,CAAC,YAAY,EAAE,CAAC;gBAC1B,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC/D,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC;YAC9G,CAAC;YAED,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACjE,IAAI,UAAU,GAAG,gBAAgB,CAAC;gBAClC,IAAI,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC;wBACH,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,CAAC;wBAC1D,SAAS,EAAE,CAAC;oBACd,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,UAAU,GAAG,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC;oBACrC,CAAC;gBACH,CAAC;gBAED,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC1G,QAAQ,IAAI,KAAK,QAAQ,iBAAiB,QAAQ,CAAC,SAAS,eAAe,UAAU,yBAAyB,CAAC;YACjH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,yBAAyB,CAAC,CAAC;IACvH,CAAC;IAEO,aAAa,CAAC,QAAgB;QACpC,IAAI,CAAC;YACH,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;YACxE,IAAI,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBACjC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1D,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YACjC,CAAC;YACD,OAAO,EAAE,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;QAChE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC;QAChE,CAAC;IACH,CAAC;IAEO,WAAW,CACjB,KAA2B,EAAE,MAA6B,EAC1D,SAAiB,EAAE,SAAiB,EAAE,IAAY,EAAE,MAAc;QAElE,OAAO;YACL,MAAM,EAAE,MAAM,EAAE,KAAK;YACrB,eAAe,EAAE,KAAK,CAAC,MAAM;YAC7B,cAAc,EAAE,SAAS;YACzB,YAAY,EAAE,IAAI;YAClB,cAAc,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACvC,CAAC;IACJ,CAAC;CACF;AAzFD,sBAyFC"}

package/dist/agents/extras.d.ts

@@ -0,0 +1,59 @@
+/**
+ * Multi-agent router, tool sandboxing, and async tool dispatch.
+ */
+import { Agent, AgentResult } from './agent';
+import { LambdaLLMContext } from '../core/context';
+export interface RouteConfig {
+    agent: Agent;
+    description: string;
+    keywords: string[];
+    priority?: number;
+}
+export declare class AgentRouter {
+    private routes;
+    private fallback?;
+    private strategy;
+    constructor(options: {
+        routes: RouteConfig[];
+        fallback?: Agent;
+        strategy?: 'keyword' | 'llm';
+    });
+    route(query: string, context: LambdaLLMContext): Promise<AgentResult>;
+    private routeByKeyword;
+    private routeByLLM;
+}
+export interface SandboxPolicy {
+    allowedActions: string[];
+    deniedActions: string[];
+    maxExecutionTimeMs: number;
+}
+export declare class ToolSandbox {
+    private policy;
+    constructor(policy?: Partial<SandboxPolicy>);
+    execute<T>(fn: () => T | Promise<T>): Promise<T>;
+    validateAction(action: string): boolean;
+}
+export interface AsyncToolRequest {
+    requestId: string;
+    toolName: string;
+    toolInput: Record<string, any>;
+}
+export declare class AsyncToolDispatcher {
+    private queueUrl?;
+    private client;
+    constructor(options?: {
+        queueUrl?: string;
+    });
+    dispatch(toolName: string, toolInput: Record<string, any>): Promise<AsyncToolRequest>;
+    private sendToSQS;
+    private generateId;
+}
+export declare class StreamingResponse {
+    private chunks;
+    private generator;
+    constructor(generator: AsyncGenerator<string>);
+    [Symbol.asyncIterator](): AsyncGenerator<string>;
+    getFullText(): Promise<string>;
+    toLambdaStream(): AsyncGenerator<Buffer>;
+}
+//# sourceMappingURL=extras.d.ts.map

package/dist/agents/extras.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"extras.d.ts","sourceRoot":"","sources":["../../src/agents/extras.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,KAAK,EAAE,WAAW,EAAkB,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAInD,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,QAAQ,CAAC,CAAQ;IACzB,OAAO,CAAC,QAAQ,CAAoB;gBAExB,OAAO,EAAE;QAAE,MAAM,EAAE,WAAW,EAAE,CAAC;QAAC,QAAQ,CAAC,EAAE,KAAK,CAAC;QAAC,QAAQ,CAAC,EAAE,SAAS,GAAG,KAAK,CAAA;KAAE;IAMxF,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAe3E,OAAO,CAAC,cAAc;YAUR,UAAU;CAUzB;AAID,MAAM,WAAW,aAAa;IAC5B,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,kBAAkB,EAAE,MAAM,CAAC;CAC5B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAgB;gBAElB,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC;IAQrC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAWtD,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO;CAUxC;AAID,MAAM,WAAW,gBAAgB;IAAG,SAAS,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAAE;AAE1G,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,QAAQ,CAAC,CAAS;IAC1B,OAAO,CAAC,MAAM,CAAa;gBAEf,OAAO,GAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO;IAIzC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;YAY7E,SAAS;IAavB,OAAO,CAAC,UAAU;CAMnB;AAID,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,SAAS,CAAyB;gBAE9B,SAAS,EAAE,cAAc,CAAC,MAAM,CAAC;IAItC,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,cAAc,CAAC,MAAM,CAAC;IAOjD,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC;IAKpC,cAAc,IAAI,cAAc,CAAC,MAAM,CAAC;CAWzC"}

package/dist/agents/extras.js

@@ -0,0 +1,152 @@
+"use strict";
+/**
+ * Multi-agent router, tool sandboxing, and async tool dispatch.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamingResponse = exports.AsyncToolDispatcher = exports.ToolSandbox = exports.AgentRouter = void 0;
+class AgentRouter {
+    constructor(options) {
+        this.routes = options.routes.sort((a, b) => (b.priority || 0) - (a.priority || 0));
+        this.fallback = options.fallback;
+        this.strategy = options.strategy || 'keyword';
+    }
+    async route(query, context) {
+        let agent;
+        if (this.strategy === 'keyword') {
+            agent = this.routeByKeyword(query);
+        }
+        else {
+            agent = await this.routeByLLM(query, context);
+        }
+        if (!agent)
+            agent = this.fallback;
+        if (!agent)
+            throw new Error('No agent matched and no fallback configured');
+        return agent.run(query, context);
+    }
+    routeByKeyword(query) {
+        const lower = query.toLowerCase();
+        for (const route of this.routes) {
+            for (const keyword of route.keywords) {
+                if (lower.includes(keyword.toLowerCase()))
+                    return route.agent;
+            }
+        }
+        return undefined;
+    }
+    async routeByLLM(query, context) {
+        const descriptions = this.routes.map(r => `- ${r.agent.name}: ${r.description}`).join('\n');
+        const prompt = `Classify this query into one of these categories:\n${descriptions}\n\nQuery: ${query}\n\nRespond with ONLY the agent name.`;
+        try {
+            const response = await context.invoke(prompt);
+            const name = response.trim().toLowerCase();
+            const match = this.routes.find(r => r.agent.name.toLowerCase() === name);
+            return match?.agent;
+        }
+        catch {
+            return undefined;
+        }
+    }
+}
+exports.AgentRouter = AgentRouter;
+class ToolSandbox {
+    constructor(policy) {
+        this.policy = {
+            allowedActions: policy?.allowedActions || [],
+            deniedActions: policy?.deniedActions || [],
+            maxExecutionTimeMs: policy?.maxExecutionTimeMs || 30000,
+        };
+    }
+    async execute(fn) {
+        const start = Date.now();
+        const result = await Promise.race([
+            Promise.resolve(fn()),
+            new Promise((_, reject) => setTimeout(() => reject(new Error(`Tool execution exceeded timeout (${this.policy.maxExecutionTimeMs}ms)`)), this.policy.maxExecutionTimeMs)),
+        ]);
+        return result;
+    }
+    validateAction(action) {
+        if (this.policy.deniedActions.includes(action))
+            return false;
+        if (this.policy.allowedActions.length === 0)
+            return true;
+        for (const allowed of this.policy.allowedActions) {
+            if (allowed === '*')
+                return true;
+            if (allowed.endsWith('*') && action.startsWith(allowed.slice(0, -1)))
+                return true;
+            if (action === allowed)
+                return true;
+        }
+        return false;
+    }
+}
+exports.ToolSandbox = ToolSandbox;
+class AsyncToolDispatcher {
+    constructor(options = {}) {
+        this.client = null;
+        this.queueUrl = options.queueUrl;
+    }
+    async dispatch(toolName, toolInput) {
+        const requestId = this.generateId();
+        const request = { requestId, toolName, toolInput };
+        if (this.queueUrl) {
+            await this.sendToSQS(request);
+        }
+        console.log(JSON.stringify({ event: 'async_tool.dispatched', ...request }));
+        return request;
+    }
+    async sendToSQS(request) {
+        try {
+            const { SQSClient, SendMessageCommand } = require('@aws-sdk/client-sqs');
+            if (!this.client)
+                this.client = new SQSClient({});
+            await this.client.send(new SendMessageCommand({
+                QueueUrl: this.queueUrl,
+                MessageBody: JSON.stringify(request),
+            }));
+        }
+        catch (e) {
+            throw new Error(`Failed to dispatch async tool: ${e.message}`);
+        }
+    }
+    generateId() {
+        return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, c => {
+            const r = Math.random() * 16 | 0;
+            return (c === 'x' ? r : (r & 0x3 | 0x8)).toString(16);
+        });
+    }
+}
+exports.AsyncToolDispatcher = AsyncToolDispatcher;
+// ===== STREAMING =====
+class StreamingResponse {
+    constructor(generator) {
+        this.chunks = [];
+        this.generator = generator;
+    }
+    async *[Symbol.asyncIterator]() {
+        for await (const chunk of this.generator) {
+            this.chunks.push(chunk);
+            yield chunk;
+        }
+    }
+    async getFullText() {
+        for await (const chunk of this.generator) {
+            this.chunks.push(chunk);
+        }
+        return this.chunks.join('');
+    }
+    toLambdaStream() {
+        const self = this;
+        return (async function* () {
+            yield Buffer.from(JSON.stringify({ type: 'metadata' }) + '\n');
+            for await (const chunk of self.generator) {
+                self.chunks.push(chunk);
+                yield Buffer.from(JSON.stringify({ type: 'content', text: chunk }) + '\n');
+            }
+            yield Buffer.from(JSON.stringify({ type: 'complete' }) + '\n');
+        })();
+    }
+}
+exports.StreamingResponse = StreamingResponse;
+//# sourceMappingURL=extras.js.map

package/dist/agents/extras.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"extras.js","sourceRoot":"","sources":["../../src/agents/extras.ts"],"names":[],"mappings":";AAAA;;GAEG;;;AAcH,MAAa,WAAW;IAKtB,YAAY,OAAkF;QAC5F,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,SAAS,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,OAAyB;QAClD,IAAI,KAAwB,CAAC;QAE7B,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAChC,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC;aAAM,CAAC;YACN,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC,KAAK;YAAE,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC;QAClC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QAE3E,OAAO,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAEO,cAAc,CAAC,KAAa;QAClC,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAClC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,KAAK,MAAM,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACrC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;oBAAE,OAAO,KAAK,CAAC,KAAK,CAAC;YAChE,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAa,EAAE,OAAyB;QAC/D,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5F,MAAM,MAAM,GAAG,sDAAsD,YAAY,cAAc,KAAK,uCAAuC,CAAC;QAC5I,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,IAAI,CAAC,CAAC;YACzE,OAAO,KAAK,EAAE,KAAK,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YAAC,OAAO,SAAS,CAAC;QAAC,CAAC;IAC/B,CAAC;CACF;AA9CD,kCA8CC;AAUD,MAAa,WAAW;IAGtB,YAAY,MAA+B;QACzC,IAAI,CAAC,MAAM,GAAG;YACZ,cAAc,EAAE,MAAM,EAAE,cAAc,IAAI,EAAE;YAC5C,aAAa,EAAE,MAAM,EAAE,aAAa,IAAI,EAAE;YAC1C,kBAAkB,EAAE,MAAM,EAAE,kBAAkB,IAAI,KAAK;SACxD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAI,EAAwB;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;YAChC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;YACrB,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAC/B,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,oCAAoC,IAAI,CAAC,MAAM,CAAC,kBAAkB,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAC7I;SACF,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,cAAc,CAAC,MAAc;QAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,KAAK,CAAC;QAC7D,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QACzD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YACjD,IAAI,OAAO,KAAK,GAAG;gBAAE,OAAO,IAAI,CAAC;YACjC,IAAI,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBAAE,OAAO,IAAI,CAAC;YAClF,IAAI,MAAM,KAAK,OAAO;gBAAE,OAAO,IAAI,CAAC;QACtC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAhCD,kCAgCC;AAMD,MAAa,mBAAmB;IAI9B,YAAY,UAAiC,EAAE;QAFvC,WAAM,GAAQ,IAAI,CAAC;QAGzB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,QAAgB,EAAE,SAA8B;QAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACpC,MAAM,OAAO,GAAqB,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;QAErE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC;QAC5E,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,OAAyB;QAC/C,IAAI,CAAC;YACH,MAAM,EAAE,SAAS,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAAC;YACzE,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,kBAAkB,CAAC;gBAC5C,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aACrC,CAAC,CAAC,CAAC;QACN,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,OAAO,sCAAsC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE;YACjE,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACjC,OAAO,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAvCD,kDAuCC;AAED,wBAAwB;AAExB,MAAa,iBAAiB;IAI5B,YAAY,SAAiC;QAHrC,WAAM,GAAa,EAAE,CAAC;QAI5B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC;QAC3B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACzC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW;QACf,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAAC,CAAC;QACtE,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC9B,CAAC;IAED,cAAc;QACZ,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,CAAC,KAAK,SAAS,CAAC;YACrB,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YAC/D,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACzC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACxB,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YAC7E,CAAC;YACD,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QACjE,CAAC,CAAC,EAAE,CAAC;IACP,CAAC;CACF;AA/BD,8CA+BC"}

package/dist/chains/chain.d.ts

@@ -0,0 +1,51 @@
+/**
+ * Multi-step chain pipeline for LambdaLLM.
+ */
+import { LambdaLLMContext } from '../core/context';
+export interface StepOptions {
+    name: string;
+    prompt?: string;
+    func?: (outputs: Record<string, any>) => any;
+    condition?: (outputs: Record<string, any>) => boolean;
+}
+export declare class Step {
+    name: string;
+    prompt?: string;
+    func?: (outputs: Record<string, any>) => any;
+    condition?: (outputs: Record<string, any>) => boolean;
+    constructor(options: StepOptions);
+    get isLlmStep(): boolean;
+    get isTransformStep(): boolean;
+}
+export interface ChainResult {
+    chainName: string;
+    status: 'completed' | 'checkpointed' | 'failed' | 'truncated';
+    steps: Array<{
+        name: string;
+        output: any;
+        costUsd: number;
+        latencyMs: number;
+        skipped: boolean;
+    }>;
+    finalOutput: any;
+    totalCostUsd: number;
+    totalLatencyMs: number;
+    completedSteps: number;
+    checkpoint?: any;
+}
+export interface ChainOptions {
+    name: string;
+    steps: Step[];
+    timeoutStrategy?: 'fail-fast' | 'checkpoint' | 'truncate';
+    maxTotalCost?: number;
+}
+export declare class Chain {
+    name: string;
+    steps: Step[];
+    timeoutStrategy: string;
+    maxTotalCost?: number;
+    constructor(options: ChainOptions);
+    run(context: LambdaLLMContext, variables: Record<string, any>): Promise<ChainResult>;
+    private resolveVariables;
+}
+//# sourceMappingURL=chain.d.ts.map

package/dist/chains/chain.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"chain.d.ts","sourceRoot":"","sources":["../../src/chains/chain.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,GAAG,CAAC;IAC7C,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC;CACvD;AAED,qBAAa,IAAI;IACR,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,GAAG,CAAC;IAC7C,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,OAAO,CAAC;gBAEjD,OAAO,EAAE,WAAW;IAUhC,IAAI,SAAS,IAAI,OAAO,CAA0B;IAClD,IAAI,eAAe,IAAI,OAAO,CAAwB;CACvD;AAED,MAAM,WAAW,WAAW;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,WAAW,GAAG,cAAc,GAAG,QAAQ,GAAG,WAAW,CAAC;IAC9D,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,GAAG,CAAC;QAAC,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;IAClG,WAAW,EAAE,GAAG,CAAC;IACjB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,GAAG,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,eAAe,CAAC,EAAE,WAAW,GAAG,YAAY,GAAG,UAAU,CAAC;IAC1D,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,KAAK;IACT,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,IAAI,EAAE,CAAC;IACd,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;gBAEjB,OAAO,EAAE,YAAY;IAQ3B,GAAG,CAAC,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,WAAW,CAAC;IAyE1F,OAAO,CAAC,gBAAgB;CAKzB"}

package/dist/chains/chain.js

@@ -0,0 +1,103 @@
+"use strict";
+/**
+ * Multi-step chain pipeline for LambdaLLM.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Chain = exports.Step = void 0;
+class Step {
+    constructor(options) {
+        if (!options.prompt && !options.func) {
+            throw new Error(`Step '${options.name}' must have either prompt or func`);
+        }
+        this.name = options.name;
+        this.prompt = options.prompt;
+        this.func = options.func;
+        this.condition = options.condition;
+    }
+    get isLlmStep() { return !!this.prompt; }
+    get isTransformStep() { return !!this.func; }
+}
+exports.Step = Step;
+class Chain {
+    constructor(options) {
+        if (!options.steps.length)
+            throw new Error('Chain must have at least one step');
+        this.name = options.name;
+        this.steps = options.steps;
+        this.timeoutStrategy = options.timeoutStrategy || 'fail-fast';
+        this.maxTotalCost = options.maxTotalCost;
+    }
+    async run(context, variables) {
+        const startTime = Date.now();
+        const outputs = { ...variables };
+        const stepResults = [];
+        let totalCost = 0;
+        for (let i = 0; i < this.steps.length; i++) {
+            const step = this.steps[i];
+            // Check timeout
+            if (this.timeoutStrategy === 'checkpoint' && context.shouldCheckpoint) {
+                return {
+                    chainName: this.name, status: 'checkpointed', steps: stepResults,
+                    finalOutput: stepResults[stepResults.length - 1]?.output,
+                    totalCostUsd: totalCost, totalLatencyMs: Date.now() - startTime,
+                    completedSteps: stepResults.filter(s => !s.skipped).length,
+                    checkpoint: { nextStepIndex: i, outputs },
+                };
+            }
+            // Check cost
+            if (this.maxTotalCost && totalCost >= this.maxTotalCost) {
+                return {
+                    chainName: this.name, status: 'truncated', steps: stepResults,
+                    finalOutput: stepResults[stepResults.length - 1]?.output,
+                    totalCostUsd: totalCost, totalLatencyMs: Date.now() - startTime,
+                    completedSteps: stepResults.filter(s => !s.skipped).length,
+                };
+            }
+            // Check condition
+            if (step.condition && !step.condition(outputs)) {
+                stepResults.push({ name: step.name, output: null, costUsd: 0, latencyMs: 0, skipped: true });
+                continue;
+            }
+            const stepStart = Date.now();
+            try {
+                let output;
+                if (step.isLlmStep) {
+                    const resolved = this.resolveVariables(step.prompt, outputs);
+                    output = await context.invoke(resolved);
+                }
+                else if (step.func) {
+                    output = step.func(outputs);
+                }
+                const stepCost = context.totalCost - totalCost;
+                totalCost = context.totalCost;
+                outputs[step.name] = output;
+                outputs[`${step.name}.output`] = output;
+                stepResults.push({
+                    name: step.name, output, costUsd: stepCost,
+                    latencyMs: Date.now() - stepStart, skipped: false,
+                });
+            }
+            catch (e) {
+                stepResults.push({ name: step.name, output: null, costUsd: 0, latencyMs: Date.now() - stepStart, skipped: false });
+                return {
+                    chainName: this.name, status: 'failed', steps: stepResults,
+                    finalOutput: null, totalCostUsd: totalCost, totalLatencyMs: Date.now() - startTime,
+                    completedSteps: stepResults.filter(s => !s.skipped).length,
+                };
+            }
+        }
+        return {
+            chainName: this.name, status: 'completed', steps: stepResults,
+            finalOutput: stepResults[stepResults.length - 1]?.output,
+            totalCostUsd: totalCost, totalLatencyMs: Date.now() - startTime,
+            completedSteps: stepResults.filter(s => !s.skipped).length,
+        };
+    }
+    resolveVariables(template, outputs) {
+        return template.replace(/\{([a-zA-Z_][a-zA-Z0-9_.]*)\}/g, (_, key) => {
+            return key in outputs ? String(outputs[key]) : `{${key}}`;
+        });
+    }
+}
+exports.Chain = Chain;
+//# sourceMappingURL=chain.js.map