@cleocode/lafs-protocol 0.5.0 → 1.1.0

package/dist/src/a2a/index.js

@@ -0,0 +1,36 @@
+/**
+ * LAFS Agent-to-Agent (A2A) Integration
+ *
+ * This module provides integration between LAFS and the official
+ * @a2a-js/sdk for Agent-to-Agent communication.
+ *
+ * @example
+ * ```typescript
+ * import { ClientFactory } from '@a2a-js/sdk/client';
+ * import { withLafsEnvelope } from '@cleocode/lafs-protocol/a2a';
+ *
+ * // Create official A2A client
+ * const factory = new ClientFactory();
+ * const a2aClient = await factory.createFromUrl('http://agent.example.com');
+ *
+ * // Wrap with LAFS support
+ * const client = withLafsEnvelope(a2aClient, {
+ *   defaultBudget: { maxTokens: 4000 }
+ * });
+ *
+ * // Send message
+ * const result = await client.sendMessage({
+ *   message: {
+ *     role: 'user',
+ *     parts: [{ text: 'Analyze data' }]
+ *   }
+ * });
+ *
+ * // Extract LAFS envelope from response
+ * const envelope = result.getLafsEnvelope();
+ * if (envelope) {
+ *   console.log(envelope._meta._tokenEstimate);
+ * }
+ * ```
+ */
+export { withLafsEnvelope, LafsA2AClient, LafsA2AResult, createLafsArtifact } from './bridge.js';

package/dist/src/budgetEnforcement.d.ts

@@ -0,0 +1,84 @@
+/**
+ * LAFS Budget Enforcement
+ *
+ * Middleware for enforcing MVI (Minimal Viable Interface) token budgets on LAFS envelopes.
+ * Provides budget checking, truncation, and error generation for exceeded budgets.
+ */
+import type { LAFSEnvelope } from "./types.js";
+import type { BudgetEnforcementOptions, TokenEstimate, BudgetEnforcementResult } from "./types.js";
+import { TokenEstimator } from "./tokenEstimator.js";
+/**
+ * Budget exceeded error code from LAFS error registry
+ */
+declare const BUDGET_EXCEEDED_CODE = "E_MVI_BUDGET_EXCEEDED";
+/**
+ * Apply budget enforcement to an envelope.
+ *
+ * @param envelope - The LAFS envelope to check
+ * @param budget - Maximum allowed tokens
+ * @param options - Budget enforcement options
+ * @returns Enforce result with potentially modified envelope
+ */
+export declare function applyBudgetEnforcement(envelope: LAFSEnvelope, budget: number, options?: BudgetEnforcementOptions): BudgetEnforcementResult;
+/**
+ * Type for middleware function
+ */
+type EnvelopeMiddleware = (envelope: LAFSEnvelope, next: () => LAFSEnvelope | Promise<LAFSEnvelope>) => Promise<LAFSEnvelope> | LAFSEnvelope;
+/**
+ * Create a budget enforcement middleware function.
+ *
+ * @param budget - Maximum allowed tokens for response
+ * @param options - Budget enforcement options
+ * @returns Middleware function that enforces budget
+ *
+ * @example
+ * ```typescript
+ * const middleware = withBudget(1000, { truncateOnExceed: true });
+ * const result = await middleware(envelope, async () => nextEnvelope);
+ * ```
+ */
+export declare function withBudget(budget: number, options?: BudgetEnforcementOptions): EnvelopeMiddleware;
+/**
+ * Check if an envelope has exceeded its budget without modifying it.
+ *
+ * @param envelope - The LAFS envelope to check
+ * @param budget - Maximum allowed tokens
+ * @returns Budget check result
+ */
+export declare function checkBudget(envelope: LAFSEnvelope, budget: number): {
+    exceeded: boolean;
+    estimated: number;
+    remaining: number;
+};
+/**
+ * Synchronous version of withBudget for non-async contexts.
+ *
+ * @param budget - Maximum allowed tokens for response
+ * @param options - Budget enforcement options
+ * @returns Middleware function that enforces budget synchronously
+ */
+export declare function withBudgetSync(budget: number, options?: BudgetEnforcementOptions): (envelope: LAFSEnvelope, next: () => LAFSEnvelope) => LAFSEnvelope;
+/**
+ * Higher-order function that wraps a handler with budget enforcement.
+ *
+ * @param handler - The handler function to wrap
+ * @param budget - Maximum allowed tokens
+ * @param options - Budget enforcement options
+ * @returns Wrapped handler with budget enforcement
+ *
+ * @example
+ * ```typescript
+ * const myHandler = async (request: Request) => ({ success: true, result: { data } });
+ * const budgetedHandler = wrapWithBudget(myHandler, 1000, { truncateOnExceed: true });
+ * const result = await budgetedHandler(request);
+ * ```
+ */
+export declare function wrapWithBudget<TArgs extends unknown[], TResult extends LAFSEnvelope>(handler: (...args: TArgs) => TResult | Promise<TResult>, budget: number, options?: BudgetEnforcementOptions): (...args: TArgs) => Promise<LAFSEnvelope>;
+/**
+ * Compose multiple middleware functions into a single middleware.
+ * Middleware is executed in order (left to right).
+ */
+export declare function composeMiddleware(...middlewares: EnvelopeMiddleware[]): EnvelopeMiddleware;
+export type { BudgetEnforcementOptions, TokenEstimate, BudgetEnforcementResult };
+export { TokenEstimator };
+export { BUDGET_EXCEEDED_CODE };

