@juspay/neurolink 7.12.0 → 7.13.0

package/CHANGELOG.md

@@ -1,3 +1,9 @@
+## [7.13.0](https://github.com/juspay/neurolink/compare/v7.12.0...v7.13.0) (2025-08-14)
+
+### Features
+
+- **(SDK):** Add context summarizer for conversation BZ-43204 ([38231c4](https://github.com/juspay/neurolink/commit/38231c475b7546c16db741010173794251a7dbaa))
+
 ## [7.12.0](https://github.com/juspay/neurolink/compare/v7.11.1...v7.12.0) (2025-08-14)
 
 ### Features

package/README.md

@@ -331,6 +331,7 @@ console.log(productData.name, productData.price, productData.features);
 - 🔍 **Smart Model Resolution** - Fuzzy matching, aliases, and capability-based search across all providers
 - 🏠 **Local AI Support** - Run completely offline with Ollama or through LiteLLM proxy
 - 🌍 **Universal Model Access** - Direct providers + 100,000+ models via Hugging Face + 100+ models via LiteLLM
+- 🧠 **Automatic Context Summarization** - Stateful, long-running conversations with automatic history summarization.
 - 📊 **Analytics & Evaluation** - Built-in usage tracking and AI-powered quality assessment
 
 ## 🛠️ MCP Integration Status ✅ **BUILT-IN TOOLS WORKING**

package/dist/context/ContextManager.d.ts

@@ -0,0 +1,28 @@
+import type { TextGenerationOptions, TextGenerationResult } from "../core/types.js";
+import type { ContextManagerConfig } from "./types.js";
+type InternalGenerator = (options: TextGenerationOptions) => Promise<TextGenerationResult>;
+/**
+ * Manages conversation context, automatically summarizing it when it
+ * exceeds a specified word count limit.
+ */
+export declare class ContextManager {
+    private static readonly SUMMARIZATION_FAILED_WARNING;
+    private static readonly SUMMARIZATION_EMPTY_WARNING;
+    private history;
+    private wordCount;
+    private readonly internalGenerator;
+    private readonly config;
+    constructor(generatorFunction: InternalGenerator, config: ContextManagerConfig, initialContext?: string);
+    addTurn(role: "user" | "assistant", message: string): Promise<void>;
+    /**
+     * Formats the history including the latest user turn for the prompt, without modifying the permanent history.
+     */
+    getContextForPrompt(role: "user", message: string): string;
+    getCurrentContext(): string;
+    private _summarize;
+    /**
+     * Truncates the history to a specific word count, preserving the most recent messages.
+     */
+    private _truncateHistory;
+}
+export {};

package/dist/context/ContextManager.js

@@ -0,0 +1,102 @@
+import { logger } from "../utils/logger.js";
+import { formatHistoryToString } from "./utils.js";
+/**
+ * Manages conversation context, automatically summarizing it when it
+ * exceeds a specified word count limit.
+ */
+export class ContextManager {
+    static SUMMARIZATION_FAILED_WARNING = "[System Warning: Context summarization failed. Conversation history has been truncated.]";
+    static SUMMARIZATION_EMPTY_WARNING = "[System Warning: Context summarization failed to return valid content. Conversation history has been truncated.]";
+    history;
+    wordCount;
+    internalGenerator;
+    config;
+    constructor(generatorFunction, config, initialContext = "This is the start of the conversation.") {
+        this.internalGenerator = generatorFunction;
+        this.config = config;
+        const initialMessage = { role: "system", content: initialContext };
+        initialMessage.wordCount = this.config.estimateWordCount([initialMessage]);
+        this.history = [initialMessage];
+        this.wordCount = initialMessage.wordCount;
+    }
+    async addTurn(role, message) {
+        const newMessage = { role, content: message };
+        newMessage.wordCount = this.config.estimateWordCount([newMessage]);
+        this.history.push(newMessage);
+        this.wordCount += newMessage.wordCount;
+        logger.info(`[ContextManager] Current word count: ${this.wordCount} / ${this.config.highWaterMarkWords}`);
+        if (this.wordCount > this.config.highWaterMarkWords) {
+            await this._summarize();
+        }
+    }
+    /**
+     * Formats the history including the latest user turn for the prompt, without modifying the permanent history.
+     */
+    getContextForPrompt(role, message) {
+        const tempHistory = [...this.history, { role, content: message }];
+        return formatHistoryToString(tempHistory);
+    }
+    getCurrentContext() {
+        // Format the history into a single string for the provider prompt
+        return formatHistoryToString(this.history);
+    }
+    async _summarize() {
+        try {
+            const prompt = this.config.getSummarizationPrompt(this.history, this.config.lowWaterMarkWords);
+            // Construct options for the internal method, bypassing the main 'generate' entry point
+            const textOptions = {
+                prompt,
+                provider: this.config.summarizationProvider,
+                model: this.config.summarizationModel,
+                // Ensure summarization does not trigger more context management or tools
+                disableTools: true,
+            };
+            // Call the internal generation function directly to avoid recursion
+            const result = await this.internalGenerator(textOptions);
+            if (typeof result.content === "string" && result.content.length > 0) {
+                // Replace the history with a single system message containing the summary
+                const newHistory = [{ role: "system", content: result.content }];
+                this.history = newHistory;
+                this.wordCount = this.config.estimateWordCount(this.history);
+                logger.info(`[ContextManager] Summarization complete. New history length: ${this.wordCount} words.`);
+            }
+            else {
+                logger.warn("[ContextManager] Summarization returned empty or non-string content; truncating history as a fallback.");
+                this._truncateHistory(this.config.lowWaterMarkWords);
+                this.history.unshift({ role: "system", content: ContextManager.SUMMARIZATION_EMPTY_WARNING });
+                this.wordCount = this.config.estimateWordCount(this.history);
+            }
+            logger.debug(`[ContextManager] New history: ${JSON.stringify(this.history)}`);
+        }
+        catch (error) {
+            logger.error("Context summarization failed:", { error });
+            // Fallback strategy: truncate the history to the target word count.
+            this._truncateHistory(this.config.lowWaterMarkWords);
+            this.history.unshift({ role: "system", content: ContextManager.SUMMARIZATION_FAILED_WARNING });
+            this.wordCount = this.config.estimateWordCount(this.history);
+        }
+    }
+    /**
+     * Truncates the history to a specific word count, preserving the most recent messages.
+     */
+    _truncateHistory(wordLimit) {
+        if (this.wordCount <= wordLimit) {
+            return;
+        }
+        let runningCount = 0;
+        let sliceIndex = this.history.length;
+        for (let i = this.history.length - 1; i >= 0; i--) {
+            let wordCount = this.history[i].wordCount;
+            if (wordCount === undefined) {
+                logger.warn(`[ContextManager] Word count cache missing for message at index ${i}. Recalculating.`);
+                wordCount = this.config.estimateWordCount([this.history[i]]);
+            }
+            runningCount += wordCount;
+            if (runningCount > wordLimit) {
+                sliceIndex = i + 1;
+                break;
+            }
+        }
+        this.history = this.history.slice(sliceIndex);
+    }
+}

package/dist/context/config.d.ts

@@ -0,0 +1,5 @@
+import type { ContextManagerConfig } from "./types.js";
+/**
+ * Default configuration for the ContextManager.
+ */
+export declare const defaultContextConfig: ContextManagerConfig;

package/dist/context/config.js

@@ -0,0 +1,38 @@
+import { formatHistoryToString } from "./utils.js";
+/**
+ * Estimates the word count of a conversation history.
+ */
+function estimateWordCount(history) {
+    if (!history || history.length === 0) {
+        return 0;
+    }
+    return history.reduce((acc, msg) => acc + (msg.content.trim().split(/\s+/).filter(word => word.length > 0).length || 0), 0);
+}
+/**
+ * Generates the default prompt for summarization.
+ */
+function getDefaultSummarizationPrompt(history, wordLimit) {
+    const formattedHistory = formatHistoryToString(history);
+    return `
+You are a context summarization AI. Your task is to condense the following conversation history for another AI assistant.
+The summary must be a concise, third-person narrative that retains all critical information. Pay special attention to retaining key entities, technical details, decisions made, and any specific dates or times mentioned.
+Ensure the summary flows logically and is ready to be used as context for the next turn in the conversation.
+Please keep the summary under ${wordLimit} words.
+
+Conversation History to Summarize:
+---
+${formattedHistory}
+---
+`.trim();
+}
+/**
+ * Default configuration for the ContextManager.
+ */
+export const defaultContextConfig = {
+    highWaterMarkWords: 3000,
+    lowWaterMarkWords: 800,
+    summarizationModel: "gemini-2.5-flash",
+    summarizationProvider: "googlevertex",
+    getSummarizationPrompt: getDefaultSummarizationPrompt,
+    estimateWordCount: estimateWordCount,
+};

package/dist/context/types.d.ts

@@ -0,0 +1,20 @@
+export interface ChatMessage {
+    /** Role of the message sender */
+    role: "user" | "assistant" | "system";
+    /** Content of the message */
+    content: string;
+    /** Cached word count for performance */
+    wordCount?: number;
+}
+/**
+ * Defines the configuration for the ContextManager.
+ * This allows for easy customization of the summarization behavior.
+ */
+export interface ContextManagerConfig {
+    highWaterMarkWords: number;
+    lowWaterMarkWords: number;
+    summarizationModel: string;
+    summarizationProvider: string;
+    getSummarizationPrompt: (history: ChatMessage[], wordLimit: number) => string;
+    estimateWordCount: (history: ChatMessage[]) => number;
+}

package/dist/context/types.js

@@ -0,0 +1 @@
+export {};

package/dist/context/utils.d.ts

@@ -0,0 +1,7 @@
+import type { ChatMessage } from "./types.js";
+/**
+ * Formats a chat history array into a single string for use in a prompt.
+ * @param history The array of ChatMessage objects.
+ * @returns A formatted string representing the conversation.
+ */
+export declare function formatHistoryToString(history: ChatMessage[]): string;

package/dist/context/utils.js

@@ -0,0 +1,8 @@
+/**
+ * Formats a chat history array into a single string for use in a prompt.
+ * @param history The array of ChatMessage objects.
+ * @returns A formatted string representing the conversation.
+ */
+export function formatHistoryToString(history) {
+    return history.map(msg => `${msg.role}: ${msg.content}`).join('\n\n');
+}

package/dist/lib/context/ContextManager.d.ts

@@ -0,0 +1,28 @@
+import type { TextGenerationOptions, TextGenerationResult } from "../core/types.js";
+import type { ContextManagerConfig } from "./types.js";
+type InternalGenerator = (options: TextGenerationOptions) => Promise<TextGenerationResult>;
+/**
+ * Manages conversation context, automatically summarizing it when it
+ * exceeds a specified word count limit.
+ */
+export declare class ContextManager {
+    private static readonly SUMMARIZATION_FAILED_WARNING;
+    private static readonly SUMMARIZATION_EMPTY_WARNING;
+    private history;
+    private wordCount;
+    private readonly internalGenerator;
+    private readonly config;
+    constructor(generatorFunction: InternalGenerator, config: ContextManagerConfig, initialContext?: string);
+    addTurn(role: "user" | "assistant", message: string): Promise<void>;
+    /**
+     * Formats the history including the latest user turn for the prompt, without modifying the permanent history.
+     */
+    getContextForPrompt(role: "user", message: string): string;
+    getCurrentContext(): string;
+    private _summarize;
+    /**
+     * Truncates the history to a specific word count, preserving the most recent messages.
+     */
+    private _truncateHistory;
+}
+export {};

package/dist/lib/context/ContextManager.js

@@ -0,0 +1,102 @@
+import { logger } from "../utils/logger.js";
+import { formatHistoryToString } from "./utils.js";
+/**
+ * Manages conversation context, automatically summarizing it when it
+ * exceeds a specified word count limit.
+ */
+export class ContextManager {
+    static SUMMARIZATION_FAILED_WARNING = "[System Warning: Context summarization failed. Conversation history has been truncated.]";
+    static SUMMARIZATION_EMPTY_WARNING = "[System Warning: Context summarization failed to return valid content. Conversation history has been truncated.]";
+    history;
+    wordCount;
+    internalGenerator;
+    config;
+    constructor(generatorFunction, config, initialContext = "This is the start of the conversation.") {
+        this.internalGenerator = generatorFunction;
+        this.config = config;
+        const initialMessage = { role: "system", content: initialContext };
+        initialMessage.wordCount = this.config.estimateWordCount([initialMessage]);
+        this.history = [initialMessage];
+        this.wordCount = initialMessage.wordCount;
+    }
+    async addTurn(role, message) {
+        const newMessage = { role, content: message };
+        newMessage.wordCount = this.config.estimateWordCount([newMessage]);
+        this.history.push(newMessage);
+        this.wordCount += newMessage.wordCount;
+        logger.info(`[ContextManager] Current word count: ${this.wordCount} / ${this.config.highWaterMarkWords}`);
+        if (this.wordCount > this.config.highWaterMarkWords) {
+            await this._summarize();
+        }
+    }
+    /**
+     * Formats the history including the latest user turn for the prompt, without modifying the permanent history.
+     */
+    getContextForPrompt(role, message) {
+        const tempHistory = [...this.history, { role, content: message }];
+        return formatHistoryToString(tempHistory);
+    }
+    getCurrentContext() {
+        // Format the history into a single string for the provider prompt
+        return formatHistoryToString(this.history);
+    }
+    async _summarize() {
+        try {
+            const prompt = this.config.getSummarizationPrompt(this.history, this.config.lowWaterMarkWords);
+            // Construct options for the internal method, bypassing the main 'generate' entry point
+            const textOptions = {
+                prompt,
+                provider: this.config.summarizationProvider,
+                model: this.config.summarizationModel,
+                // Ensure summarization does not trigger more context management or tools
+                disableTools: true,
+            };
+            // Call the internal generation function directly to avoid recursion
+            const result = await this.internalGenerator(textOptions);
+            if (typeof result.content === "string" && result.content.length > 0) {
+                // Replace the history with a single system message containing the summary
+                const newHistory = [{ role: "system", content: result.content }];
+                this.history = newHistory;
+                this.wordCount = this.config.estimateWordCount(this.history);
+                logger.info(`[ContextManager] Summarization complete. New history length: ${this.wordCount} words.`);
+            }
+            else {
+                logger.warn("[ContextManager] Summarization returned empty or non-string content; truncating history as a fallback.");
+                this._truncateHistory(this.config.lowWaterMarkWords);
+                this.history.unshift({ role: "system", content: ContextManager.SUMMARIZATION_EMPTY_WARNING });
+                this.wordCount = this.config.estimateWordCount(this.history);
+            }
+            logger.debug(`[ContextManager] New history: ${JSON.stringify(this.history)}`);
+        }
+        catch (error) {
+            logger.error("Context summarization failed:", { error });
+            // Fallback strategy: truncate the history to the target word count.
+            this._truncateHistory(this.config.lowWaterMarkWords);
+            this.history.unshift({ role: "system", content: ContextManager.SUMMARIZATION_FAILED_WARNING });
+            this.wordCount = this.config.estimateWordCount(this.history);
+        }
+    }
+    /**
+     * Truncates the history to a specific word count, preserving the most recent messages.
+     */
+    _truncateHistory(wordLimit) {
+        if (this.wordCount <= wordLimit) {
+            return;
+        }
+        let runningCount = 0;
+        let sliceIndex = this.history.length;
+        for (let i = this.history.length - 1; i >= 0; i--) {
+            let wordCount = this.history[i].wordCount;
+            if (wordCount === undefined) {
+                logger.warn(`[ContextManager] Word count cache missing for message at index ${i}. Recalculating.`);
+                wordCount = this.config.estimateWordCount([this.history[i]]);
+            }
+            runningCount += wordCount;
+            if (runningCount > wordLimit) {
+                sliceIndex = i + 1;
+                break;
+            }
+        }
+        this.history = this.history.slice(sliceIndex);
+    }
+}

package/dist/lib/context/config.d.ts

@@ -0,0 +1,5 @@
+import type { ContextManagerConfig } from "./types.js";
+/**
+ * Default configuration for the ContextManager.
+ */
+export declare const defaultContextConfig: ContextManagerConfig;

package/dist/lib/context/config.js

@@ -0,0 +1,38 @@
+import { formatHistoryToString } from "./utils.js";
+/**
+ * Estimates the word count of a conversation history.
+ */
+function estimateWordCount(history) {
+    if (!history || history.length === 0) {
+        return 0;
+    }
+    return history.reduce((acc, msg) => acc + (msg.content.trim().split(/\s+/).filter(word => word.length > 0).length || 0), 0);
+}
+/**
+ * Generates the default prompt for summarization.
+ */
+function getDefaultSummarizationPrompt(history, wordLimit) {
+    const formattedHistory = formatHistoryToString(history);
+    return `
+You are a context summarization AI. Your task is to condense the following conversation history for another AI assistant.
+The summary must be a concise, third-person narrative that retains all critical information. Pay special attention to retaining key entities, technical details, decisions made, and any specific dates or times mentioned.
+Ensure the summary flows logically and is ready to be used as context for the next turn in the conversation.
+Please keep the summary under ${wordLimit} words.
+
+Conversation History to Summarize:
+---
+${formattedHistory}
+---
+`.trim();
+}
+/**
+ * Default configuration for the ContextManager.
+ */
+export const defaultContextConfig = {
+    highWaterMarkWords: 3000,
+    lowWaterMarkWords: 800,
+    summarizationModel: "gemini-2.5-flash",
+    summarizationProvider: "googlevertex",
+    getSummarizationPrompt: getDefaultSummarizationPrompt,
+    estimateWordCount: estimateWordCount,
+};

package/dist/lib/context/types.d.ts

@@ -0,0 +1,20 @@
+export interface ChatMessage {
+    /** Role of the message sender */
+    role: "user" | "assistant" | "system";
+    /** Content of the message */
+    content: string;
+    /** Cached word count for performance */
+    wordCount?: number;
+}
+/**
+ * Defines the configuration for the ContextManager.
+ * This allows for easy customization of the summarization behavior.
+ */
+export interface ContextManagerConfig {
+    highWaterMarkWords: number;
+    lowWaterMarkWords: number;
+    summarizationModel: string;
+    summarizationProvider: string;
+    getSummarizationPrompt: (history: ChatMessage[], wordLimit: number) => string;
+    estimateWordCount: (history: ChatMessage[]) => number;
+}

package/dist/lib/context/types.js

@@ -0,0 +1 @@
+export {};

package/dist/lib/context/utils.d.ts

@@ -0,0 +1,7 @@
+import type { ChatMessage } from "./types.js";
+/**
+ * Formats a chat history array into a single string for use in a prompt.
+ * @param history The array of ChatMessage objects.
+ * @returns A formatted string representing the conversation.
+ */
+export declare function formatHistoryToString(history: ChatMessage[]): string;

package/dist/lib/context/utils.js

@@ -0,0 +1,8 @@
+/**
+ * Formats a chat history array into a single string for use in a prompt.
+ * @param history The array of ChatMessage objects.
+ * @returns A formatted string representing the conversation.
+ */
+export function formatHistoryToString(history) {
+    return history.map(msg => `${msg.role}: ${msg.content}`).join('\n\n');
+}

package/dist/lib/neurolink.d.ts

@@ -41,9 +41,11 @@ export interface MCPServerInfo {
     hasServer: boolean;
     metadata?: unknown;
 }
+import type { ContextManagerConfig } from "./context/types.js";
 export declare class NeuroLink {
     private mcpInitialized;
     private emitter;
+    private contextManager;
     private customTools;
     private inMemoryServers;
     private toolCircuitBreakers;
@@ -69,6 +71,21 @@ export declare class NeuroLink {
      * MAIN ENTRY POINT: Enhanced generate method with new function signature
      * Replaces both generateText and legacy methods
      */
+    /**
+     * Extracts the original prompt text from the provided input.
+     * If a string is provided, it returns the string directly.
+     * If a GenerateOptions object is provided, it returns the input text from the object.
+     * @param optionsOrPrompt The prompt input, either as a string or a GenerateOptions object.
+     * @returns The original prompt text as a string.
+     */
+    private _extractOriginalPrompt;
+    /**
+     * Enables automatic context summarization for the NeuroLink instance.
+     * Once enabled, the instance will maintain conversation history and
+     * automatically summarize it when it exceeds token limits.
+     * @param config Optional configuration to override default summarization settings.
+     */
+    enableContextSummarization(config?: Partial<ContextManagerConfig>): void;
     generate(optionsOrPrompt: GenerateOptions | string): Promise<GenerateResult>;
     /**
      * BACKWARD COMPATIBILITY: Legacy generateText method

package/dist/lib/neurolink.js

@@ -29,10 +29,13 @@ import { ErrorFactory, NeuroLinkError, withTimeout, withRetry, isRetriableError,
 import { EventEmitter } from "events";
 import { ConversationMemoryManager } from "./core/conversationMemoryManager.js";
 import { applyConversationMemoryDefaults, getConversationMessages, storeConversationTurn, } from "./utils/conversationMemoryUtils.js";
+import { ContextManager } from "./context/ContextManager.js";
+import { defaultContextConfig } from "./context/config.js";
 // Core types imported from core/types.js
 export class NeuroLink {
     mcpInitialized = false;
     emitter = new EventEmitter();
+    contextManager = null;
     // Tool registration support
     customTools = new Map();
     inMemoryServers = new Map();
@@ -120,7 +123,33 @@ export class NeuroLink {
      * MAIN ENTRY POINT: Enhanced generate method with new function signature
      * Replaces both generateText and legacy methods
      */
+    /**
+     * Extracts the original prompt text from the provided input.
+     * If a string is provided, it returns the string directly.
+     * If a GenerateOptions object is provided, it returns the input text from the object.
+     * @param optionsOrPrompt The prompt input, either as a string or a GenerateOptions object.
+     * @returns The original prompt text as a string.
+     */
+    _extractOriginalPrompt(optionsOrPrompt) {
+        return typeof optionsOrPrompt === 'string' ? optionsOrPrompt : optionsOrPrompt.input.text;
+    }
+    /**
+     * Enables automatic context summarization for the NeuroLink instance.
+     * Once enabled, the instance will maintain conversation history and
+     * automatically summarize it when it exceeds token limits.
+     * @param config Optional configuration to override default summarization settings.
+     */
+    enableContextSummarization(config) {
+        const contextConfig = {
+            ...defaultContextConfig,
+            ...config,
+        };
+        // Pass the internal generator function directly, bound to the correct `this` context.
+        this.contextManager = new ContextManager(this.generateTextInternal.bind(this), contextConfig);
+        logger.info("[NeuroLink] Automatic context summarization enabled.");
+    }
     async generate(optionsOrPrompt) {
+        const originalPrompt = this._extractOriginalPrompt(optionsOrPrompt);
         // Convert string prompt to full options
         const options = typeof optionsOrPrompt === "string"
             ? { input: { text: optionsOrPrompt } }
@@ -129,6 +158,11 @@ export class NeuroLink {
         if (!options.input?.text || typeof options.input.text !== "string") {
             throw new Error("Input text is required and must be a non-empty string");
         }
+        // Handle Context Management if enabled
+        if (this.contextManager) {
+            // Get the full context for the prompt without permanently adding the user's turn yet
+            options.input.text = this.contextManager.getContextForPrompt("user", options.input.text);
+        }
         const startTime = Date.now();
         // Emit generation start event
         this.emitter.emit("generation:start", {
@@ -240,6 +274,11 @@ export class NeuroLink {
                 }
                 : undefined,
         };
+        // Add both the user's turn and the AI's response to the permanent history
+        if (this.contextManager) {
+            await this.contextManager.addTurn("user", originalPrompt);
+            await this.contextManager.addTurn("assistant", generateResult.content);
+        }
         return generateResult;
     }
     /**

package/dist/neurolink.d.ts

@@ -41,9 +41,11 @@ export interface MCPServerInfo {
     hasServer: boolean;
     metadata?: unknown;
 }
+import type { ContextManagerConfig } from "./context/types.js";
 export declare class NeuroLink {
     private mcpInitialized;
     private emitter;
+    private contextManager;
     private customTools;
     private inMemoryServers;
     private toolCircuitBreakers;
@@ -69,6 +71,21 @@ export declare class NeuroLink {
      * MAIN ENTRY POINT: Enhanced generate method with new function signature
      * Replaces both generateText and legacy methods
      */
+    /**
+     * Extracts the original prompt text from the provided input.
+     * If a string is provided, it returns the string directly.
+     * If a GenerateOptions object is provided, it returns the input text from the object.
+     * @param optionsOrPrompt The prompt input, either as a string or a GenerateOptions object.
+     * @returns The original prompt text as a string.
+     */
+    private _extractOriginalPrompt;
+    /**
+     * Enables automatic context summarization for the NeuroLink instance.
+     * Once enabled, the instance will maintain conversation history and
+     * automatically summarize it when it exceeds token limits.
+     * @param config Optional configuration to override default summarization settings.
+     */
+    enableContextSummarization(config?: Partial<ContextManagerConfig>): void;
     generate(optionsOrPrompt: GenerateOptions | string): Promise<GenerateResult>;
     /**
      * BACKWARD COMPATIBILITY: Legacy generateText method

package/dist/neurolink.js

@@ -29,10 +29,13 @@ import { ErrorFactory, NeuroLinkError, withTimeout, withRetry, isRetriableError,
 import { EventEmitter } from "events";
 import { ConversationMemoryManager } from "./core/conversationMemoryManager.js";
 import { applyConversationMemoryDefaults, getConversationMessages, storeConversationTurn, } from "./utils/conversationMemoryUtils.js";
+import { ContextManager } from "./context/ContextManager.js";
+import { defaultContextConfig } from "./context/config.js";
 // Core types imported from core/types.js
 export class NeuroLink {
     mcpInitialized = false;
     emitter = new EventEmitter();
+    contextManager = null;
     // Tool registration support
     customTools = new Map();
     inMemoryServers = new Map();
@@ -120,7 +123,33 @@ export class NeuroLink {
      * MAIN ENTRY POINT: Enhanced generate method with new function signature
      * Replaces both generateText and legacy methods
      */
+    /**
+     * Extracts the original prompt text from the provided input.
+     * If a string is provided, it returns the string directly.
+     * If a GenerateOptions object is provided, it returns the input text from the object.
+     * @param optionsOrPrompt The prompt input, either as a string or a GenerateOptions object.
+     * @returns The original prompt text as a string.
+     */
+    _extractOriginalPrompt(optionsOrPrompt) {
+        return typeof optionsOrPrompt === 'string' ? optionsOrPrompt : optionsOrPrompt.input.text;
+    }
+    /**
+     * Enables automatic context summarization for the NeuroLink instance.
+     * Once enabled, the instance will maintain conversation history and
+     * automatically summarize it when it exceeds token limits.
+     * @param config Optional configuration to override default summarization settings.
+     */
+    enableContextSummarization(config) {
+        const contextConfig = {
+            ...defaultContextConfig,
+            ...config,
+        };
+        // Pass the internal generator function directly, bound to the correct `this` context.
+        this.contextManager = new ContextManager(this.generateTextInternal.bind(this), contextConfig);
+        logger.info("[NeuroLink] Automatic context summarization enabled.");
+    }
     async generate(optionsOrPrompt) {
+        const originalPrompt = this._extractOriginalPrompt(optionsOrPrompt);
         // Convert string prompt to full options
         const options = typeof optionsOrPrompt === "string"
             ? { input: { text: optionsOrPrompt } }
@@ -129,6 +158,11 @@ export class NeuroLink {
         if (!options.input?.text || typeof options.input.text !== "string") {
             throw new Error("Input text is required and must be a non-empty string");
         }
+        // Handle Context Management if enabled
+        if (this.contextManager) {
+            // Get the full context for the prompt without permanently adding the user's turn yet
+            options.input.text = this.contextManager.getContextForPrompt("user", options.input.text);
+        }
         const startTime = Date.now();
         // Emit generation start event
         this.emitter.emit("generation:start", {
@@ -240,6 +274,11 @@ export class NeuroLink {
                 }
                 : undefined,
         };
+        // Add both the user's turn and the AI's response to the permanent history
+        if (this.contextManager) {
+            await this.contextManager.addTurn("user", originalPrompt);
+            await this.contextManager.addTurn("assistant", generateResult.content);
+        }
         return generateResult;
     }
     /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@juspay/neurolink",
-  "version": "7.12.0",
+  "version": "7.13.0",
   "description": "Universal AI Development Platform with working MCP integration, multi-provider support, and professional CLI. Built-in tools operational, 58+ external MCP servers discoverable. Connect to filesystem, GitHub, database operations, and more. Build, test, and deploy AI applications with 9 major providers: OpenAI, Anthropic, Google AI, AWS Bedrock, Azure, Hugging Face, Ollama, and Mistral AI.",
   "author": {
     "name": "Juspay Technologies",