@juspay/neurolink 7.7.1 → 7.9.0

package/dist/cli/factories/commandFactory.js

@@ -28,6 +28,7 @@ export class CLICommandFactory {
                 "ollama",
                 "mistral",
                 "litellm",
+                "sagemaker",
             ],
             default: "auto",
             description: "AI provider to use (auto-selects best available)",
@@ -287,8 +288,12 @@ export class CLICommandFactory {
         let analyticsText = "\n\n📊 Analytics:\n";
         // Provider and model info
         analyticsText += `   Provider: ${analytics.provider}`;
-        if (result.model) {
-            analyticsText += ` (${result.model})`;
+        // Check for model in multiple locations: result.model, analytics.model, or any model property
+        const modelName = result.model ||
+            analytics.model ||
+            analytics.modelName;
+        if (modelName) {
+            analyticsText += ` (${modelName})`;
         }
         analyticsText += "\n";
         // Token usage

package/dist/cli/index.js

@@ -9,6 +9,7 @@ import yargs from "yargs";
 import { hideBin } from "yargs/helpers";
 import chalk from "chalk";
 import { addOllamaCommands } from "./commands/ollama.js";
+import { addSageMakerCommands } from "./commands/sagemaker.js";
 import { CLICommandFactory } from "./factories/commandFactory.js";
 import { logger } from "../lib/utils/logger.js";
 // Load environment variables from .env file
@@ -228,6 +229,8 @@ const cli = yargs(args)
     .command(CLICommandFactory.createCompletionCommand());
 // Add Ollama Commands
 addOllamaCommands(cli);
+// Add SageMaker Commands
+addSageMakerCommands(cli);
 // Execute CLI
 (async () => {
     try {

package/dist/cli/utils/interactiveSetup.js

@@ -176,6 +176,34 @@ export const PROVIDER_CONFIGS = [
             },
         ],
     },
+    {
+        id: AIProviderName.SAGEMAKER,
+        name: "Amazon SageMaker",
+        description: "Custom models deployed on Amazon SageMaker endpoints",
+        envVars: [
+            {
+                key: "AWS_ACCESS_KEY_ID",
+                prompt: "AWS Access Key ID",
+                secure: true,
+            },
+            {
+                key: "AWS_SECRET_ACCESS_KEY",
+                prompt: "AWS Secret Access Key",
+                secure: true,
+            },
+            {
+                key: "AWS_REGION",
+                prompt: "AWS Region",
+                default: "us-east-1",
+                secure: false,
+            },
+            {
+                key: "SAGEMAKER_DEFAULT_ENDPOINT",
+                prompt: "SageMaker Endpoint Name",
+                secure: false,
+            },
+        ],
+    },
 ];
 /**
  * Run the interactive setup wizard

package/dist/core/baseProvider.d.ts

@@ -1,6 +1,6 @@
 import type { ZodType, ZodTypeDef } from "zod";
 import type { Schema } from "ai";
-import type { Tool, LanguageModel } from "ai";
+import type { Tool, LanguageModelV1 } from "ai";
 import type { AIProvider, TextGenerationOptions, EnhancedGenerateResult, AnalyticsData, AIProviderName, EvaluationData } from "../core/types.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
 import type { JsonValue, UnknownRecord } from "../types/common.js";
@@ -419,7 +419,7 @@ export declare abstract class BaseProvider implements AIProvider {
      * REQUIRED: Every provider MUST implement this method
      * Returns the Vercel AI SDK model instance for this provider
      */
-    protected abstract getAISDKModel(): LanguageModel | Promise<LanguageModel>;
+    protected abstract getAISDKModel(): LanguageModelV1 | Promise<LanguageModelV1>;
     /**
      * Get all available tools - direct tools are ALWAYS available
      * MCP tools are added when available (without blocking)

package/dist/core/types.d.ts

@@ -51,6 +51,7 @@ export declare enum AIProviderName {
     OLLAMA = "ollama",
     MISTRAL = "mistral",
     LITELLM = "litellm",
+    SAGEMAKER = "sagemaker",
     AUTO = "auto"
 }
 /**
@@ -77,19 +78,30 @@ export declare enum OpenAIModels {
  */
 export declare enum VertexModels {
     CLAUDE_4_0_SONNET = "claude-sonnet-4@20250514",
-    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    CLAUDE_4_0_OPUS = "claude-opus-4@20250514",
+    CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20241022",
+    CLAUDE_3_5_HAIKU = "claude-3-5-haiku-20241022",
+    CLAUDE_3_SONNET = "claude-3-sonnet-20240229",
+    CLAUDE_3_OPUS = "claude-3-opus-20240229",
+    CLAUDE_3_HAIKU = "claude-3-haiku-20240307",
     GEMINI_2_5_PRO = "gemini-2.5-pro",
     GEMINI_2_5_FLASH = "gemini-2.5-flash",
-    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite",
+    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    GEMINI_1_5_PRO = "gemini-1.5-pro",
+    GEMINI_1_5_FLASH = "gemini-1.5-flash"
 }
 /**
  * Supported Models for Google AI Studio
  */
 export declare enum GoogleAIModels {
-    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
     GEMINI_2_5_PRO = "gemini-2.5-pro",
     GEMINI_2_5_FLASH = "gemini-2.5-flash",
-    GEMINI_1_5_FLASH_LITE = "gemini-2.5-flash-lite"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite",
+    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    GEMINI_1_5_PRO = "gemini-1.5-pro",
+    GEMINI_1_5_FLASH = "gemini-1.5-flash",
+    GEMINI_1_5_FLASH_LITE = "gemini-1.5-flash-lite"
 }
 /**
  * Union type of all supported model names

package/dist/core/types.js

@@ -14,6 +14,7 @@ export var AIProviderName;
     AIProviderName["OLLAMA"] = "ollama";
     AIProviderName["MISTRAL"] = "mistral";
     AIProviderName["LITELLM"] = "litellm";
+    AIProviderName["SAGEMAKER"] = "sagemaker";
     AIProviderName["AUTO"] = "auto";
 })(AIProviderName || (AIProviderName = {}));
 /**
@@ -42,21 +43,41 @@ export var OpenAIModels;
  */
 export var VertexModels;
 (function (VertexModels) {
+    // Claude 4 Series (Latest - May 2025)
     VertexModels["CLAUDE_4_0_SONNET"] = "claude-sonnet-4@20250514";
-    VertexModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    VertexModels["CLAUDE_4_0_OPUS"] = "claude-opus-4@20250514";
+    // Claude 3.5 Series (Still supported)
+    VertexModels["CLAUDE_3_5_SONNET"] = "claude-3-5-sonnet-20241022";
+    VertexModels["CLAUDE_3_5_HAIKU"] = "claude-3-5-haiku-20241022";
+    // Claude 3 Series (Legacy support)
+    VertexModels["CLAUDE_3_SONNET"] = "claude-3-sonnet-20240229";
+    VertexModels["CLAUDE_3_OPUS"] = "claude-3-opus-20240229";
+    VertexModels["CLAUDE_3_HAIKU"] = "claude-3-haiku-20240307";
+    // Gemini 2.5 Series (Latest - 2025)
     VertexModels["GEMINI_2_5_PRO"] = "gemini-2.5-pro";
     VertexModels["GEMINI_2_5_FLASH"] = "gemini-2.5-flash";
     VertexModels["GEMINI_2_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    // Gemini 2.0 Series
+    VertexModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 1.5 Series (Legacy support)
+    VertexModels["GEMINI_1_5_PRO"] = "gemini-1.5-pro";
+    VertexModels["GEMINI_1_5_FLASH"] = "gemini-1.5-flash";
 })(VertexModels || (VertexModels = {}));
 /**
  * Supported Models for Google AI Studio
  */
 export var GoogleAIModels;
 (function (GoogleAIModels) {
-    GoogleAIModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 2.5 Series (Latest - 2025)
     GoogleAIModels["GEMINI_2_5_PRO"] = "gemini-2.5-pro";
     GoogleAIModels["GEMINI_2_5_FLASH"] = "gemini-2.5-flash";
-    GoogleAIModels["GEMINI_1_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    GoogleAIModels["GEMINI_2_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    // Gemini 2.0 Series
+    GoogleAIModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 1.5 Series (Legacy support)
+    GoogleAIModels["GEMINI_1_5_PRO"] = "gemini-1.5-pro";
+    GoogleAIModels["GEMINI_1_5_FLASH"] = "gemini-1.5-flash";
+    GoogleAIModels["GEMINI_1_5_FLASH_LITE"] = "gemini-1.5-flash-lite";
 })(GoogleAIModels || (GoogleAIModels = {}));
 /**
  * Default provider configurations

package/dist/factories/providerFactory.js

@@ -35,7 +35,16 @@ export class ProviderFactory {
         if (!registration) {
             throw new Error(`Unknown provider: ${providerName}. Available providers: ${this.getAvailableProviders().join(", ")}`);
         }
-        const model = modelName || registration.defaultModel;
+        // Respect environment variables before falling back to registry default
+        let model = modelName;
+        if (!model) {
+            // Check for provider-specific environment variables
+            if (providerName.toLowerCase().includes("vertex")) {
+                model = process.env.VERTEX_MODEL;
+            }
+            // Fallback to registry default if no env var
+            model = model || registration.defaultModel;
+        }
         try {
             // Try calling as factory function first, then fallback to constructor
             let result;

package/dist/factories/providerRegistry.js

@@ -55,7 +55,7 @@ export class ProviderRegistry {
             ProviderFactory.registerProvider(AIProviderName.VERTEX, async (modelName) => {
                 const { GoogleVertexProvider } = await import("../providers/googleVertex.js");
                 return new GoogleVertexProvider(modelName);
-            }, "gemini-2.5-flash", ["vertex", "googleVertex"]);
+            }, "claude-sonnet-4@20250514", ["vertex", "googleVertex"]);
             // Register Hugging Face provider (Unified Router implementation)
             ProviderFactory.registerProvider(AIProviderName.HUGGINGFACE, async (modelName) => {
                 const { HuggingFaceProvider } = await import("../providers/huggingFace.js");
@@ -82,6 +82,11 @@ export class ProviderRegistry {
                 return new OpenAICompatibleProvider(modelName, sdk);
             }, process.env.OPENAI_COMPATIBLE_MODEL || undefined, // Enable auto-discovery when no model specified
             ["openai-compatible", "openrouter", "vllm", "compatible"]);
+            // Register Amazon SageMaker provider
+            ProviderFactory.registerProvider(AIProviderName.SAGEMAKER, async (modelName, providerName, sdk) => {
+                const { AmazonSageMakerProvider } = await import("../providers/amazonSagemaker.js");
+                return new AmazonSageMakerProvider(modelName);
+            }, process.env.SAGEMAKER_MODEL || "sagemaker-model", ["sagemaker", "aws-sagemaker"]);
             logger.debug("All providers registered successfully");
             this.registered = true;
         }

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

@@ -1,6 +1,6 @@
 import type { ZodType, ZodTypeDef } from "zod";
 import type { Schema } from "ai";
-import type { Tool, LanguageModel } from "ai";
+import type { Tool, LanguageModelV1 } from "ai";
 import type { AIProvider, TextGenerationOptions, EnhancedGenerateResult, AnalyticsData, AIProviderName, EvaluationData } from "../core/types.js";
 import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
 import type { JsonValue, UnknownRecord } from "../types/common.js";
@@ -419,7 +419,7 @@ export declare abstract class BaseProvider implements AIProvider {
      * REQUIRED: Every provider MUST implement this method
      * Returns the Vercel AI SDK model instance for this provider
      */
-    protected abstract getAISDKModel(): LanguageModel | Promise<LanguageModel>;
+    protected abstract getAISDKModel(): LanguageModelV1 | Promise<LanguageModelV1>;
     /**
      * Get all available tools - direct tools are ALWAYS available
      * MCP tools are added when available (without blocking)

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

@@ -51,6 +51,7 @@ export declare enum AIProviderName {
     OLLAMA = "ollama",
     MISTRAL = "mistral",
     LITELLM = "litellm",
+    SAGEMAKER = "sagemaker",
     AUTO = "auto"
 }
 /**
@@ -77,19 +78,30 @@ export declare enum OpenAIModels {
  */
 export declare enum VertexModels {
     CLAUDE_4_0_SONNET = "claude-sonnet-4@20250514",
-    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    CLAUDE_4_0_OPUS = "claude-opus-4@20250514",
+    CLAUDE_3_5_SONNET = "claude-3-5-sonnet-20241022",
+    CLAUDE_3_5_HAIKU = "claude-3-5-haiku-20241022",
+    CLAUDE_3_SONNET = "claude-3-sonnet-20240229",
+    CLAUDE_3_OPUS = "claude-3-opus-20240229",
+    CLAUDE_3_HAIKU = "claude-3-haiku-20240307",
     GEMINI_2_5_PRO = "gemini-2.5-pro",
     GEMINI_2_5_FLASH = "gemini-2.5-flash",
-    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite",
+    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    GEMINI_1_5_PRO = "gemini-1.5-pro",
+    GEMINI_1_5_FLASH = "gemini-1.5-flash"
 }
 /**
  * Supported Models for Google AI Studio
  */
 export declare enum GoogleAIModels {
-    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
     GEMINI_2_5_PRO = "gemini-2.5-pro",
     GEMINI_2_5_FLASH = "gemini-2.5-flash",
-    GEMINI_1_5_FLASH_LITE = "gemini-2.5-flash-lite"
+    GEMINI_2_5_FLASH_LITE = "gemini-2.5-flash-lite",
+    GEMINI_2_0_FLASH_001 = "gemini-2.0-flash-001",
+    GEMINI_1_5_PRO = "gemini-1.5-pro",
+    GEMINI_1_5_FLASH = "gemini-1.5-flash",
+    GEMINI_1_5_FLASH_LITE = "gemini-1.5-flash-lite"
 }
 /**
  * Union type of all supported model names

package/dist/lib/core/types.js

@@ -14,6 +14,7 @@ export var AIProviderName;
     AIProviderName["OLLAMA"] = "ollama";
     AIProviderName["MISTRAL"] = "mistral";
     AIProviderName["LITELLM"] = "litellm";
+    AIProviderName["SAGEMAKER"] = "sagemaker";
     AIProviderName["AUTO"] = "auto";
 })(AIProviderName || (AIProviderName = {}));
 /**
@@ -42,21 +43,41 @@ export var OpenAIModels;
  */
 export var VertexModels;
 (function (VertexModels) {
+    // Claude 4 Series (Latest - May 2025)
     VertexModels["CLAUDE_4_0_SONNET"] = "claude-sonnet-4@20250514";
-    VertexModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    VertexModels["CLAUDE_4_0_OPUS"] = "claude-opus-4@20250514";
+    // Claude 3.5 Series (Still supported)
+    VertexModels["CLAUDE_3_5_SONNET"] = "claude-3-5-sonnet-20241022";
+    VertexModels["CLAUDE_3_5_HAIKU"] = "claude-3-5-haiku-20241022";
+    // Claude 3 Series (Legacy support)
+    VertexModels["CLAUDE_3_SONNET"] = "claude-3-sonnet-20240229";
+    VertexModels["CLAUDE_3_OPUS"] = "claude-3-opus-20240229";
+    VertexModels["CLAUDE_3_HAIKU"] = "claude-3-haiku-20240307";
+    // Gemini 2.5 Series (Latest - 2025)
     VertexModels["GEMINI_2_5_PRO"] = "gemini-2.5-pro";
     VertexModels["GEMINI_2_5_FLASH"] = "gemini-2.5-flash";
     VertexModels["GEMINI_2_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    // Gemini 2.0 Series
+    VertexModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 1.5 Series (Legacy support)
+    VertexModels["GEMINI_1_5_PRO"] = "gemini-1.5-pro";
+    VertexModels["GEMINI_1_5_FLASH"] = "gemini-1.5-flash";
 })(VertexModels || (VertexModels = {}));
 /**
  * Supported Models for Google AI Studio
  */
 export var GoogleAIModels;
 (function (GoogleAIModels) {
-    GoogleAIModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 2.5 Series (Latest - 2025)
     GoogleAIModels["GEMINI_2_5_PRO"] = "gemini-2.5-pro";
     GoogleAIModels["GEMINI_2_5_FLASH"] = "gemini-2.5-flash";
-    GoogleAIModels["GEMINI_1_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    GoogleAIModels["GEMINI_2_5_FLASH_LITE"] = "gemini-2.5-flash-lite";
+    // Gemini 2.0 Series
+    GoogleAIModels["GEMINI_2_0_FLASH_001"] = "gemini-2.0-flash-001";
+    // Gemini 1.5 Series (Legacy support)
+    GoogleAIModels["GEMINI_1_5_PRO"] = "gemini-1.5-pro";
+    GoogleAIModels["GEMINI_1_5_FLASH"] = "gemini-1.5-flash";
+    GoogleAIModels["GEMINI_1_5_FLASH_LITE"] = "gemini-1.5-flash-lite";
 })(GoogleAIModels || (GoogleAIModels = {}));
 /**
  * Default provider configurations

package/dist/lib/factories/providerFactory.js

@@ -35,7 +35,16 @@ export class ProviderFactory {
         if (!registration) {
             throw new Error(`Unknown provider: ${providerName}. Available providers: ${this.getAvailableProviders().join(", ")}`);
         }
-        const model = modelName || registration.defaultModel;
+        // Respect environment variables before falling back to registry default
+        let model = modelName;
+        if (!model) {
+            // Check for provider-specific environment variables
+            if (providerName.toLowerCase().includes("vertex")) {
+                model = process.env.VERTEX_MODEL;
+            }
+            // Fallback to registry default if no env var
+            model = model || registration.defaultModel;
+        }
         try {
             // Try calling as factory function first, then fallback to constructor
             let result;

package/dist/lib/factories/providerRegistry.js

@@ -55,7 +55,7 @@ export class ProviderRegistry {
             ProviderFactory.registerProvider(AIProviderName.VERTEX, async (modelName) => {
                 const { GoogleVertexProvider } = await import("../providers/googleVertex.js");
                 return new GoogleVertexProvider(modelName);
-            }, "gemini-2.5-flash", ["vertex", "googleVertex"]);
+            }, "claude-sonnet-4@20250514", ["vertex", "googleVertex"]);
             // Register Hugging Face provider (Unified Router implementation)
             ProviderFactory.registerProvider(AIProviderName.HUGGINGFACE, async (modelName) => {
                 const { HuggingFaceProvider } = await import("../providers/huggingFace.js");
@@ -82,6 +82,11 @@ export class ProviderRegistry {
                 return new OpenAICompatibleProvider(modelName, sdk);
             }, process.env.OPENAI_COMPATIBLE_MODEL || undefined, // Enable auto-discovery when no model specified
             ["openai-compatible", "openrouter", "vllm", "compatible"]);
+            // Register Amazon SageMaker provider
+            ProviderFactory.registerProvider(AIProviderName.SAGEMAKER, async (modelName, providerName, sdk) => {
+                const { AmazonSageMakerProvider } = await import("../providers/amazonSagemaker.js");
+                return new AmazonSageMakerProvider(modelName);
+            }, process.env.SAGEMAKER_MODEL || "sagemaker-model", ["sagemaker", "aws-sagemaker"]);
             logger.debug("All providers registered successfully");
             this.registered = true;
         }

package/dist/lib/neurolink.d.ts

@@ -10,6 +10,7 @@ import type { GenerateOptions, GenerateResult } from "./types/generateTypes.js";
 import type { StreamOptions, StreamResult } from "./types/streamTypes.js";
 import type { SimpleTool } from "./sdk/toolRegistration.js";
 import type { InMemoryMCPServerConfig } from "./types/mcpTypes.js";
+import { EventEmitter } from "events";
 export interface ProviderStatus {
     provider: string;
     status: "working" | "failed" | "not-configured";
@@ -41,10 +42,19 @@ export interface MCPServerInfo {
 }
 export declare class NeuroLink {
     private mcpInitialized;
+    private emitter;
     private customTools;
     private inMemoryServers;
     private toolCircuitBreakers;
     private toolExecutionMetrics;
+    /**
+     * Helper method to emit tool end event in a consistent way
+     * Used by executeTool in both success and error paths
+     * @param toolName - Name of the tool
+     * @param startTime - Timestamp when tool execution started
+     * @param success - Whether the tool execution was successful
+     */
+    private emitToolEndEvent;
     constructor();
     /**
      * Initialize MCP registry with enhanced error handling and resource cleanup
@@ -101,6 +111,11 @@ export declare class NeuroLink {
      * Future-ready for multi-modal capabilities with current text focus
      */
     stream(options: StreamOptions): Promise<StreamResult>;
+    /**
+     * Get the EventEmitter to listen to NeuroLink events
+     * @returns EventEmitter instance
+     */
+    getEventEmitter(): EventEmitter<[never]>;
     /**
      * Register a custom tool that will be available to all AI providers
      * @param name - Unique name for the tool

package/dist/lib/neurolink.js

@@ -26,15 +26,32 @@ import { validateTool, createMCPServerFromTools, } from "./sdk/toolRegistration.
 import { processFactoryOptions, enhanceTextGenerationOptions, validateFactoryConfig, processStreamingFactoryOptions, createCleanStreamOptions, } from "./utils/factoryProcessing.js";
 // Enhanced error handling imports
 import { ErrorFactory, NeuroLinkError, withTimeout, withRetry, isRetriableError, logStructuredError, CircuitBreaker, } from "./utils/errorHandling.js";
+import { EventEmitter } from "events";
 // Core types imported from core/types.js
 export class NeuroLink {
     mcpInitialized = false;
+    emitter = new EventEmitter();
     // Tool registration support
     customTools = new Map();
     inMemoryServers = new Map();
     // Enhanced error handling support
     toolCircuitBreakers = new Map();
     toolExecutionMetrics = new Map();
+    /**
+     * Helper method to emit tool end event in a consistent way
+     * Used by executeTool in both success and error paths
+     * @param toolName - Name of the tool
+     * @param startTime - Timestamp when tool execution started
+     * @param success - Whether the tool execution was successful
+     */
+    emitToolEndEvent(toolName, startTime, success) {
+        this.emitter.emit("tool:end", {
+            toolName,
+            responseTime: Date.now() - startTime,
+            success,
+            timestamp: Date.now(),
+        });
+    }
     constructor() {
         // SDK always disables manual MCP config for security
         ProviderRegistry.setOptions({
@@ -91,7 +108,6 @@ export class NeuroLink {
      * Replaces both generateText and legacy methods
      */
     async generate(optionsOrPrompt) {
-        const startTime = Date.now();
         // Convert string prompt to full options
         const options = typeof optionsOrPrompt === "string"
             ? { input: { text: optionsOrPrompt } }
@@ -100,6 +116,12 @@ 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");
         }
+        const startTime = Date.now();
+        // Emit generation start event
+        this.emitter.emit("generation:start", {
+            provider: options.provider || "auto",
+            timestamp: startTime,
+        });
         // Process factory configuration
         const factoryResult = processFactoryOptions(options);
         // Validate factory configuration if present
@@ -142,6 +164,13 @@ export class NeuroLink {
         }
         // Use redesigned generation logic
         const textResult = await this.generateTextInternal(textOptions);
+        // Emit generation completion event
+        this.emitter.emit("generation:end", {
+            provider: textResult.provider,
+            responseTime: Date.now() - startTime,
+            toolsUsed: textResult.toolsUsed,
+            timestamp: Date.now(),
+        });
         // Convert back to GenerateResult
         const generateResult = {
             content: textResult.content,
@@ -509,6 +538,11 @@ export class NeuroLink {
             options.input.text.trim() === "") {
             throw new Error("Stream options must include input.text as a non-empty string");
         }
+        // Emit stream start event
+        this.emitter.emit("stream:start", {
+            provider: options.provider || "auto",
+            timestamp: startTime,
+        });
         // Process factory configuration for streaming
         const factoryResult = processFactoryOptions(options);
         const streamingResult = processStreamingFactoryOptions(options);
@@ -576,6 +610,11 @@ export class NeuroLink {
                 responseTime,
                 provider: providerName,
             });
+            // Emit stream completion event
+            this.emitter.emit("stream:end", {
+                provider: providerName,
+                responseTime,
+            });
             // Convert to StreamResult format - Include analytics and evaluation from provider
             return {
                 stream,
@@ -615,6 +654,12 @@ export class NeuroLink {
             const cleanOptions = createCleanStreamOptions(enhancedOptions);
             const streamResult = await provider.stream(cleanOptions);
             const responseTime = Date.now() - startTime;
+            // Emit stream completion event for fallback
+            this.emitter.emit("stream:end", {
+                provider: providerName,
+                responseTime,
+                fallback: true,
+            });
             return {
                 stream: streamResult.stream,
                 provider: providerName,
@@ -643,6 +688,13 @@ export class NeuroLink {
             };
         }
     }
+    /**
+     * Get the EventEmitter to listen to NeuroLink events
+     * @returns EventEmitter instance
+     */
+    getEventEmitter() {
+        return this.emitter;
+    }
     // ========================================
     // Tool Registration API
     // ========================================
@@ -652,6 +704,11 @@ export class NeuroLink {
      * @param tool - Tool configuration
      */
     registerTool(name, tool) {
+        // Emit tool registration start event
+        this.emitter.emit("tools-register:start", {
+            toolName: name,
+            timestamp: Date.now(),
+        });
         try {
             // Validate tool configuration
             validateTool(name, tool);
@@ -666,6 +723,12 @@ export class NeuroLink {
             // Store as in-memory server
             this.inMemoryServers.set(serverId, mcpServer);
             logger.info(`Registered custom tool: ${name}`);
+            // Emit tool registration success event
+            this.emitter.emit("tools-register:end", {
+                toolName: name,
+                success: true,
+                timestamp: Date.now(),
+            });
         }
         catch (error) {
             logger.error(`Failed to register tool ${name}:`, error);
@@ -758,6 +821,11 @@ export class NeuroLink {
     async executeTool(toolName, params = {}, options) {
         const functionTag = "NeuroLink.executeTool";
         const executionStartTime = Date.now();
+        // Emit tool start event
+        this.emitter.emit("tool:start", {
+            toolName,
+            timestamp: executionStartTime,
+        });
         // Set default options
         const finalOptions = {
             timeout: options?.timeout || 30000, // 30 second default timeout
@@ -832,6 +900,8 @@ export class NeuroLink {
                 memoryDelta,
                 circuitBreakerState: circuitBreaker.getState(),
             });
+            // Emit tool end event using the helper method
+            this.emitToolEndEvent(toolName, executionStartTime, true);
             return result;
         }
         catch (error) {
@@ -867,6 +937,8 @@ export class NeuroLink {
             else {
                 structuredError = ErrorFactory.toolExecutionFailed(toolName, new Error(String(error)));
             }
+            // Emit tool end event using the helper method
+            this.emitToolEndEvent(toolName, executionStartTime, false);
             // Add execution context to structured error
             structuredError = new NeuroLinkError({
                 ...structuredError,

package/dist/lib/providers/amazonSagemaker.d.ts

@@ -0,0 +1,67 @@
+/**
+ * Amazon SageMaker Provider Implementation (Simplified)
+ *
+ * This module provides a simplified SageMaker provider that extends BaseProvider
+ * and integrates with the NeuroLink ecosystem using existing patterns.
+ */
+import type { ZodType, ZodTypeDef } from "zod";
+import type { Schema, LanguageModelV1 } from "ai";
+import type { AIProviderName } from "../core/types.js";
+import type { StreamOptions, StreamResult } from "../types/streamTypes.js";
+import { BaseProvider } from "../core/baseProvider.js";
+/**
+ * Amazon SageMaker Provider extending BaseProvider
+ */
+export declare class AmazonSageMakerProvider extends BaseProvider {
+    private sagemakerModel;
+    private sagemakerConfig;
+    private modelConfig;
+    constructor(modelName?: string, endpointName?: string);
+    protected getProviderName(): AIProviderName;
+    protected getDefaultModel(): string;
+    protected getAISDKModel(): LanguageModelV1;
+    protected executeStream(options: StreamOptions, analysisSchema?: ZodType<unknown, ZodTypeDef, unknown> | Schema<unknown>): Promise<StreamResult>;
+    protected handleProviderError(error: unknown): Error;
+    /**
+     * Get SageMaker-specific provider information
+     */
+    getSageMakerInfo(): {
+        endpointName: string;
+        modelType: string;
+        region: string;
+        configured: boolean;
+    };
+    /**
+     * Test basic configuration
+     */
+    testConnection(): Promise<{
+        connected: boolean;
+        error?: string;
+    }>;
+    /**
+     * Public method to get the AI SDK model for CLI and external usage
+     */
+    getModel(): Promise<LanguageModelV1>;
+    /**
+     * Test connectivity to the SageMaker endpoint
+     */
+    testConnectivity(): Promise<{
+        success: boolean;
+        error?: string;
+    }>;
+    /**
+     * Get model capabilities and information
+     */
+    getModelCapabilities(): {
+        capabilities: {
+            streaming: boolean;
+            toolCalling: boolean;
+            structuredOutput: boolean;
+            batchInference: boolean;
+            supportedResponseFormats: string[];
+            supportedToolTypes: string[];
+            maxBatchSize: number;
+        };
+    };
+}
+export default AmazonSageMakerProvider;