package/dist/src/budgetEnforcement.js

@@ -0,0 +1,328 @@
+/**
+ * LAFS Budget Enforcement
+ *
+ * Middleware for enforcing MVI (Minimal Viable Interface) token budgets on LAFS envelopes.
+ * Provides budget checking, truncation, and error generation for exceeded budgets.
+ */
+import { TokenEstimator } from "./tokenEstimator.js";
+/**
+ * Budget exceeded error code from LAFS error registry
+ */
+const BUDGET_EXCEEDED_CODE = "E_MVI_BUDGET_EXCEEDED";
+/**
+ * Default category for budget exceeded errors
+ */
+const BUDGET_ERROR_CATEGORY = "VALIDATION";
+/**
+ * Create a budget exceeded error object
+ */
+function createBudgetExceededError(estimated, budget) {
+    return {
+        code: BUDGET_EXCEEDED_CODE,
+        message: `Response exceeds declared MVI budget: estimated ${estimated} tokens, budget ${budget} tokens`,
+        category: BUDGET_ERROR_CATEGORY,
+        retryable: false,
+        retryAfterMs: null,
+        details: {
+            estimatedTokens: estimated,
+            budgetTokens: budget,
+            exceededBy: estimated - budget,
+            exceededByPercent: Math.round(((estimated - budget) / budget) * 100),
+        },
+    };
+}
+/**
+ * Truncate a result to fit within budget.
+ * Returns the truncated result and whether truncation occurred.
+ */
+function truncateResult(result, targetTokens, estimator) {
+    if (result === null) {
+        return { result: null, wasTruncated: false };
+    }
+    const currentEstimate = estimator.estimate(result);
+    // If already within budget, no truncation needed
+    if (currentEstimate <= targetTokens) {
+        return { result, wasTruncated: false };
+    }
+    // Calculate target size (conservative: assume 10% overhead)
+    const targetChars = Math.floor(targetTokens * 4 * 0.9);
+    if (Array.isArray(result)) {
+        return truncateArray(result, targetChars, targetTokens, estimator);
+    }
+    return truncateObject(result, targetChars, targetTokens, estimator);
+}
+/**
+ * Truncate an array to fit within budget.
+ */
+function truncateArray(arr, targetChars, targetTokens, estimator) {
+    if (arr.length === 0) {
+        return { result: arr, wasTruncated: false };
+    }
+    // Binary search to find how many items fit
+    let left = 0;
+    let right = arr.length;
+    let bestFit = 0;
+    while (left <= right) {
+        const mid = Math.floor((left + right) / 2);
+        const subset = arr.slice(0, mid);
+        const estimate = estimator.estimate(subset);
+        if (estimate <= targetTokens) {
+            bestFit = mid;
+            left = mid + 1;
+        }
+        else {
+            right = mid - 1;
+        }
+    }
+    // If we can fit all items, no truncation needed
+    if (bestFit >= arr.length) {
+        return { result: arr, wasTruncated: false };
+    }
+    // Create truncated result
+    const truncated = arr.slice(0, bestFit);
+    // If we couldn't fit any items, return minimal response
+    if (bestFit === 0 && arr.length > 0) {
+        return {
+            result: [{ _truncated: true, reason: "budget_exceeded" }],
+            wasTruncated: true
+        };
+    }
+    // Add truncation indicator to last element if it's an object
+    if (bestFit > 0 && typeof truncated[bestFit - 1] === 'object' && truncated[bestFit - 1] !== null) {
+        const lastItem = truncated[bestFit - 1];
+        truncated[bestFit - 1] = {
+            ...lastItem,
+            _truncated: true,
+            remainingItems: arr.length - bestFit,
+        };
+    }
+    return { result: truncated, wasTruncated: true };
+}
+/**
+ * Truncate an object to fit within budget.
+ */
+function truncateObject(obj, targetChars, targetTokens, estimator) {
+    const keys = Object.keys(obj);
+    if (keys.length === 0) {
+        return { result: obj, wasTruncated: false };
+    }
+    // Try to fit as many top-level properties as possible
+    let left = 0;
+    let right = keys.length;
+    let bestFit = 0;
+    while (left <= right) {
+        const mid = Math.floor((left + right) / 2);
+        const subsetKeys = keys.slice(0, mid);
+        const subset = {};
+        for (const key of subsetKeys) {
+            subset[key] = obj[key];
+        }
+        const estimate = estimator.estimate(subset);
+        if (estimate <= targetTokens) {
+            bestFit = mid;
+            left = mid + 1;
+        }
+        else {
+            right = mid - 1;
+        }
+    }
+    // If we can fit all properties, no truncation needed
+    if (bestFit >= keys.length) {
+        return { result: obj, wasTruncated: false };
+    }
+    // Create truncated result
+    const subsetKeys = keys.slice(0, bestFit);
+    const truncated = {};
+    for (const key of subsetKeys) {
+        truncated[key] = obj[key];
+    }
+    // If we couldn't fit any properties, return minimal response
+    if (bestFit === 0) {
+        return {
+            result: { _truncated: true, reason: "budget_exceeded" },
+            wasTruncated: true
+        };
+    }
+    // Add truncation metadata
+    truncated._truncated = true;
+    truncated._truncatedFields = keys.slice(bestFit);
+    return { result: truncated, wasTruncated: true };
+}
+/**
+ * Apply budget enforcement to an envelope.
+ *
+ * @param envelope - The LAFS envelope to check
+ * @param budget - Maximum allowed tokens
+ * @param options - Budget enforcement options
+ * @returns Enforce result with potentially modified envelope
+ */
+export function applyBudgetEnforcement(envelope, budget, options = {}) {
+    const { truncateOnExceed = false, onBudgetExceeded } = options;
+    const estimator = new TokenEstimator();
+    // Estimate the result payload
+    const estimatedTokens = estimator.estimate(envelope.result);
+    // Add estimate to metadata
+    const tokenEstimate = {
+        estimated: estimatedTokens,
+    };
+    // Check if within budget
+    const withinBudget = estimatedTokens <= budget;
+    // If within budget, just add the estimate to metadata
+    if (withinBudget) {
+        return {
+            envelope: {
+                ...envelope,
+                _meta: {
+                    ...envelope._meta,
+                    _tokenEstimate: tokenEstimate,
+                },
+            },
+            withinBudget: true,
+            estimatedTokens,
+            budget,
+            truncated: false,
+        };
+    }
+    // Budget exceeded - call callback if provided
+    if (onBudgetExceeded) {
+        onBudgetExceeded(estimatedTokens, budget);
+    }
+    // If truncation is enabled, try to truncate
+    if (truncateOnExceed) {
+        const { result, wasTruncated } = truncateResult(envelope.result, budget, estimator);
+        const truncatedEstimate = estimator.estimate(result);
+        if (truncatedEstimate <= budget) {
+            return {
+                envelope: {
+                    ...envelope,
+                    result,
+                    _meta: {
+                        ...envelope._meta,
+                        _tokenEstimate: {
+                            estimated: truncatedEstimate,
+                            truncated: true,
+                            originalEstimate: estimatedTokens,
+                        },
+                    },
+                },
+                withinBudget: true,
+                estimatedTokens: truncatedEstimate,
+                budget,
+                truncated: true,
+            };
+        }
+    }
+    // Return budget exceeded error
+    return {
+        envelope: {
+            ...envelope,
+            success: false,
+            result: null,
+            error: createBudgetExceededError(estimatedTokens, budget),
+            _meta: {
+                ...envelope._meta,
+                _tokenEstimate: tokenEstimate,
+            },
+        },
+        withinBudget: false,
+        estimatedTokens,
+        budget,
+        truncated: false,
+    };
+}
+/**
+ * Create a budget enforcement middleware function.
+ *
+ * @param budget - Maximum allowed tokens for response
+ * @param options - Budget enforcement options
+ * @returns Middleware function that enforces budget
+ *
+ * @example
+ * ```typescript
+ * const middleware = withBudget(1000, { truncateOnExceed: true });
+ * const result = await middleware(envelope, async () => nextEnvelope);
+ * ```
+ */
+export function withBudget(budget, options = {}) {
+    return async (envelope, next) => {
+        // Execute next middleware/handler
+        const result = await next();
+        // Apply budget enforcement to the result
+        const enforcement = applyBudgetEnforcement(result, budget, options);
+        return enforcement.envelope;
+    };
+}
+/**
+ * Check if an envelope has exceeded its budget without modifying it.
+ *
+ * @param envelope - The LAFS envelope to check
+ * @param budget - Maximum allowed tokens
+ * @returns Budget check result
+ */
+export function checkBudget(envelope, budget) {
+    const estimator = new TokenEstimator();
+    const estimated = estimator.estimate(envelope.result);
+    return {
+        exceeded: estimated > budget,
+        estimated,
+        remaining: Math.max(0, budget - estimated),
+    };
+}
+/**
+ * Synchronous version of withBudget for non-async contexts.
+ *
+ * @param budget - Maximum allowed tokens for response
+ * @param options - Budget enforcement options
+ * @returns Middleware function that enforces budget synchronously
+ */
+export function withBudgetSync(budget, options = {}) {
+    return (envelope, next) => {
+        const result = next();
+        const enforcement = applyBudgetEnforcement(result, budget, options);
+        return enforcement.envelope;
+    };
+}
+/**
+ * Higher-order function that wraps a handler with budget enforcement.
+ *
+ * @param handler - The handler function to wrap
+ * @param budget - Maximum allowed tokens
+ * @param options - Budget enforcement options
+ * @returns Wrapped handler with budget enforcement
+ *
+ * @example
+ * ```typescript
+ * const myHandler = async (request: Request) => ({ success: true, result: { data } });
+ * const budgetedHandler = wrapWithBudget(myHandler, 1000, { truncateOnExceed: true });
+ * const result = await budgetedHandler(request);
+ * ```
+ */
+export function wrapWithBudget(handler, budget, options = {}) {
+    return async (...args) => {
+        const result = await handler(...args);
+        const enforcement = applyBudgetEnforcement(result, budget, options);
+        return enforcement.envelope;
+    };
+}
+/**
+ * Compose multiple middleware functions into a single middleware.
+ * Middleware is executed in order (left to right).
+ */
+export function composeMiddleware(...middlewares) {
+    return async (envelope, next) => {
+        let index = 0;
+        async function dispatch(i) {
+            if (i >= middlewares.length) {
+                return next();
+            }
+            const middleware = middlewares[i];
+            if (!middleware) {
+                return dispatch(i + 1);
+            }
+            return middleware(envelope, () => dispatch(i + 1));
+        }
+        return dispatch(0);
+    };
+}
+export { TokenEstimator };
+export { BUDGET_EXCEEDED_CODE };

