@juspay/neurolink 7.8.0 → 7.10.0

package/dist/cli/commands/sagemaker.js

@@ -8,6 +8,7 @@ import chalk from "chalk";
 import ora from "ora";
 import inquirer from "inquirer";
 import { SageMakerClient, ListEndpointsCommand, } from "@aws-sdk/client-sagemaker";
+import { logger } from "../../lib/utils/logger.js";
 import { checkSageMakerConfiguration, getSageMakerConfig, getConfigurationSummary, clearConfigurationCache, } from "../../lib/providers/sagemaker/config.js";
 import { AmazonSageMakerProvider } from "../../lib/providers/sagemaker/index.js";
 import { runQuickDiagnostics, formatDiagnosticReport, } from "../../lib/providers/sagemaker/diagnostics.js";
@@ -176,35 +177,35 @@ async function statusHandler() {
     try {
         const status = checkSageMakerConfiguration();
         spinner.stop();
-        console.log(chalk.blue("\n🔍 SageMaker Configuration Status\n"));
+        logger.always(chalk.blue("\n🔍 SageMaker Configuration Status\n"));
         if (status.configured) {
-            console.log(chalk.green("✅ Configuration: Valid"));
+            logger.always(chalk.green("✅ Configuration: Valid"));
         }
         else {
-            console.log(chalk.red("❌ Configuration: Invalid"));
+            logger.always(chalk.red("❌ Configuration: Invalid"));
         }
         if (status.issues.length > 0) {
-            console.log(chalk.yellow("\n⚠️  Issues found:"));
+            logger.always(chalk.yellow("\n⚠️  Issues found:"));
             status.issues.forEach((issue) => {
-                console.log(`   • ${issue}`);
+                logger.always(`   • ${issue}`);
             });
         }
         // Show configuration summary (safe for display)
         if (status.summary) {
-            console.log(chalk.blue("\n📋 Configuration Summary:"));
+            logger.always(chalk.blue("\n📋 Configuration Summary:"));
             if (typeof status.summary === "object" && status.summary.aws) {
                 const aws = status.summary.aws;
-                console.log(`   Region: ${aws.region}`);
-                console.log(`   Access Key: ${aws.accessKeyId}`);
-                console.log(`   Timeout: ${aws.timeout}ms`);
-                console.log(`   Max Retries: ${aws.maxRetries}`);
+                logger.always(`   Region: ${aws.region}`);
+                logger.always(`   Access Key: ${aws.accessKeyId}`);
+                logger.always(`   Timeout: ${aws.timeout}ms`);
+                logger.always(`   Max Retries: ${aws.maxRetries}`);
             }
         }
         process.exit(status.configured ? 0 : 1);
     }
     catch (error) {
         spinner.fail("Failed to check SageMaker configuration");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -219,7 +220,7 @@ async function testEndpointHandler(argv) {
         const status = checkSageMakerConfiguration();
         if (!status.configured) {
             spinner.fail("SageMaker configuration is invalid");
-            console.error(chalk.red("Please run 'neurolink sagemaker setup' first"));
+            logger.error(chalk.red("Please run 'neurolink sagemaker setup' first"));
             process.exit(1);
         }
         // Create provider and test connectivity
@@ -244,26 +245,26 @@ async function testEndpointHandler(argv) {
                     maxTokens: 50,
                 });
                 spinner.succeed("✅ Text generation test successful");
-                console.log(chalk.blue("\n📝 Test Response:"));
-                console.log(`   Input: "${prompt}"`);
-                console.log(`   Output: "${result.text?.substring(0, 100)}${result.text && result.text.length > 100 ? "..." : ""}"`);
-                console.log(`   Tokens: ${result.usage.promptTokens} → ${result.usage.completionTokens} (${result.usage.totalTokens ?? result.usage.promptTokens + result.usage.completionTokens} total)`);
-                console.log(`   Finish Reason: ${result.finishReason}`);
+                logger.always(chalk.blue("\n📝 Test Response:"));
+                logger.always(`   Input: "${prompt}"`);
+                logger.always(`   Output: "${result.text?.substring(0, 100)}${result.text && result.text.length > 100 ? "..." : ""}"`);
+                logger.always(`   Tokens: ${result.usage.promptTokens} → ${result.usage.completionTokens} (${result.usage.totalTokens ?? result.usage.promptTokens + result.usage.completionTokens} total)`);
+                logger.always(`   Finish Reason: ${result.finishReason}`);
             }
             catch (genError) {
                 spinner.fail("❌ Text generation test failed");
-                console.error(chalk.red(`Generation Error: ${genError instanceof Error ? genError.message : String(genError)}`));
+                logger.error(chalk.red(`Generation Error: ${genError instanceof Error ? genError.message : String(genError)}`));
             }
         }
         else {
             spinner.fail(`❌ Endpoint '${endpoint}' is not accessible`);
-            console.error(chalk.red(`Error: ${testResult.error}`));
+            logger.error(chalk.red(`Error: ${testResult.error}`));
             process.exit(1);
         }
     }
     catch (error) {
         spinner.fail("Failed to test SageMaker endpoint");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -290,34 +291,34 @@ async function listEndpointsHandler() {
             const endpoints = { Endpoints: response.Endpoints || [] };
             spinner.stop();
             if (endpoints.Endpoints && endpoints.Endpoints.length > 0) {
-                console.log(chalk.blue("\n🔗 Available SageMaker Endpoints:\n"));
+                logger.always(chalk.blue("\n🔗 Available SageMaker Endpoints:\n"));
                 endpoints.Endpoints.forEach((endpoint, index) => {
-                    console.log(`${index + 1}. ${chalk.green(endpoint.EndpointName)}`);
-                    console.log(`   Status: ${endpoint.EndpointStatus}`);
-                    console.log(`   Created: ${endpoint.CreationTime?.toLocaleDateString() ?? "Unknown"}`);
+                    logger.always(`${index + 1}. ${chalk.green(endpoint.EndpointName)}`);
+                    logger.always(`   Status: ${endpoint.EndpointStatus}`);
+                    logger.always(`   Created: ${endpoint.CreationTime?.toLocaleDateString() ?? "Unknown"}`);
                     if (endpoint.LastModifiedTime) {
-                        console.log(`   Modified: ${endpoint.LastModifiedTime.toLocaleDateString()}`);
+                        logger.always(`   Modified: ${endpoint.LastModifiedTime.toLocaleDateString()}`);
                     }
-                    console.log();
+                    logger.always();
                 });
             }
             else {
-                console.log(chalk.yellow("No SageMaker endpoints found"));
+                logger.always(chalk.yellow("No SageMaker endpoints found"));
             }
         }
-        catch (awsError) {
+        catch (_awsError) {
             spinner.fail("Failed to list endpoints");
-            console.error(chalk.red("AWS SDK credentials error or insufficient permissions"));
-            console.log(chalk.yellow("\nTo list endpoints, please:"));
-            console.log("1. Set AWS_ACCESS_KEY_ID environment variable");
-            console.log("2. Set AWS_SECRET_ACCESS_KEY environment variable");
-            console.log("3. Set AWS_REGION environment variable (or use default)");
-            console.log("4. Ensure you have sagemaker:ListEndpoints permission");
+            logger.error(chalk.red("AWS SDK credentials error or insufficient permissions"));
+            logger.always(chalk.yellow("\nTo list endpoints, please:"));
+            logger.always("1. Set AWS_ACCESS_KEY_ID environment variable");
+            logger.always("2. Set AWS_SECRET_ACCESS_KEY environment variable");
+            logger.always("3. Set AWS_REGION environment variable (or use default)");
+            logger.always("4. Ensure you have sagemaker:ListEndpoints permission");
         }
     }
     catch (error) {
         spinner.fail("Failed to list SageMaker endpoints");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -330,9 +331,9 @@ async function configHandler(args) {
     try {
         const summary = getConfigurationSummary();
         spinner.stop();
-        console.log(chalk.blue("\n⚙️  SageMaker Configuration\n"));
+        logger.always(chalk.blue("\n⚙️  SageMaker Configuration\n"));
         if (format === "json") {
-            console.log(JSON.stringify(summary, null, 2));
+            logger.always(JSON.stringify(summary, null, 2));
         }
         else if (format === "yaml") {
             // Simple YAML-like output
@@ -340,11 +341,11 @@ async function configHandler(args) {
                 const spaces = " ".repeat(indent);
                 for (const [key, value] of Object.entries(obj)) {
                     if (typeof value === "object" && value !== null) {
-                        console.log(`${spaces}${key}:`);
+                        logger.always(`${spaces}${key}:`);
                         printYaml(value, indent + 2);
                     }
                     else {
-                        console.log(`${spaces}${key}: ${value}`);
+                        logger.always(`${spaces}${key}: ${value}`);
                     }
                 }
             }
@@ -358,33 +359,33 @@ async function configHandler(args) {
                     {});
                 const environment = (summary.environment ||
                     {});
-                console.log(chalk.green("AWS Configuration:"));
-                console.log(`  Region: ${aws.region}`);
-                console.log(`  Access Key: ${aws.accessKeyId}`);
-                console.log(`  Secret Key: ${aws.secretAccessKey}`);
-                console.log(`  Session Token: ${aws.sessionToken}`);
-                console.log(`  Timeout: ${aws.timeout}ms`);
-                console.log(`  Max Retries: ${aws.maxRetries}`);
-                console.log(`  Custom Endpoint: ${aws.endpoint || "None"}`);
-                console.log(chalk.blue("\nSageMaker Configuration:"));
-                console.log(`  Default Endpoint: ${sagemaker.defaultEndpoint}`);
-                console.log(`  Model Name: ${sagemaker.model}`);
+                logger.always(chalk.green("AWS Configuration:"));
+                logger.always(`  Region: ${aws.region}`);
+                logger.always(`  Access Key: ${aws.accessKeyId}`);
+                logger.always(`  Secret Key: ${aws.secretAccessKey}`);
+                logger.always(`  Session Token: ${aws.sessionToken}`);
+                logger.always(`  Timeout: ${aws.timeout}ms`);
+                logger.always(`  Max Retries: ${aws.maxRetries}`);
+                logger.always(`  Custom Endpoint: ${aws.endpoint || "None"}`);
+                logger.always(chalk.blue("\nSageMaker Configuration:"));
+                logger.always(`  Default Endpoint: ${sagemaker.defaultEndpoint}`);
+                logger.always(`  Model Name: ${sagemaker.model}`);
                 if (sagemaker.modelConfig) {
                     const modelConfig = sagemaker.modelConfig;
-                    console.log(`  Model Type: ${modelConfig.modelType}`);
-                    console.log(`  Content Type: ${modelConfig.contentType}`);
-                    console.log(`  Accept: ${modelConfig.accept}`);
+                    logger.always(`  Model Type: ${modelConfig.modelType}`);
+                    logger.always(`  Content Type: ${modelConfig.contentType}`);
+                    logger.always(`  Accept: ${modelConfig.accept}`);
                 }
-                console.log(chalk.yellow("\nEnvironment:"));
-                console.log(`  Node Environment: ${environment.nodeEnv}`);
-                console.log(`  SageMaker Configured: ${environment.sagemakerConfigured ? "Yes" : "No"}`);
-                console.log(`  AWS Configured: ${environment.awsConfigured ? "Yes" : "No"}`);
+                logger.always(chalk.yellow("\nEnvironment:"));
+                logger.always(`  Node Environment: ${environment.nodeEnv}`);
+                logger.always(`  SageMaker Configured: ${environment.sagemakerConfigured ? "Yes" : "No"}`);
+                logger.always(`  AWS Configured: ${environment.awsConfigured ? "Yes" : "No"}`);
             }
         }
     }
     catch (error) {
         spinner.fail("Failed to load SageMaker configuration");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -392,9 +393,9 @@ async function configHandler(args) {
  * Handler for interactive setup
  */
 async function setupHandler() {
-    console.log(chalk.blue("\n🚀 SageMaker Interactive Setup\n"));
+    logger.always(chalk.blue("\n🚀 SageMaker Interactive Setup\n"));
     // Pre-setup security advisory
-    console.log(chalk.yellow.bold("🔒 SECURITY NOTICE: You will be prompted to enter AWS credentials.\n" +
+    logger.always(chalk.yellow.bold("🔒 SECURITY NOTICE: You will be prompted to enter AWS credentials.\n" +
         "These credentials will be stored temporarily in memory only.\n" +
         "For production use, consider using AWS credential files or IAM roles.\n"));
     // Ask for user confirmation before proceeding
@@ -407,11 +408,11 @@ async function setupHandler() {
         },
     ]);
     if (!confirmSetup) {
-        console.log(chalk.blue("\nSetup cancelled. Consider using alternative credential methods:"));
-        console.log("• AWS credential files: ~/.aws/credentials");
-        console.log("• Environment variables in .env file");
-        console.log("• AWS CLI configuration: aws configure");
-        console.log("• IAM roles for production environments");
+        logger.always(chalk.blue("\nSetup cancelled. Consider using alternative credential methods:"));
+        logger.always("• AWS credential files: ~/.aws/credentials");
+        logger.always("• Environment variables in .env file");
+        logger.always("• AWS CLI configuration: aws configure");
+        logger.always("• IAM roles for production environments");
         return;
     }
     try {
@@ -456,8 +457,8 @@ async function setupHandler() {
         const spinner = ora("Setting up SageMaker configuration...").start();
         // Enhanced security warnings for credential handling
         spinner.stop();
-        console.log(chalk.red.bold("\n🔒 CRITICAL SECURITY WARNINGS\n"));
-        console.log(chalk.yellow.bold("⚠️  CREDENTIAL PERSISTENCE: AWS credentials will only be set for this session.\n" +
+        logger.always(chalk.red.bold("\n🔒 CRITICAL SECURITY WARNINGS\n"));
+        logger.always(chalk.yellow.bold("⚠️  CREDENTIAL PERSISTENCE: AWS credentials will only be set for this session.\n" +
             "   They will NOT persist after you exit the CLI.\n\n" +
             "🔐 SECURE STORAGE OPTIONS:\n" +
             "   • Use environment variables in a secure .env file (never commit to git)\n" +
@@ -495,17 +496,17 @@ async function setupHandler() {
         try {
             validateSecureConfiguration(secureConfig); // Validate configuration is loadable
             spinner.succeed("✅ Configuration validated successfully");
-            console.log(chalk.green("\n🎉 SageMaker setup complete!"));
-            console.log(chalk.yellow("\n💡 Next steps:"));
-            console.log("1. Test your endpoint: neurolink sagemaker test <endpoint-name>");
-            console.log("2. Check status: neurolink sagemaker status");
-            console.log("3. List endpoints: neurolink sagemaker list-endpoints");
-            console.log(chalk.blue("\n🔒 Secure configuration validated:"));
-            console.log("  ✓ AWS credentials verified");
-            console.log("  ✓ AWS region validated");
-            console.log("  ✓ SageMaker endpoint configured");
-            console.log("  ✓ Timeout and retry settings applied");
-            console.log(chalk.yellow("\n⚠️  For persistent configuration, add these to your .env file:\n" +
+            logger.always(chalk.green("\n🎉 SageMaker setup complete!"));
+            logger.always(chalk.yellow("\n💡 Next steps:"));
+            logger.always("1. Test your endpoint: neurolink sagemaker test <endpoint-name>");
+            logger.always("2. Check status: neurolink sagemaker status");
+            logger.always("3. List endpoints: neurolink sagemaker list-endpoints");
+            logger.always(chalk.blue("\n🔒 Secure configuration validated:"));
+            logger.always("  ✓ AWS credentials verified");
+            logger.always("  ✓ AWS region validated");
+            logger.always("  ✓ SageMaker endpoint configured");
+            logger.always("  ✓ Timeout and retry settings applied");
+            logger.always(chalk.yellow("\n⚠️  For persistent configuration, add these to your .env file:\n" +
                 "  AWS_ACCESS_KEY_ID=your_access_key\n" +
                 "  AWS_SECRET_ACCESS_KEY=your_secret_key\n" +
                 "  AWS_REGION=" +
@@ -525,14 +526,14 @@ async function setupHandler() {
         }
         catch (configError) {
             spinner.fail("❌ Configuration validation failed");
-            console.error(chalk.red(`Error: ${configError instanceof Error ? configError.message : String(configError)}`));
+            logger.error(chalk.red(`Error: ${configError instanceof Error ? configError.message : String(configError)}`));
             // Clear secure credentials from memory on error
             clearSecureCredentials(secureConfig.sessionId);
             process.exit(1);
         }
     }
     catch (error) {
-        console.error(chalk.red(`Setup failed: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Setup failed: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -544,27 +545,27 @@ async function validateHandler() {
     try {
         const status = checkSageMakerConfiguration();
         spinner.stop();
-        console.log(chalk.blue("\n🔍 Configuration Validation Results\n"));
+        logger.always(chalk.blue("\n🔍 Configuration Validation Results\n"));
         if (status.configured) {
-            console.log(chalk.green("✅ All checks passed"));
-            console.log(chalk.blue("🚀 SageMaker is ready to use"));
+            logger.always(chalk.green("✅ All checks passed"));
+            logger.always(chalk.blue("🚀 SageMaker is ready to use"));
         }
         else {
-            console.log(chalk.red("❌ Configuration validation failed"));
+            logger.always(chalk.red("❌ Configuration validation failed"));
             if (status.issues.length > 0) {
-                console.log(chalk.yellow("\n🔧 Issues to fix:"));
+                logger.always(chalk.yellow("\n🔧 Issues to fix:"));
                 status.issues.forEach((issue, index) => {
-                    console.log(`${index + 1}. ${issue}`);
+                    logger.always(`${index + 1}. ${issue}`);
                 });
             }
-            console.log(chalk.blue("\n💡 How to fix:"));
-            console.log("Run: neurolink sagemaker setup");
+            logger.always(chalk.blue("\n💡 How to fix:"));
+            logger.always("Run: neurolink sagemaker setup");
             process.exit(1);
         }
     }
     catch (error) {
         spinner.fail("Validation failed");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -573,18 +574,18 @@ async function validateHandler() {
  */
 async function benchmarkHandler(argv) {
     const { endpoint, requests = 10, concurrency = 2, maxTokens = 100 } = argv;
-    console.log(chalk.blue(`\n⚡ SageMaker Performance Benchmark\n`));
-    console.log(`Endpoint: ${endpoint}`);
-    console.log(`Requests: ${requests}`);
-    console.log(`Concurrency: ${concurrency}`);
-    console.log(`Max Tokens: ${maxTokens}\n`);
+    logger.always(chalk.blue(`\n⚡ SageMaker Performance Benchmark\n`));
+    logger.always(`Endpoint: ${endpoint}`);
+    logger.always(`Requests: ${requests}`);
+    logger.always(`Concurrency: ${concurrency}`);
+    logger.always(`Max Tokens: ${maxTokens}\n`);
     const spinner = ora("Setting up benchmark...").start();
     try {
         // Check configuration first
         const status = checkSageMakerConfiguration();
         if (!status.configured) {
             spinner.fail("SageMaker configuration is invalid");
-            console.error(chalk.red("Please run 'neurolink sagemaker setup' first"));
+            logger.error(chalk.red("Please run 'neurolink sagemaker setup' first"));
             process.exit(1);
         }
         const provider = new AmazonSageMakerProvider(undefined, endpoint);
@@ -593,7 +594,7 @@ async function benchmarkHandler(argv) {
         const connectivityTest = await provider.testConnectivity();
         if (!connectivityTest.success) {
             spinner.fail(`Endpoint '${endpoint}' is not accessible`);
-            console.error(chalk.red(`Error: ${connectivityTest.error}`));
+            logger.error(chalk.red(`Error: ${connectivityTest.error}`));
             process.exit(1);
         }
         spinner.text = "Starting benchmark...";
@@ -651,35 +652,35 @@ async function benchmarkHandler(argv) {
         const failed = results.filter((r) => !r.success);
         const durations = successful.map((r) => r.duration);
         const totalTokens = successful.reduce((sum, r) => sum + r.tokens, 0);
-        console.log(chalk.green("\n📊 Benchmark Results\n"));
-        console.log(`Total Time: ${totalTime}ms`);
-        console.log(`Successful Requests: ${successful.length}/${requests}`);
-        console.log(`Failed Requests: ${failed.length}`);
-        console.log(`Success Rate: ${((successful.length / requests) * 100).toFixed(1)}%`);
+        logger.always(chalk.green("\n📊 Benchmark Results\n"));
+        logger.always(`Total Time: ${totalTime}ms`);
+        logger.always(`Successful Requests: ${successful.length}/${requests}`);
+        logger.always(`Failed Requests: ${failed.length}`);
+        logger.always(`Success Rate: ${((successful.length / requests) * 100).toFixed(1)}%`);
         if (successful.length > 0) {
-            console.log(`\nLatency Statistics:`);
-            console.log(`  Average: ${(durations.reduce((a, b) => a + b, 0) / durations.length).toFixed(0)}ms`);
-            console.log(`  Minimum: ${Math.min(...durations)}ms`);
-            console.log(`  Maximum: ${Math.max(...durations)}ms`);
-            console.log(`  Median: ${durations.sort((a, b) => a - b)[Math.floor(durations.length / 2)]}ms`);
-            console.log(`\nThroughput:`);
-            console.log(`  Requests/sec: ${(successful.length / (totalTime / 1000)).toFixed(2)}`);
-            console.log(`  Tokens/sec: ${(totalTokens / (totalTime / 1000)).toFixed(2)}`);
-            console.log(`  Average tokens/request: ${(totalTokens / successful.length).toFixed(1)}`);
+            logger.always(`\nLatency Statistics:`);
+            logger.always(`  Average: ${(durations.reduce((a, b) => a + b, 0) / durations.length).toFixed(0)}ms`);
+            logger.always(`  Minimum: ${Math.min(...durations)}ms`);
+            logger.always(`  Maximum: ${Math.max(...durations)}ms`);
+            logger.always(`  Median: ${durations.sort((a, b) => a - b)[Math.floor(durations.length / 2)]}ms`);
+            logger.always(`\nThroughput:`);
+            logger.always(`  Requests/sec: ${(successful.length / (totalTime / 1000)).toFixed(2)}`);
+            logger.always(`  Tokens/sec: ${(totalTokens / (totalTime / 1000)).toFixed(2)}`);
+            logger.always(`  Average tokens/request: ${(totalTokens / successful.length).toFixed(1)}`);
         }
         if (failed.length > 0) {
-            console.log(chalk.red(`\n❌ Failed Requests (${failed.length}):`));
+            logger.always(chalk.red(`\n❌ Failed Requests (${failed.length}):`));
             failed.slice(0, 5).forEach((failure, index) => {
-                console.log(`  ${index + 1}. ${failure.error}`);
+                logger.always(`  ${index + 1}. ${failure.error}`);
             });
             if (failed.length > 5) {
-                console.log(`  ... and ${failed.length - 5} more`);
+                logger.always(`  ... and ${failed.length - 5} more`);
             }
         }
     }
     catch (error) {
         spinner.fail("Benchmark failed");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -691,11 +692,11 @@ async function clearCacheHandler() {
     try {
         clearConfigurationCache();
         spinner.succeed("✅ Configuration cache cleared");
-        console.log(chalk.blue("Configuration will be reloaded on next use"));
+        logger.always(chalk.blue("Configuration will be reloaded on next use"));
     }
     catch (error) {
         spinner.fail("Failed to clear cache");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
         process.exit(1);
     }
 }
@@ -704,15 +705,15 @@ async function clearCacheHandler() {
  */
 async function diagnoseHandler(argv) {
     const { endpoint, quick, full, timeout } = argv;
-    console.log(chalk.blue(`\n🔍 SageMaker Streaming Diagnostics\n`));
+    logger.always(chalk.blue(`\n🔍 SageMaker Streaming Diagnostics\n`));
     if (endpoint) {
-        console.log(`Endpoint: ${endpoint}`);
+        logger.always(`Endpoint: ${endpoint}`);
     }
     else {
-        console.log("Endpoint: Not specified (configuration tests only)");
+        logger.always("Endpoint: Not specified (configuration tests only)");
     }
-    console.log(`Mode: ${quick ? "Quick" : full ? "Full" : "Standard"}`);
-    console.log(`Timeout: ${timeout}ms\n`);
+    logger.always(`Mode: ${quick ? "Quick" : full ? "Full" : "Standard"}`);
+    logger.always(`Timeout: ${timeout}ms\n`);
     const spinner = ora("Starting diagnostics...").start();
     try {
         // Run diagnostics (simplified - advanced streaming diagnostics removed)
@@ -720,59 +721,59 @@ async function diagnoseHandler(argv) {
         spinner.stop();
         // Display results
         const formatted = formatDiagnosticReport(report);
-        console.log(formatted);
+        logger.always(formatted);
         // Additional insights based on results
         if (report.overallStatus === "critical") {
-            console.log(chalk.red("🚨 Critical Issues Detected"));
-            console.log(chalk.red("   Your streaming configuration has serious problems that need immediate attention."));
-            console.log(chalk.yellow("   See the recommendations above for resolution steps.\n"));
+            logger.always(chalk.red("🚨 Critical Issues Detected"));
+            logger.always(chalk.red("   Your streaming configuration has serious problems that need immediate attention."));
+            logger.always(chalk.yellow("   See the recommendations above for resolution steps.\n"));
         }
         else if (report.overallStatus === "issues") {
-            console.log(chalk.yellow("⚠️  Issues Detected"));
-            console.log(chalk.yellow("   Your streaming setup works but has some issues that could affect performance."));
-            console.log(chalk.blue("   Consider addressing the recommendations above.\n"));
+            logger.always(chalk.yellow("⚠️  Issues Detected"));
+            logger.always(chalk.yellow("   Your streaming setup works but has some issues that could affect performance."));
+            logger.always(chalk.blue("   Consider addressing the recommendations above.\n"));
         }
         else {
-            console.log(chalk.green("✅ All Systems Go"));
-            console.log(chalk.green("   Your SageMaker streaming configuration looks healthy!"));
+            logger.always(chalk.green("✅ All Systems Go"));
+            logger.always(chalk.green("   Your SageMaker streaming configuration looks healthy!"));
             if (endpoint) {
-                console.log(chalk.blue("   You can now use streaming features with confidence.\n"));
-                console.log(chalk.dim("   Try: neurolink sagemaker stream " + endpoint));
+                logger.always(chalk.blue("   You can now use streaming features with confidence.\n"));
+                logger.always(chalk.dim("   Try: neurolink sagemaker stream " + endpoint));
             }
         }
         // Show additional help based on findings
         const failedTests = report.results.filter((r) => r.status === "fail");
         if (failedTests.length > 0) {
-            console.log(chalk.blue("📚 Additional Resources:"));
+            logger.always(chalk.blue("📚 Additional Resources:"));
             const hasConnectivityIssues = failedTests.some((t) => t.category === "connectivity");
             const hasStreamingIssues = failedTests.some((t) => t.category === "streaming");
             const hasConfigIssues = failedTests.some((t) => t.category === "configuration");
             if (hasConfigIssues) {
-                console.log("   • Configuration: neurolink sagemaker setup");
+                logger.always("   • Configuration: neurolink sagemaker setup");
             }
             if (hasConnectivityIssues) {
-                console.log("   • Connectivity: neurolink sagemaker test " +
+                logger.always("   • Connectivity: neurolink sagemaker test " +
                     (endpoint || "your-endpoint"));
             }
             if (hasStreamingIssues) {
-                console.log("   • Streaming Guide: docs/providers/sagemaker/streaming-troubleshooting.md");
+                logger.always("   • Streaming Guide: docs/providers/sagemaker/streaming-troubleshooting.md");
             }
-            console.log("   • Full Diagnostics: neurolink sagemaker diagnose " +
+            logger.always("   • Full Diagnostics: neurolink sagemaker diagnose " +
                 (endpoint || "") +
                 " --full");
-            console.log();
+            logger.always();
         }
         // Exit with appropriate code
         process.exit(report.overallStatus === "critical" ? 1 : 0);
     }
     catch (error) {
         spinner.fail("Diagnostics failed");
-        console.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
-        console.log(chalk.yellow("\n💡 Diagnostic troubleshooting:"));
-        console.log("   • Check your SageMaker configuration: neurolink sagemaker status");
-        console.log("   • Verify AWS credentials and permissions");
-        console.log("   • Try with a specific endpoint: neurolink sagemaker diagnose your-endpoint");
-        console.log("   • Run quick mode: neurolink sagemaker diagnose --quick");
+        logger.error(chalk.red(`Error: ${error instanceof Error ? error.message : String(error)}`));
+        logger.always(chalk.yellow("\n💡 Diagnostic troubleshooting:"));
+        logger.always("   • Check your SageMaker configuration: neurolink sagemaker status");
+        logger.always("   • Verify AWS credentials and permissions");
+        logger.always("   • Try with a specific endpoint: neurolink sagemaker diagnose your-endpoint");
+        logger.always("   • Run quick mode: neurolink sagemaker diagnose --quick");
         process.exit(1);
     }
 }

package/dist/cli/factories/commandFactory.js

@@ -288,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
@@ -448,15 +452,15 @@ export class CLICommandFactory {
             describe: "Manage NeuroLink configuration",
             builder: (yargs) => {
                 return yargs
-                    .command("init", "Interactive configuration setup wizard", (y) => this.buildOptions(y), async (argv) => {
+                    .command("init", "Interactive configuration setup wizard", (y) => this.buildOptions(y), async (_argv) => {
                     const { configManager } = await import("../commands/config.js");
                     await configManager.initInteractive();
                 })
-                    .command("show", "Display current configuration", (y) => this.buildOptions(y), async (argv) => {
+                    .command("show", "Display current configuration", (y) => this.buildOptions(y), async (_argv) => {
                     const { configManager } = await import("../commands/config.js");
                     configManager.showConfig();
                 })
-                    .command("validate", "Validate current configuration", (y) => this.buildOptions(y), async (argv) => {
+                    .command("validate", "Validate current configuration", (y) => this.buildOptions(y), async (_argv) => {
                     const { configManager } = await import("../commands/config.js");
                     const result = configManager.validateConfig();
                     if (result.valid) {
@@ -468,7 +472,7 @@ export class CLICommandFactory {
                         process.exit(1);
                     }
                 })
-                    .command("reset", "Reset configuration to defaults", (y) => this.buildOptions(y), async (argv) => {
+                    .command("reset", "Reset configuration to defaults", (y) => this.buildOptions(y), async (_argv) => {
                     const { configManager } = await import("../commands/config.js");
                     configManager.resetConfig();
                 })

package/dist/core/dynamicModels.js

@@ -100,6 +100,12 @@ export class DynamicModelProvider {
             // Setup timeout and abort controller
             const controller = new AbortController();
             const timeoutId = setTimeout(() => controller.abort(), timeoutMs);
+            // Load from URL
+            const response = await fetch(source, {
+                headers: {
+                    "User-Agent": "NeuroLink/1.0 (+https://github.com/juspay/neurolink)",
+                },
+            });
             try {
                 // Add health check for localhost before attempting full request
                 if (source.includes("localhost") || source.includes("127.0.0.1")) {