@juspay/neurolink 7.28.1 → 7.29.1

package/CHANGELOG.md

@@ -1,3 +1,15 @@
+## [7.29.1](https://github.com/juspay/neurolink/compare/v7.29.0...v7.29.1) (2025-08-28)
+
+### Bug Fixes
+
+- **(vertex):** restored support for adc ([238666a](https://github.com/juspay/neurolink/commit/238666ab907fc16945d5de6c5f79637be128f4e6))
+
+## [7.29.0](https://github.com/juspay/neurolink/compare/v7.28.1...v7.29.0) (2025-08-26)
+
+### Features
+
+- **(guardrails):** added guardrails as a middleware ([ac60f6b](https://github.com/juspay/neurolink/commit/ac60f6b143a58f86e17481ddb3e067e5307391cf))
+
 ## [7.28.1](https://github.com/juspay/neurolink/compare/v7.28.0...v7.28.1) (2025-08-26)
 
 ### Bug Fixes

package/dist/core/baseProvider.js

@@ -413,13 +413,15 @@ export class BaseProvider {
             return baseModel;
         }
         try {
+            // Create a new factory instance with the specified options
+            const factory = new MiddlewareFactory(middlewareOptions);
             // Create middleware context
-            const context = MiddlewareFactory.createContext(this.providerName, this.modelName, options, {
+            const context = factory.createContext(this.providerName, this.modelName, options, {
                 sessionId: this.sessionId,
                 userId: this.userId,
             });
             // Apply middleware to the model
-            const wrappedModel = MiddlewareFactory.applyMiddleware(baseModel, context, middlewareOptions);
+            const wrappedModel = factory.applyMiddleware(baseModel, context, middlewareOptions);
             logger.debug(`Applied middleware to ${this.providerName} model`, {
                 provider: this.providerName,
                 model: this.modelName,

package/dist/index.d.ts

@@ -18,9 +18,8 @@ export { getBestProvider, getAvailableProviders, isValidProvider, } from "./util
 export { NeuroLink } from "./neurolink.js";
 export type { ProviderStatus, MCPStatus } from "./neurolink.js";
 export type { MCPServerInfo } from "./types/mcpTypes.js";
-export type { NeuroLinkMiddleware, MiddlewareContext, MiddlewareFactoryOptions, } from "./middleware/types.js";
-export { MiddlewareRegistry, MiddlewareFactory } from "./middleware/index.js";
-export { createAnalyticsMiddleware } from "./middleware/builtin/analytics.js";
+export type { NeuroLinkMiddleware, MiddlewareContext, MiddlewareFactoryOptions, MiddlewarePreset, MiddlewareConfig, } from "./middleware/types.js";
+export { MiddlewareFactory } from "./middleware/factory.js";
 export declare const VERSION = "1.0.0";
 /**
  * Quick start factory function

package/dist/index.js

@@ -16,8 +16,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 // Main NeuroLink wrapper class and diagnostic types
 export { NeuroLink } from "./neurolink.js";
-export { MiddlewareRegistry, MiddlewareFactory } from "./middleware/index.js";
-export { createAnalyticsMiddleware } from "./middleware/builtin/analytics.js";
+export { MiddlewareFactory } from "./middleware/factory.js";
 // Version
 export const VERSION = "1.0.0";
 /**

package/dist/lib/core/baseProvider.js

@@ -413,13 +413,15 @@ export class BaseProvider {
             return baseModel;
         }
         try {
+            // Create a new factory instance with the specified options
+            const factory = new MiddlewareFactory(middlewareOptions);
             // Create middleware context
-            const context = MiddlewareFactory.createContext(this.providerName, this.modelName, options, {
+            const context = factory.createContext(this.providerName, this.modelName, options, {
                 sessionId: this.sessionId,
                 userId: this.userId,
             });
             // Apply middleware to the model
-            const wrappedModel = MiddlewareFactory.applyMiddleware(baseModel, context, middlewareOptions);
+            const wrappedModel = factory.applyMiddleware(baseModel, context, middlewareOptions);
             logger.debug(`Applied middleware to ${this.providerName} model`, {
                 provider: this.providerName,
                 model: this.modelName,

package/dist/lib/core/dynamicModels.d.ts

@@ -20,8 +20,8 @@ declare const ModelConfigSchema: z.ZodObject<{
     contextWindow: z.ZodNumber;
     releaseDate: z.ZodString;
 }, "strip", z.ZodTypeAny, {
-    capabilities: string[];
     id: string;
+    capabilities: string[];
     displayName: string;
     deprecated: boolean;
     pricing: {
@@ -31,8 +31,8 @@ declare const ModelConfigSchema: z.ZodObject<{
     contextWindow: number;
     releaseDate: string;
 }, {
-    capabilities: string[];
     id: string;
+    capabilities: string[];
     displayName: string;
     deprecated: boolean;
     pricing: {
@@ -63,8 +63,8 @@ declare const ModelRegistrySchema: z.ZodObject<{
         contextWindow: z.ZodNumber;
         releaseDate: z.ZodString;
     }, "strip", z.ZodTypeAny, {
-        capabilities: string[];
         id: string;
+        capabilities: string[];
         displayName: string;
         deprecated: boolean;
         pricing: {
@@ -74,8 +74,8 @@ declare const ModelRegistrySchema: z.ZodObject<{
         contextWindow: number;
         releaseDate: string;
     }, {
-        capabilities: string[];
         id: string;
+        capabilities: string[];
         displayName: string;
         deprecated: boolean;
         pricing: {
@@ -89,8 +89,8 @@ declare const ModelRegistrySchema: z.ZodObject<{
     defaults: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
 }, "strip", z.ZodTypeAny, {
     models: Record<string, Record<string, {
-        capabilities: string[];
         id: string;
+        capabilities: string[];
         displayName: string;
         deprecated: boolean;
         pricing: {
@@ -106,8 +106,8 @@ declare const ModelRegistrySchema: z.ZodObject<{
     aliases?: Record<string, string> | undefined;
 }, {
     models: Record<string, Record<string, {
-        capabilities: string[];
         id: string;
+        capabilities: string[];
         displayName: string;
         deprecated: boolean;
         pricing: {

package/dist/lib/index.d.ts

@@ -18,9 +18,8 @@ export { getBestProvider, getAvailableProviders, isValidProvider, } from "./util
 export { NeuroLink } from "./neurolink.js";
 export type { ProviderStatus, MCPStatus } from "./neurolink.js";
 export type { MCPServerInfo } from "./types/mcpTypes.js";
-export type { NeuroLinkMiddleware, MiddlewareContext, MiddlewareFactoryOptions, } from "./middleware/types.js";
-export { MiddlewareRegistry, MiddlewareFactory } from "./middleware/index.js";
-export { createAnalyticsMiddleware } from "./middleware/builtin/analytics.js";
+export type { NeuroLinkMiddleware, MiddlewareContext, MiddlewareFactoryOptions, MiddlewarePreset, MiddlewareConfig, } from "./middleware/types.js";
+export { MiddlewareFactory } from "./middleware/factory.js";
 export declare const VERSION = "1.0.0";
 /**
  * Quick start factory function

package/dist/lib/index.js

@@ -16,8 +16,7 @@ export { BedrockModels, OpenAIModels, VertexModels, DEFAULT_PROVIDER_CONFIGS, }
 export { getBestProvider, getAvailableProviders, isValidProvider, } from "./utils/providerUtils.js";
 // Main NeuroLink wrapper class and diagnostic types
 export { NeuroLink } from "./neurolink.js";
-export { MiddlewareRegistry, MiddlewareFactory } from "./middleware/index.js";
-export { createAnalyticsMiddleware } from "./middleware/builtin/analytics.js";
+export { MiddlewareFactory } from "./middleware/factory.js";
 // Version
 export const VERSION = "1.0.0";
 /**

package/dist/lib/middleware/builtin/analytics.js

@@ -78,20 +78,19 @@ export function createAnalyticsMiddleware() {
                     streamingMode: true,
                 };
                 requestMetrics.set(requestId, streamAnalytics);
-                // The 'result' is a stream, so we can't directly modify it.
-                // Analytics for streams are typically handled after the stream is consumed.
-                // For this middleware, we'll log the start and rely on other mechanisms
-                // to capture the end-to-end stream metrics if needed.
-                // We will pass a new property in the `rawResponse`
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                const updatedResult = { ...result };
-                if (!updatedResult.rawResponse) {
-                    updatedResult.rawResponse = {};
-                }
-                if (!updatedResult.rawResponse.neurolink) {
-                    updatedResult.rawResponse.neurolink = {};
-                }
-                updatedResult.rawResponse.neurolink.analytics = streamAnalytics;
+                // Ensure the result includes all required properties
+                const updatedResult = {
+                    ...result,
+                    stream: result.stream || new ReadableStream(),
+                    rawCall: result.rawCall || { rawPrompt: null, rawSettings: {} },
+                    rawResponse: {
+                        ...(result.rawResponse || {}),
+                        neurolink: {
+                            ...(result.rawResponse?.neurolink ?? {}),
+                            analytics: streamAnalytics,
+                        },
+                    },
+                };
                 return updatedResult;
             }
             catch (error) {

package/dist/lib/middleware/builtin/guardrails.d.ts

@@ -0,0 +1,20 @@
+import type { LanguageModelV1 } from "ai";
+import type { NeuroLinkMiddleware } from "../types.js";
+/**
+ * Configuration for the Guardrails middleware.
+ */
+export interface GuardrailsMiddlewareConfig {
+    badWords?: {
+        enabled?: boolean;
+        list?: string[];
+    };
+    modelFilter?: {
+        enabled?: boolean;
+        filterModel?: LanguageModelV1;
+    };
+}
+/**
+ * Create Guardrails AI middleware for content filtering and policy enforcement.
+ * @param config - Configuration for the guardrails middleware.
+ */
+export declare function createGuardrailsMiddleware(config?: GuardrailsMiddlewareConfig): NeuroLinkMiddleware;

package/dist/lib/middleware/builtin/guardrails.js

@@ -0,0 +1,87 @@
+import { generateText } from "ai";
+import { logger } from "../../utils/logger.js";
+/**
+ * Create Guardrails AI middleware for content filtering and policy enforcement.
+ * @param config - Configuration for the guardrails middleware.
+ */
+export function createGuardrailsMiddleware(config = {}) {
+    const metadata = {
+        id: "guardrails",
+        name: "Guardrails AI",
+        description: "Provides content filtering and policy enforcement using custom rules and AI models.",
+        priority: 90,
+        defaultEnabled: true,
+    };
+    const middleware = {
+        wrapGenerate: async ({ doGenerate, params }) => {
+            logger.debug(`[GuardrailsMiddleware] Applying to generate call.`, {
+                badWordsEnabled: !!config.badWords?.enabled,
+                modelFilterEnabled: !!config.modelFilter?.enabled,
+            });
+            let result = await doGenerate();
+            // 1. Bad Word Filtering
+            if (config.badWords?.enabled && config.badWords.list) {
+                let filteredText = result.text;
+                for (const term of config.badWords.list) {
+                    const regex = new RegExp(term, "gi");
+                    filteredText = filteredText?.replace(regex, "*".repeat(term.length));
+                }
+                if (result.text !== filteredText) {
+                    logger.debug(`[GuardrailsMiddleware] Redacted bad words from text.`);
+                    result = { ...result, text: filteredText ?? "" };
+                }
+            }
+            // 2. Model-based Filtering
+            if (config.modelFilter?.enabled && config.modelFilter.filterModel) {
+                logger.debug(`[GuardrailsMiddleware] Invoking model-based filter.`);
+                try {
+                    const filterPrompt = `Is the following text safe? Respond with only "safe" or "unsafe".\n\nText: "${result.text}"`;
+                    const { text: filterResponse } = await generateText({
+                        model: config.modelFilter.filterModel, // Cast as any to satisfy type, will be resolved in integration
+                        prompt: filterPrompt,
+                    });
+                    if (filterResponse.toLowerCase().trim() === "unsafe") {
+                        logger.warn(`[GuardrailsMiddleware] Model-based filter flagged content as unsafe.`);
+                        result = { ...result, text: "<REDACTED BY AI GUARDRAIL>" };
+                    }
+                }
+                catch (error) {
+                    logger.error(`[GuardrailsMiddleware] Model-based filter failed.`, {
+                        error,
+                    });
+                }
+            }
+            return result;
+        },
+        wrapStream: async ({ doStream, params }) => {
+            logger.debug(`[GuardrailsMiddleware] Applying to stream call.`, {
+                badWordsEnabled: !!config.badWords?.enabled,
+            });
+            const { stream, ...rest } = await doStream();
+            // Note: Model-based filtering is not applied to streams in this version
+            // as it requires the full text for analysis.
+            const transformStream = new TransformStream({
+                transform(chunk, controller) {
+                    let filteredChunk = chunk;
+                    if (config.badWords?.enabled && config.badWords.list) {
+                        for (const term of config.badWords.list) {
+                            const regex = new RegExp(term, "gi");
+                            if (typeof filteredChunk === "string") {
+                                filteredChunk = filteredChunk.replace(regex, "*".repeat(term.length));
+                            }
+                        }
+                    }
+                    controller.enqueue(filteredChunk);
+                },
+            });
+            return {
+                stream: stream.pipeThrough(transformStream),
+                ...rest,
+            };
+        },
+    };
+    return {
+        ...middleware,
+        metadata,
+    };
+}

package/dist/lib/middleware/factory.d.ts

@@ -1,36 +1,51 @@
 import type { LanguageModelV1 } from "ai";
-import type { MiddlewareContext, MiddlewareConfig, MiddlewareFactoryOptions, MiddlewareChainStats } from "./types.js";
+import type { MiddlewareContext, MiddlewareConfig, MiddlewareFactoryOptions, MiddlewareChainStats, MiddlewarePreset, NeuroLinkMiddleware, MiddlewareRegistrationOptions } from "./types.js";
+import { MiddlewareRegistry } from "./registry.js";
 /**
- * Middleware factory for creating and applying middleware chains
+ * Middleware factory for creating and applying middleware chains.
+ * Each factory instance manages its own registry and configuration.
  */
 export declare class MiddlewareFactory {
+    registry: MiddlewareRegistry;
+    presets: Map<string, MiddlewarePreset>;
+    private options;
+    constructor(options?: MiddlewareFactoryOptions);
+    /**
+     * Initialize the factory with built-in middleware and presets
+     */
+    private initialize;
+    /**
+     * Register a custom preset
+     */
+    registerPreset(preset: MiddlewarePreset, replace?: boolean): void;
+    /**
+     * Register a custom middleware
+     */
+    register(middleware: NeuroLinkMiddleware, options?: MiddlewareRegistrationOptions): void;
     /**
      * Apply middleware to a language model
      */
-    static applyMiddleware(model: LanguageModelV1, context: MiddlewareContext, options?: MiddlewareFactoryOptions): LanguageModelV1;
+    applyMiddleware(model: LanguageModelV1, context: MiddlewareContext, options?: MiddlewareFactoryOptions): LanguageModelV1;
+    private getCreator;
     /**
      * Build middleware configuration from factory options
      */
-    private static buildMiddlewareConfig;
+    private buildMiddlewareConfig;
     /**
      * Get preset configuration
      */
-    private static getPresetConfig;
-    /**
-     * Get built-in preset configurations
-     */
-    private static getBuiltInPresets;
+    private getPresetConfig;
     /**
      * Create middleware context from provider and options
      */
-    static createContext(provider: string, model: string, options?: Record<string, unknown>, session?: {
+    createContext(provider: string, model: string, options?: Record<string, unknown>, session?: {
         sessionId?: string;
         userId?: string;
     }): MiddlewareContext;
     /**
      * Validate middleware configuration
      */
-    static validateConfig(config: Record<string, MiddlewareConfig>): {
+    validateConfig(config: Record<string, MiddlewareConfig>): {
         isValid: boolean;
         errors: string[];
         warnings: string[];
@@ -38,7 +53,7 @@ export declare class MiddlewareFactory {
     /**
      * Get available presets
      */
-    static getAvailablePresets(): Array<{
+    getAvailablePresets(): Array<{
         name: string;
         description: string;
         middleware: string[];
@@ -46,9 +61,9 @@ export declare class MiddlewareFactory {
     /**
      * Get middleware chain statistics
      */
-    static getChainStats(context: MiddlewareContext, config: Record<string, MiddlewareConfig>): MiddlewareChainStats;
+    getChainStats(context: MiddlewareContext, config: Record<string, MiddlewareConfig>): MiddlewareChainStats;
     /**
      * Create a middleware-enabled model factory function
      */
-    static createModelFactory(baseModelFactory: () => Promise<LanguageModelV1>, defaultOptions?: MiddlewareFactoryOptions): (context: MiddlewareContext, options?: MiddlewareFactoryOptions) => Promise<LanguageModelV1>;
+    createModelFactory(baseModelFactory: () => Promise<LanguageModelV1>, defaultOptions?: MiddlewareFactoryOptions): (context: MiddlewareContext, options?: MiddlewareFactoryOptions) => Promise<LanguageModelV1>;
 }