@lleverage-ai/agent-sdk 0.0.1

package/dist/hooks/cache.js

@@ -0,0 +1,273 @@
+/**
+ * Cache hook utilities.
+ *
+ * Provides cache hooks that replace the cache middleware functionality
+ * using the unified hook system.
+ *
+ * @packageDocumentation
+ */
+/**
+ * Default in-memory cache store implementation with LRU eviction.
+ *
+ * @category Hooks
+ */
+export class InMemoryCacheStore {
+    cache = new Map();
+    maxSize;
+    constructor(maxSize = 100) {
+        this.maxSize = maxSize;
+    }
+    get(key) {
+        return this.cache.get(key);
+    }
+    set(key, value) {
+        // LRU eviction: remove oldest entry if at capacity
+        if (this.cache.size >= this.maxSize) {
+            const firstKey = this.cache.keys().next().value;
+            if (firstKey !== undefined) {
+                this.cache.delete(firstKey);
+            }
+        }
+        this.cache.set(key, value);
+    }
+    delete(key) {
+        return this.cache.delete(key);
+    }
+    clear() {
+        this.cache.clear();
+    }
+    /** Get current cache size */
+    get size() {
+        return this.cache.size;
+    }
+}
+/**
+ * Default cache key generator.
+ *
+ * Creates a stable key from generation options including model identifier.
+ * Addresses the cache key issue mentioned in CODE_REVIEW.md.
+ */
+function defaultKeyGenerator(input, context) {
+    const opts = input.options;
+    // Extract model identifier from agent context
+    // The AI SDK LanguageModel may have modelId as a property or be a string
+    const model = context?.agent?.options?.model;
+    let modelId = "unknown";
+    if (model) {
+        if (typeof model === "string") {
+            modelId = model;
+        }
+        else if ("modelId" in model && typeof model.modelId === "string") {
+            modelId = model.modelId;
+        }
+        else if ("specificationVersion" in model) {
+            // For LanguageModel objects, try to extract identifier from provider
+            modelId = String(model);
+        }
+    }
+    const keyData = {
+        // Include model to differentiate between different models
+        model: modelId,
+        // Include session to differentiate conversations
+        session: input.session_id,
+        messages: opts.messages,
+        temperature: opts.temperature,
+        maxTokens: opts.maxTokens,
+    };
+    return JSON.stringify(keyData);
+}
+/**
+ * Creates cache hooks for PreGenerate and PostGenerate events.
+ *
+ * The PreGenerate hook checks for cached results and returns them via
+ * `respondWith` to short-circuit generation. The PostGenerate hook stores
+ * results in the cache.
+ *
+ * This replaces the cache middleware with hook-based caching that works
+ * correctly with streaming (hooks fire at lifecycle boundaries).
+ *
+ * @param options - Configuration options
+ * @returns Array of two hooks: [PreGenerate cache check, PostGenerate cache store]
+ *
+ * @example
+ * ```typescript
+ * const [cacheCheck, cacheStore] = createCacheHooks({
+ *   ttl: 60000,  // 1 minute
+ *   maxSize: 50,
+ * });
+ *
+ * const agent = createAgent({
+ *   model,
+ *   hooks: {
+ *     PreGenerate: [{ hooks: [cacheCheck] }],
+ *     PostGenerate: [{ hooks: [cacheStore] }],
+ *   },
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Custom cache key based on last message
+ * const hooks = createCacheHooks({
+ *   keyGenerator: (input) => {
+ *     const lastMsg = input.options.messages?.[input.options.messages.length - 1];
+ *     return JSON.stringify(lastMsg);
+ *   },
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Conditional caching (only cache when no tools)
+ * const hooks = createCacheHooks({
+ *   shouldCache: (input) => !input.options.tools || Object.keys(input.options.tools).length === 0,
+ *   ttl: 300000,
+ * });
+ * ```
+ *
+ * @category Hooks
+ */
+export function createCacheHooks(options = {}) {
+    const { ttl = 300000, // 5 minutes
+    maxSize = 100, keyGenerator = defaultKeyGenerator, store = new InMemoryCacheStore(maxSize), shouldCache = () => true, } = options;
+    // PreGenerate: Check cache and short-circuit if hit
+    const cacheCheck = async (input, _toolUseId, context) => {
+        if (input.hook_event_name !== "PreGenerate")
+            return {};
+        const preGenInput = input;
+        // Check if this request should be cached
+        if (!shouldCache(preGenInput)) {
+            return {};
+        }
+        const key = keyGenerator(preGenInput, context);
+        // Check cache
+        const cached = await store.get(key);
+        if (cached && Date.now() - cached.timestamp < ttl) {
+            // Cache hit - return cached result via respondWith
+            return {
+                hookSpecificOutput: {
+                    hookEventName: "PreGenerate",
+                    respondWith: cached.result,
+                },
+            };
+        }
+        // Cache miss - continue with generation
+        return {};
+    };
+    // PostGenerate: Store result in cache
+    const cacheStore = async (input, _toolUseId, context) => {
+        if (input.hook_event_name !== "PostGenerate")
+            return {};
+        const postGenInput = input;
+        // Check if this request should be cached
+        // Convert PostGenerateInput to PreGenerateInput format for key generation
+        const preGenFormat = {
+            hook_event_name: "PreGenerate",
+            session_id: postGenInput.session_id,
+            cwd: postGenInput.cwd,
+            options: postGenInput.options,
+        };
+        if (!shouldCache(preGenFormat)) {
+            return {};
+        }
+        const key = keyGenerator(preGenFormat, context);
+        // Store result in cache
+        await store.set(key, {
+            result: postGenInput.result,
+            timestamp: Date.now(),
+        });
+        return {};
+    };
+    return [cacheCheck, cacheStore];
+}
+/**
+ * Creates managed cache hooks with programmatic cache control.
+ *
+ * Returns hooks along with functions to clear cache, delete entries,
+ * and get statistics.
+ *
+ * @param options - Configuration options
+ * @returns Object with hooks and cache control functions
+ *
+ * @example
+ * ```typescript
+ * const { hooks, clearCache, getStats } = createManagedCacheHooks();
+ *
+ * const agent = createAgent({
+ *   model,
+ *   hooks: {
+ *     PreGenerate: [{ hooks: [hooks[0]] }],
+ *     PostGenerate: [{ hooks: [hooks[1]] }],
+ *   },
+ * });
+ *
+ * // Clear cache when needed
+ * await clearCache();
+ *
+ * // Get cache statistics
+ * const { size, hits, misses } = getStats();
+ * console.log(`Cache hit rate: ${hits / (hits + misses)}`);
+ * ```
+ *
+ * @category Hooks
+ */
+export function createManagedCacheHooks(options = {}) {
+    const { ttl = 300000, maxSize = 100, keyGenerator = defaultKeyGenerator, store = new InMemoryCacheStore(maxSize), shouldCache = () => true, } = options;
+    let hits = 0;
+    let misses = 0;
+    // PreGenerate: Check cache with stats tracking
+    const cacheCheck = async (input, _toolUseId, context) => {
+        if (input.hook_event_name !== "PreGenerate")
+            return {};
+        const preGenInput = input;
+        if (!shouldCache(preGenInput)) {
+            return {};
+        }
+        const key = keyGenerator(preGenInput, context);
+        const cached = await store.get(key);
+        if (cached && Date.now() - cached.timestamp < ttl) {
+            hits++;
+            return {
+                hookSpecificOutput: {
+                    hookEventName: "PreGenerate",
+                    respondWith: cached.result,
+                },
+            };
+        }
+        misses++;
+        return {};
+    };
+    // PostGenerate: Store result
+    const cacheStore = async (input, _toolUseId, context) => {
+        if (input.hook_event_name !== "PostGenerate")
+            return {};
+        const postGenInput = input;
+        // Convert PostGenerateInput to PreGenerateInput format for key generation
+        const preGenFormat = {
+            hook_event_name: "PreGenerate",
+            session_id: postGenInput.session_id,
+            cwd: postGenInput.cwd,
+            options: postGenInput.options,
+        };
+        if (!shouldCache(preGenFormat)) {
+            return {};
+        }
+        const key = keyGenerator(preGenFormat, context);
+        await store.set(key, {
+            result: postGenInput.result,
+            timestamp: Date.now(),
+        });
+        return {};
+    };
+    return {
+        hooks: [cacheCheck, cacheStore],
+        clearCache: () => store.clear(),
+        deleteEntry: (key) => store.delete(key),
+        getStats: () => ({
+            size: store instanceof InMemoryCacheStore ? store.size : -1,
+            hits,
+            misses,
+        }),
+    };
+}
+//# sourceMappingURL=cache.js.map