package/dist/src/circuit-breaker/index.d.ts

@@ -0,0 +1,121 @@
+/**
+ * LAFS Circuit Breaker Module
+ *
+ * Provides circuit breaker pattern for resilient service calls
+ */
+export type CircuitState = 'CLOSED' | 'OPEN' | 'HALF_OPEN';
+export interface CircuitBreakerConfig {
+    name: string;
+    failureThreshold?: number;
+    resetTimeout?: number;
+    halfOpenMaxCalls?: number;
+    successThreshold?: number;
+}
+export interface CircuitBreakerMetrics {
+    state: CircuitState;
+    failures: number;
+    successes: number;
+    lastFailureTime?: Date;
+    consecutiveSuccesses: number;
+    totalCalls: number;
+}
+export declare class CircuitBreakerError extends Error {
+    constructor(message: string);
+}
+/**
+ * Circuit breaker for protecting against cascading failures
+ *
+ * @example
+ * ```typescript
+ * import { CircuitBreaker } from '@cleocode/lafs-protocol/circuit-breaker';
+ *
+ * const breaker = new CircuitBreaker({
+ *   name: 'external-api',
+ *   failureThreshold: 5,
+ *   resetTimeout: 30000
+ * });
+ *
+ * try {
+ *   const result = await breaker.execute(async () => {
+ *     return await externalApi.call();
+ *   });
+ * } catch (error) {
+ *   if (error instanceof CircuitBreakerError) {
+ *     console.log('Circuit breaker is open');
+ *   }
+ * }
+ * ```
+ */
+export declare class CircuitBreaker {
+    private config;
+    private state;
+    private failures;
+    private successes;
+    private lastFailureTime?;
+    private consecutiveSuccesses;
+    private totalCalls;
+    private halfOpenCalls;
+    private resetTimer?;
+    constructor(config: CircuitBreakerConfig);
+    /**
+     * Execute a function with circuit breaker protection
+     */
+    execute<T>(fn: () => Promise<T>): Promise<T>;
+    /**
+     * Get current circuit breaker state
+     */
+    getState(): CircuitState;
+    /**
+     * Get circuit breaker metrics
+     */
+    getMetrics(): CircuitBreakerMetrics;
+    /**
+     * Manually open the circuit breaker
+     */
+    forceOpen(): void;
+    /**
+     * Manually close the circuit breaker
+     */
+    forceClose(): void;
+    private onSuccess;
+    private onFailure;
+    private transitionTo;
+    private shouldAttemptReset;
+    private scheduleReset;
+    private reset;
+}
+/**
+ * Circuit breaker registry for managing multiple breakers
+ *
+ * @example
+ * ```typescript
+ * const registry = new CircuitBreakerRegistry();
+ *
+ * registry.add('payment-api', {
+ *   failureThreshold: 3,
+ *   resetTimeout: 60000
+ * });
+ *
+ * const paymentBreaker = registry.get('payment-api');
+ * ```
+ */
+export declare class CircuitBreakerRegistry {
+    private breakers;
+    add(name: string, config: Omit<CircuitBreakerConfig, 'name'>): CircuitBreaker;
+    get(name: string): CircuitBreaker | undefined;
+    getOrCreate(name: string, config: Omit<CircuitBreakerConfig, 'name'>): CircuitBreaker;
+    getAllMetrics(): Record<string, CircuitBreakerMetrics>;
+    resetAll(): void;
+}
+/**
+ * Create a circuit breaker middleware for Express
+ *
+ * @example
+ * ```typescript
+ * app.use('/external-api', circuitBreakerMiddleware({
+ *   name: 'external-api',
+ *   failureThreshold: 5
+ * }));
+ * ```
+ */
+export declare function circuitBreakerMiddleware(config: CircuitBreakerConfig): (req: any, res: any, next: any) => Promise<void>;