@juspay/neurolink 1.6.0 → 1.9.0

package/dist/cli/utils/interactive-setup.js

@@ -0,0 +1,302 @@
+/**
+ * Interactive Setup Utilities for NeuroLink CLI
+ *
+ * Provides interactive configuration wizard with provider selection,
+ * credential collection, testing, and environment file management.
+ */
+import { AIProviderName } from "../../lib/core/types.js";
+import { NeuroLink } from "../../lib/neurolink.js";
+import chalk from "chalk";
+import ora from "ora";
+export const PROVIDER_CONFIGS = [
+    {
+        id: AIProviderName.OPENAI,
+        name: "OpenAI",
+        description: "GPT-4, GPT-3.5, and other OpenAI models",
+        envVars: [
+            {
+                key: "OPENAI_API_KEY",
+                prompt: "OpenAI API Key (sk-...)",
+                secure: true,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.GOOGLE_AI,
+        name: "Google AI Studio",
+        description: "Gemini models via Google AI Studio",
+        envVars: [
+            {
+                key: "GOOGLE_AI_API_KEY",
+                prompt: "Google AI Studio API Key (AIza-...)",
+                secure: true,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.ANTHROPIC,
+        name: "Anthropic",
+        description: "Claude models via Anthropic API",
+        envVars: [
+            {
+                key: "ANTHROPIC_API_KEY",
+                prompt: "Anthropic API Key (sk-ant-...)",
+                secure: true,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.BEDROCK,
+        name: "AWS Bedrock",
+        description: "Claude, Llama, and other models via AWS Bedrock",
+        envVars: [
+            { key: "AWS_ACCESS_KEY_ID", prompt: "AWS Access Key ID", secure: false },
+            {
+                key: "AWS_SECRET_ACCESS_KEY",
+                prompt: "AWS Secret Access Key",
+                secure: true,
+            },
+            {
+                key: "AWS_REGION",
+                prompt: "AWS Region",
+                default: "us-east-1",
+                secure: false,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.VERTEX,
+        name: "Google Vertex AI",
+        description: "Gemini models via Google Cloud Vertex AI",
+        envVars: [
+            {
+                key: "GOOGLE_APPLICATION_CREDENTIALS",
+                prompt: "Path to Service Account JSON file",
+                secure: false,
+            },
+            {
+                key: "GOOGLE_VERTEX_PROJECT",
+                prompt: "Google Cloud Project ID",
+                secure: false,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.AZURE,
+        name: "Azure OpenAI",
+        description: "GPT models via Azure OpenAI Service",
+        envVars: [
+            {
+                key: "AZURE_OPENAI_API_KEY",
+                prompt: "Azure OpenAI API Key",
+                secure: true,
+            },
+            {
+                key: "AZURE_OPENAI_ENDPOINT",
+                prompt: "Azure OpenAI Endpoint URL",
+                secure: false,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.HUGGINGFACE,
+        name: "Hugging Face",
+        description: "100,000+ open source models",
+        envVars: [
+            {
+                key: "HUGGINGFACE_API_KEY",
+                prompt: "Hugging Face API Token (hf_...)",
+                secure: true,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.OLLAMA,
+        name: "Ollama",
+        description: "Local AI models running on your machine",
+        envVars: [
+            {
+                key: "OLLAMA_BASE_URL",
+                prompt: "Ollama Server URL",
+                default: "http://localhost:11434",
+                secure: false,
+                optional: true,
+            },
+        ],
+    },
+    {
+        id: AIProviderName.MISTRAL,
+        name: "Mistral AI",
+        description: "European GDPR-compliant AI models",
+        envVars: [
+            { key: "MISTRAL_API_KEY", prompt: "Mistral AI API Key", secure: true },
+        ],
+    },
+];
+/**
+ * Run the interactive setup wizard
+ */
+export async function runInteractiveSetup(quiet = false) {
+    // Dynamic import to avoid bundling inquirer in SDK
+    const { default: inquirer } = await import("inquirer");
+    const result = {
+        selectedProviders: [],
+        credentials: {},
+        testResults: [],
+    };
+    if (!quiet) {
+        console.log(chalk.blue("\n🎉 Welcome to NeuroLink Interactive Setup!"));
+        console.log(chalk.gray("This wizard will help you configure AI providers for NeuroLink.\n"));
+    }
+    // Step 1: Provider Selection
+    const providerChoices = PROVIDER_CONFIGS.map((config) => ({
+        name: `${config.name} - ${config.description}`,
+        value: config.id,
+        checked: false,
+    }));
+    const { selectedProviders } = await inquirer.prompt([
+        {
+            type: "checkbox",
+            name: "selectedProviders",
+            message: "Which AI providers would you like to configure?",
+            choices: providerChoices,
+            validate: (answers) => {
+                if (answers.length === 0) {
+                    return "Please select at least one provider.";
+                }
+                return true;
+            },
+        },
+    ]);
+    result.selectedProviders = selectedProviders;
+    // Step 2: Credential Collection
+    if (!quiet) {
+        console.log(chalk.blue("\n🔑 Collecting credentials for selected providers...\n"));
+    }
+    for (const providerId of selectedProviders) {
+        const config = PROVIDER_CONFIGS.find((c) => c.id === providerId);
+        if (!config) {
+            continue;
+        }
+        if (!quiet) {
+            console.log(chalk.yellow(`\n📋 Configuring ${config.name}:`));
+        }
+        for (const envVar of config.envVars) {
+            const currentValue = process.env[envVar.key];
+            const hasCurrentValue = !!currentValue;
+            let promptMessage = envVar.prompt;
+            if (hasCurrentValue) {
+                promptMessage += chalk.gray(` (current: ${envVar.secure ? "***" : currentValue})`);
+            }
+            if (envVar.default) {
+                promptMessage += chalk.gray(` (default: ${envVar.default})`);
+            }
+            const { value } = await inquirer.prompt([
+                {
+                    type: envVar.secure ? "password" : "input",
+                    name: "value",
+                    message: promptMessage,
+                    default: envVar.default,
+                    validate: (input) => {
+                        if (!envVar.optional && !input && !hasCurrentValue) {
+                            return `${envVar.key} is required.`;
+                        }
+                        return true;
+                    },
+                    when: () => {
+                        if (hasCurrentValue && !envVar.optional) {
+                            // Ask if they want to update existing value
+                            return inquirer
+                                .prompt([
+                                {
+                                    type: "confirm",
+                                    name: "update",
+                                    message: `Update existing ${envVar.key}?`,
+                                    default: false,
+                                },
+                            ])
+                                .then((answer) => answer.update);
+                        }
+                        return true;
+                    },
+                },
+            ]);
+            if (value || envVar.default) {
+                result.credentials[envVar.key] = value || envVar.default || "";
+            }
+        }
+    }
+    return result;
+}
+/**
+ * Test provider connectivity using existing logic
+ */
+export async function testProviderConnectivity(providers, quiet = false) {
+    const sdk = new NeuroLink();
+    const results = [];
+    if (!quiet) {
+        console.log(chalk.blue("\n🧪 Testing provider connectivity...\n"));
+    }
+    const spinner = quiet ? null : ora().start();
+    for (const provider of providers) {
+        if (spinner) {
+            spinner.text = `Testing ${provider}...`;
+        }
+        try {
+            const start = Date.now();
+            await sdk.generateText({ prompt: "test", provider, maxTokens: 1 });
+            const duration = Date.now() - start;
+            results.push({ provider, status: "working", responseTime: duration });
+            if (spinner) {
+                spinner.succeed(`${provider}: ${chalk.green("✅ Working")} (${duration}ms)`);
+                spinner.start(); // Restart for next provider
+            }
+            else if (!quiet) {
+                console.log(`${provider}: ${chalk.green("✅ Working")} (${duration}ms)`);
+            }
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            results.push({ provider, status: "failed", error: errorMessage });
+            if (spinner) {
+                spinner.fail(`${provider}: ${chalk.red("❌ Failed")} - ${errorMessage.split("\n")[0]}`);
+                spinner.start(); // Restart for next provider
+            }
+            else if (!quiet) {
+                console.error(`${provider}: ${chalk.red("❌ Failed")} - ${errorMessage.split("\n")[0]}`);
+            }
+        }
+    }
+    if (spinner) {
+        spinner.stop();
+    }
+    return results;
+}
+/**
+ * Display setup summary
+ */
+export function displaySetupSummary(result, quiet = false) {
+    if (quiet) {
+        return;
+    }
+    const working = result.testResults.filter((r) => r.status === "working").length;
+    const total = result.testResults.length;
+    console.log(chalk.blue("\n📊 Setup Summary:"));
+    console.log(chalk.blue("================"));
+    console.log(`Selected providers: ${result.selectedProviders.length}`);
+    console.log(`Working providers: ${working}/${total}`);
+    if (result.envFileBackup) {
+        console.log(chalk.gray(`Environment backup: ${result.envFileBackup}`));
+    }
+    if (working > 0) {
+        console.log(chalk.green("\n✅ Setup completed successfully!"));
+        console.log(chalk.yellow("💡 You can now use NeuroLink with your configured providers."));
+        console.log(chalk.gray('   Try: neurolink generate-text "Hello, AI!"'));
+    }
+    else {
+        console.log(chalk.red("\n❌ No providers are working."));
+        console.log(chalk.yellow("💡 Please check your credentials and try again."));
+        console.log(chalk.gray("   Run: neurolink config setup"));
+    }
+    console.log(chalk.blue("\n📚 Documentation: https://github.com/juspay/neurolink#setup"));
+}

package/dist/core/dynamic-models.d.ts

@@ -0,0 +1,208 @@
+import { z } from "zod";
+/**
+ * Model configuration schema for validation
+ */
+declare const ModelConfigSchema: z.ZodObject<{
+    id: z.ZodString;
+    displayName: z.ZodString;
+    capabilities: z.ZodArray<z.ZodString, "many">;
+    deprecated: z.ZodBoolean;
+    pricing: z.ZodObject<{
+        input: z.ZodNumber;
+        output: z.ZodNumber;
+    }, "strip", z.ZodTypeAny, {
+        input: number;
+        output: number;
+    }, {
+        input: number;
+        output: number;
+    }>;
+    contextWindow: z.ZodNumber;
+    releaseDate: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    id: string;
+    displayName: string;
+    capabilities: string[];
+    deprecated: boolean;
+    pricing: {
+        input: number;
+        output: number;
+    };
+    contextWindow: number;
+    releaseDate: string;
+}, {
+    id: string;
+    displayName: string;
+    capabilities: string[];
+    deprecated: boolean;
+    pricing: {
+        input: number;
+        output: number;
+    };
+    contextWindow: number;
+    releaseDate: string;
+}>;
+declare const ModelRegistrySchema: z.ZodObject<{
+    version: z.ZodString;
+    lastUpdated: z.ZodString;
+    models: z.ZodRecord<z.ZodString, z.ZodRecord<z.ZodString, z.ZodObject<{
+        id: z.ZodString;
+        displayName: z.ZodString;
+        capabilities: z.ZodArray<z.ZodString, "many">;
+        deprecated: z.ZodBoolean;
+        pricing: z.ZodObject<{
+            input: z.ZodNumber;
+            output: z.ZodNumber;
+        }, "strip", z.ZodTypeAny, {
+            input: number;
+            output: number;
+        }, {
+            input: number;
+            output: number;
+        }>;
+        contextWindow: z.ZodNumber;
+        releaseDate: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        id: string;
+        displayName: string;
+        capabilities: string[];
+        deprecated: boolean;
+        pricing: {
+            input: number;
+            output: number;
+        };
+        contextWindow: number;
+        releaseDate: string;
+    }, {
+        id: string;
+        displayName: string;
+        capabilities: string[];
+        deprecated: boolean;
+        pricing: {
+            input: number;
+            output: number;
+        };
+        contextWindow: number;
+        releaseDate: string;
+    }>>>;
+    aliases: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+    defaults: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+}, "strip", z.ZodTypeAny, {
+    version: string;
+    lastUpdated: string;
+    models: Record<string, Record<string, {
+        id: string;
+        displayName: string;
+        capabilities: string[];
+        deprecated: boolean;
+        pricing: {
+            input: number;
+            output: number;
+        };
+        contextWindow: number;
+        releaseDate: string;
+    }>>;
+    aliases?: Record<string, string> | undefined;
+    defaults?: Record<string, string> | undefined;
+}, {
+    version: string;
+    lastUpdated: string;
+    models: Record<string, Record<string, {
+        id: string;
+        displayName: string;
+        capabilities: string[];
+        deprecated: boolean;
+        pricing: {
+            input: number;
+            output: number;
+        };
+        contextWindow: number;
+        releaseDate: string;
+    }>>;
+    aliases?: Record<string, string> | undefined;
+    defaults?: Record<string, string> | undefined;
+}>;
+export type ModelConfig = z.infer<typeof ModelConfigSchema>;
+export type ModelRegistry = z.infer<typeof ModelRegistrySchema>;
+/**
+ * Dynamic Model Provider
+ * Loads and manages model configurations from external sources
+ */
+export declare class DynamicModelProvider {
+    private static instance;
+    private modelRegistry;
+    private lastFetch;
+    private readonly CACHE_DURATION;
+    private constructor();
+    static getInstance(): DynamicModelProvider;
+    /**
+     * Initialize the model registry from multiple sources
+     */
+    initialize(): Promise<void>;
+    /**
+     * Load configuration from a source (URL or file path)
+     */
+    private loadFromSource;
+    /**
+     * Get all available models for a provider
+     */
+    getModelsForProvider(provider: string): Record<string, ModelConfig>;
+    /**
+     * Resolve a model by provider and model hint
+     */
+    resolveModel(provider: string, modelHint?: string): ModelConfig | null;
+    /**
+     * Search models by capabilities
+     */
+    searchByCapability(capability: string, options?: {
+        provider?: string;
+        maxPrice?: number;
+        excludeDeprecated?: boolean;
+    }): Array<{
+        provider: string;
+        model: string;
+        config: ModelConfig;
+    }>;
+    /**
+     * Get the best model for a specific use case
+     */
+    getBestModelFor(useCase: "coding" | "analysis" | "vision" | "fastest" | "cheapest"): {
+        provider: string;
+        model: string;
+        config: ModelConfig;
+    } | null;
+    /**
+     * Get all models across all providers
+     */
+    getAllModels(): Array<{
+        provider: string;
+        model: string;
+        config: ModelConfig;
+    }>;
+    /**
+     * Get total number of models
+     */
+    getTotalModelCount(): number;
+    /**
+     * Check if cache needs refresh
+     */
+    needsRefresh(): boolean;
+    /**
+     * Force refresh the model registry
+     */
+    refresh(): Promise<void>;
+    /**
+     * Ensure the registry is initialized
+     */
+    private ensureInitialized;
+    /**
+     * Get registry metadata
+     */
+    getMetadata(): {
+        version: string;
+        lastUpdated: string;
+        modelCount: number;
+    } | null;
+}
+export declare const dynamicModelProvider: DynamicModelProvider;
+export {};