package/dist/hooks/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../../src/hooks/cache.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAwCH;;;;GAIG;AACH,MAAM,OAAO,kBAAkB;IACrB,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IACtC,OAAO,CAAS;IAExB,YAAY,OAAO,GAAG,GAAG;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,GAAG,CAAC,GAAW;QACb,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,GAAW,EAAE,KAAiB;QAChC,mDAAmD;QACnD,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YAChD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,CAAC,GAAW;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,6BAA6B;IAC7B,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;CACF;AAuCD;;;;;GAKG;AACH,SAAS,mBAAmB,CAAC,KAAuB,EAAE,OAA6B;IACjF,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC;IAC3B,8CAA8C;IAC9C,yEAAyE;IACzE,MAAM,KAAK,GAAG,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,CAAC;IAC7C,IAAI,OAAO,GAAG,SAAS,CAAC;IACxB,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC;aAAM,IAAI,SAAS,IAAI,KAAK,IAAI,OAAQ,KAA8B,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC7F,OAAO,GAAI,KAA6B,CAAC,OAAO,CAAC;QACnD,CAAC;aAAM,IAAI,sBAAsB,IAAI,KAAK,EAAE,CAAC;YAC3C,qEAAqE;YACrE,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,MAAM,OAAO,GAAG;QACd,0DAA0D;QAC1D,KAAK,EAAE,OAAO;QACd,iDAAiD;QACjD,OAAO,EAAE,KAAK,CAAC,UAAU;QACzB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;KAC1B,CAAC;IACF,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkDG;AACH,MAAM,UAAU,gBAAgB,CAAC,UAA6B,EAAE;IAC9D,MAAM,EACJ,GAAG,GAAG,MAAM,EAAE,YAAY;IAC1B,OAAO,GAAG,GAAG,EACb,YAAY,GAAG,mBAAmB,EAClC,KAAK,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,EACvC,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,GACzB,GAAG,OAAO,CAAC;IAEZ,oDAAoD;IACpD,MAAM,UAAU,GAAiB,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE;QACpE,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa;YAAE,OAAO,EAAE,CAAC;QAEvD,MAAM,WAAW,GAAG,KAAyB,CAAC;QAE9C,yCAAyC;QACzC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,GAAG,GAAG,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAE/C,cAAc;QACd,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;YAClD,mDAAmD;YACnD,OAAO;gBACL,kBAAkB,EAAE;oBAClB,aAAa,EAAE,aAAa;oBAC5B,WAAW,EAAE,MAAM,CAAC,MAAM;iBAC3B;aACF,CAAC;QACJ,CAAC;QAED,wCAAwC;QACxC,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,sCAAsC;IACtC,MAAM,UAAU,GAAiB,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE;QACpE,IAAI,KAAK,CAAC,eAAe,KAAK,cAAc;YAAE,OAAO,EAAE,CAAC;QAExD,MAAM,YAAY,GAAG,KAA0B,CAAC;QAEhD,yCAAyC;QACzC,0EAA0E;QAC1E,MAAM,YAAY,GAAqB;YACrC,eAAe,EAAE,aAAa;YAC9B,UAAU,EAAE,YAAY,CAAC,UAAU;YACnC,GAAG,EAAE,YAAY,CAAC,GAAG;YACrB,OAAO,EAAE,YAAY,CAAC,OAAO;SAC9B,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,GAAG,GAAG,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAEhD,wBAAwB;QACxB,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;YACnB,MAAM,EAAE,YAAY,CAAC,MAAM;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,OAAO,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;AAClC,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,UAAU,uBAAuB,CAAC,UAA6B,EAAE;IAMrE,MAAM,EACJ,GAAG,GAAG,MAAM,EACZ,OAAO,GAAG,GAAG,EACb,YAAY,GAAG,mBAAmB,EAClC,KAAK,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,EACvC,WAAW,GAAG,GAAG,EAAE,CAAC,IAAI,GACzB,GAAG,OAAO,CAAC;IAEZ,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,MAAM,GAAG,CAAC,CAAC;IAEf,+CAA+C;IAC/C,MAAM,UAAU,GAAiB,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE;QACpE,IAAI,KAAK,CAAC,eAAe,KAAK,aAAa;YAAE,OAAO,EAAE,CAAC;QAEvD,MAAM,WAAW,GAAG,KAAyB,CAAC;QAE9C,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;YAC9B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,GAAG,GAAG,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC/C,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAEpC,IAAI,MAAM,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,SAAS,GAAG,GAAG,EAAE,CAAC;YAClD,IAAI,EAAE,CAAC;YACP,OAAO;gBACL,kBAAkB,EAAE;oBAClB,aAAa,EAAE,aAAa;oBAC5B,WAAW,EAAE,MAAM,CAAC,MAAM;iBAC3B;aACF,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,CAAC;QACT,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,6BAA6B;IAC7B,MAAM,UAAU,GAAiB,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,EAAE;QACpE,IAAI,KAAK,CAAC,eAAe,KAAK,cAAc;YAAE,OAAO,EAAE,CAAC;QAExD,MAAM,YAAY,GAAG,KAA0B,CAAC;QAEhD,0EAA0E;QAC1E,MAAM,YAAY,GAAqB;YACrC,eAAe,EAAE,aAAa;YAC9B,UAAU,EAAE,YAAY,CAAC,UAAU;YACnC,GAAG,EAAE,YAAY,CAAC,GAAG;YACrB,OAAO,EAAE,YAAY,CAAC,OAAO;SAC9B,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,GAAG,GAAG,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAChD,MAAM,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE;YACnB,MAAM,EAAE,YAAY,CAAC,MAAM;YAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IAEF,OAAO;QACL,KAAK,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC;QAC/B,UAAU,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE;QAC/B,WAAW,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC;QAC/C,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;YACf,IAAI,EAAE,KAAK,YAAY,kBAAkB,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,IAAI;YACJ,MAAM;SACP,CAAC;KACH,CAAC;AACJ,CAAC"}

package/dist/hooks/guardrails.d.ts

@@ -0,0 +1,145 @@
+/**
+ * Guardrails hook utilities.
+ *
+ * Provides content filtering and safety hooks using the unified hook system.
+ *
+ * @packageDocumentation
+ */
+import type { HookCallback, PostGenerateInput, PreGenerateInput } from "../types.js";
+/**
+ * Options for creating guardrails hooks.
+ *
+ * @category Hooks
+ */
+export interface GuardrailsHooksOptions {
+    /**
+     * Regex patterns to block in input (user messages).
+     * If matched, generation is denied.
+     */
+    blockedInputPatterns?: RegExp[];
+    /**
+     * Regex patterns to filter in output (model responses).
+     * If matched, output is replaced with filterMessage.
+     */
+    blockedOutputPatterns?: RegExp[];
+    /**
+     * Message to show when input is blocked.
+     * @defaultValue "Request blocked by content policy"
+     */
+    blockedInputMessage?: string;
+    /**
+     * Message to replace filtered output with.
+     * @defaultValue "[Content filtered]"
+     */
+    filteredOutputMessage?: string;
+    /**
+     * Custom input validation function.
+     * Throw error to block, or return options to allow/transform.
+     */
+    checkInput?: (input: PreGenerateInput) => PreGenerateInput | undefined;
+    /**
+     * Custom output validation function.
+     * Return modified result to filter/transform output.
+     */
+    checkOutput?: (input: PostGenerateInput) => PostGenerateInput["result"] | undefined;
+}
+/**
+ * Creates guardrails hooks for input and output content filtering.
+ *
+ * The PreGenerate hook blocks requests matching input patterns.
+ * The PostGenerate hook filters output matching output patterns.
+ *
+ * This replaces guardrails middleware with hook-based filtering that
+ * works correctly with the unified hook system.
+ *
+ * @param options - Configuration options
+ * @returns Array of two hooks: [PreGenerate input filter, PostGenerate output filter]
+ *
+ * @example
+ * ```typescript
+ * const [inputFilter, outputFilter] = createGuardrailsHooks({
+ *   blockedInputPatterns: [
+ *     /password/i,
+ *     /api.?key/i,
+ *     /secret/i,
+ *   ],
+ *   blockedOutputPatterns: [
+ *     /\b\d{16}\b/,  // Credit card numbers
+ *     /\b\d{3}-\d{2}-\d{4}\b/,  // SSN
+ *   ],
+ * });
+ *
+ * const agent = createAgent({
+ *   model,
+ *   hooks: {
+ *     PreGenerate: [{ hooks: [inputFilter] }],
+ *     PostGenerate: [{ hooks: [outputFilter] }],
+ *   },
+ * });
+ * ```
+ *
+ * @example
+ * ```typescript
+ * // Custom validation with transformation
+ * const hooks = createGuardrailsHooks({
+ *   checkInput: (input) => {
+ *     // Allow but transform: remove PII from messages
+ *     const cleaned = cleanPII(input.options.messages);
+ *     return {
+ *       ...input,
+ *       options: { ...input.options, messages: cleaned },
+ *     };
+ *   },
+ *   checkOutput: (input) => {
+ *     // Filter harmful content
+ *     if (isHarmful(input.result.text)) {
+ *       return { ...input.result, text: "[Content filtered for safety]" };
+ *     }
+ *   },
+ * });
+ * ```
+ *
+ * @category Hooks
+ */
+export declare function createGuardrailsHooks(options?: GuardrailsHooksOptions): [HookCallback, HookCallback];
+/**
+ * Creates guardrails hooks with statistics tracking.
+ *
+ * Returns hooks along with functions to get filtering statistics.
+ *
+ * @param options - Configuration options
+ * @returns Object with hooks and statistics getter
+ *
+ * @example
+ * ```typescript
+ * const { hooks, getStats } = createManagedGuardrailsHooks({
+ *   blockedInputPatterns: [/password/i, /secret/i],
+ *   blockedOutputPatterns: [/\b\d{16}\b/],
+ * });
+ *
+ * const agent = createAgent({
+ *   model,
+ *   hooks: {
+ *     PreGenerate: [{ hooks: [hooks[0]] }],
+ *     PostGenerate: [{ hooks: [hooks[1]] }],
+ *   },
+ * });
+ *
+ * // Check statistics
+ * const stats = getStats();
+ * console.log(`Blocked inputs: ${stats.blockedInputs}`);
+ * console.log(`Filtered outputs: ${stats.filteredOutputs}`);
+ * ```
+ *
+ * @category Hooks
+ */
+export declare function createManagedGuardrailsHooks(options?: GuardrailsHooksOptions): {
+    hooks: [HookCallback, HookCallback];
+    getStats: () => {
+        blockedInputs: number;
+        filteredOutputs: number;
+        totalInputs: number;
+        totalOutputs: number;
+    };
+};
+//# sourceMappingURL=guardrails.d.ts.map

package/dist/hooks/guardrails.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"guardrails.d.ts","sourceRoot":"","sources":["../../src/hooks/guardrails.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAErF;;;;GAIG;AACH,MAAM,WAAW,sBAAsB;IACrC;;;OAGG;IACH,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAEhC;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,EAAE,CAAC;IAEjC;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAE7B;;;OAGG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAE/B;;;OAGG;IACH,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,gBAAgB,KAAK,gBAAgB,GAAG,SAAS,CAAC;IAEvE;;;OAGG;IACH,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,iBAAiB,CAAC,QAAQ,CAAC,GAAG,SAAS,CAAC;CACrF;AA2ED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyDG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,GAAE,sBAA2B,GACnC,CAAC,YAAY,EAAE,YAAY,CAAC,CAmG9B;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,wBAAgB,4BAA4B,CAAC,OAAO,GAAE,sBAA2B,GAAG;IAClF,KAAK,EAAE,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IACpC,QAAQ,EAAE,MAAM;QACd,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH,CA2GA"}